perftests/src/test/java/org/apache/qpid/disttest/message/JsonHandlerTest.java - qpid-broker-j - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership.  The ASF licenses this file
  * to you under the Apache License, Version 2.0 (the
  * "License"); you may not use this file except in compliance
  * with the License.  You may obtain a copy of the License at
  *
  *   http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing,
  * software distributed under the License is distributed on an
  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  * KIND, either express or implied.  See the License for the
  * specific language governing permissions and limitations
  * under the License.
  *
  */
 package org.apache.qpid.disttest.message;

 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;

 import org.junit.Assert;

 import org.apache.qpid.disttest.client.property.ListPropertyValue;
 import org.apache.qpid.disttest.client.property.PropertyValue;
 import org.apache.qpid.disttest.json.JsonHandler;

 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.After;
 import org.junit.Test;

 import org.apache.qpid.test.utils.UnitTestBase;
 import static org.junit.Assert.assertSame;
 import static org.junit.Assert.assertNotSame;
 import static org.junit.Assert.fail;
 import static org.junit.Assert.assertNotNull;

 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;

 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNull;

 public class JsonHandlerTest extends UnitTestBase
 {
     private JsonHandler _jsonHandler = null;
     private SendChristmasCards _testCommand = null;

     @Before
     public void setUp() throws Exception
     {

         _jsonHandler = new JsonHandler();

         _testCommand = new SendChristmasCards(CommandType.START_TEST, Collections.singletonMap(SendChristmasCards.CardType.FUNNY, 5));
         _testCommand._persons = Arrays.asList(new Person("Phil"), new Person("Andrew"));
     }

     @Test
     public void testMarshallUnmarshall() throws Exception
     {
         final String jsonString = _jsonHandler.marshall(_testCommand);

         final SendChristmasCards unmarshalledCommand = _jsonHandler.unmarshall(jsonString, SendChristmasCards.class);

         assertEquals("Unmarshalled command should be equal to the original object",
                             _testCommand,
                             unmarshalledCommand);

     }

     @Test
     public void testSimplePropertyValueMarshallUnmarshall() throws Exception
     {
         String json = "{'_messageProperties': {'test': 1}}";
         final TestCommand unmarshalledCommand = _jsonHandler.unmarshall(json, TestCommand.class);

         Map<String, PropertyValue> properties = unmarshalledCommand.getMessageProperties();
         assertNotNull("Properties should not be null", properties);
         assertFalse("Properties should not be empty", properties.isEmpty());
         assertEquals("Unexpected properties size", (long) 1, (long) properties.size());
         PropertyValue testProperty = properties.get("test");
         assertNotNull("Unexpected property test", testProperty);
         final boolean condition = testProperty.getValue() instanceof Number;
         assertTrue("Unexpected property test", condition);
         assertEquals("Unexpected property value", (long) 1, (long) ((Number) testProperty.getValue()).intValue());

         String newJson =_jsonHandler.marshall(unmarshalledCommand);
         final TestCommand newUnmarshalledCommand = _jsonHandler.unmarshall(newJson, TestCommand.class);
         assertEquals("Unmarshalled command should be equal to the original object",
                             unmarshalledCommand,
                             newUnmarshalledCommand);
     }

     @Test
     public void testGeneratorDesrialization() throws Exception
     {
         String json = "{'_messageProperties': {'test': 1, 'generator': {'@def': 'list',  '_cyclic': false, '_items': ['first', " +
                 "{'@def': 'range', '_upper':10, '_type':'int'}]}}}";
         final TestCommand unmarshalledCommand = _jsonHandler.unmarshall(json, TestCommand.class);

         Map<String, PropertyValue> properties = unmarshalledCommand.getMessageProperties();
         assertNotNull("Properties should not be null", properties);
         assertFalse("Properties should not be empty", properties.isEmpty());
         assertEquals("Unexpected properties size", (long) 2, (long) properties.size());
         PropertyValue testProperty = properties.get("test");
         assertNotNull("Unexpected property test", testProperty);
         final boolean condition1 = testProperty.getValue() instanceof Number;
         assertTrue("Unexpected property test", condition1);
         assertEquals("Unexpected property value", (long) 1, (long) ((Number) testProperty.getValue()).intValue());
         Object generatorObject = properties.get("generator");

         final boolean condition = generatorObject instanceof ListPropertyValue;
         assertTrue("Unexpected generator object : " + generatorObject, condition);

         PropertyValue generator = (PropertyValue)generatorObject;
         assertEquals("Unexpected generator value", "first", generator.getValue());
         for (int i = 0; i < 10; i++)
         {
             assertEquals("Unexpected generator value", Integer.valueOf(i), generator.getValue());
         }

         String newJson =_jsonHandler.marshall(unmarshalledCommand);
         final TestCommand newUnmarshalledCommand = _jsonHandler.unmarshall(newJson, TestCommand.class);
         assertEquals("Unmarshalled command should be equal to the original object",
                             unmarshalledCommand,
                             newUnmarshalledCommand);
     }

     /**
      * A {@link Command} designed to exercise {@link JsonHandler}, e.g does it handle a map of enums?.
      *
      * This class is non-private to avoid auto-deletion of "unused" fields/methods
      */
     static class SendChristmasCards extends Command
     {
         enum CardType {FUNNY, TRADITIONAL}

         private Map<CardType, Integer> _cardTypes;
         private List<Person> _persons;

         public SendChristmasCards()
         {
             this(null, null);
         }

         public SendChristmasCards(final CommandType type, Map<CardType, Integer> cardTypes)
         {
             super(type);
             _cardTypes = cardTypes;
         }

         public Map<CardType, Integer> getCardTypes()
         {
             return _cardTypes;
         }

         public List<Person> getPersons()
         {
             return _persons;
         }

         @Override
         public boolean equals(final Object o)
         {
             if (this == o)
             {
                 return true;
             }
             if (o == null || getClass() != o.getClass())
             {
                 return false;
             }

             final SendChristmasCards that = (SendChristmasCards) o;

             if (getCardTypes() != null ? !getCardTypes().equals(that.getCardTypes()) : that.getCardTypes() != null)
             {
                 return false;
             }
             return !(getPersons() != null ? !getPersons().equals(that.getPersons()) : that.getPersons() != null);

         }

         @Override
         public int hashCode()
         {
             int result = getCardTypes() != null ? getCardTypes().hashCode() : 0;
             result = 31 * result + (getPersons() != null ? getPersons().hashCode() : 0);
             return result;
         }
     }

     /**
      * This class is non-private to avoid auto-deletion of "unused" fields/methods
      */
     static class Person
     {
         private String _firstName;

         public Person()
         {
             this(null);
         }

         public Person(final String firstName)
         {
             _firstName = firstName;
         }

         public String getFirstName()
         {
             return _firstName;
         }

         @Override
         public boolean equals(final Object o)
         {
             if (this == o)
             {
                 return true;
             }
             if (o == null || getClass() != o.getClass())
             {
                 return false;
             }

             final Person person = (Person) o;

             return !(getFirstName() != null
                     ? !getFirstName().equals(person.getFirstName())
                     : person.getFirstName() != null);

         }

         @Override
         public int hashCode()
         {
             return getFirstName() != null ? getFirstName().hashCode() : 0;
         }
     }

     /**
      * Yet another test class
      */
     static class TestCommand extends Command
     {

         private Map<String, PropertyValue> _messageProperties;

         public TestCommand()
         {
             super(null);
         }

         public TestCommand(CommandType type)
         {
             super(type);
         }

         public Map<String, PropertyValue> getMessageProperties()
         {
             return _messageProperties;
         }

         public void setMessageProperties(Map<String, PropertyValue> _messageProperties)
         {
             this._messageProperties = _messageProperties;
         }

         @Override
         public boolean equals(final Object obj)
         {
             if (obj == null || !(obj instanceof TestCommand))
             {
                 return false;
             }
             TestCommand other = (TestCommand)obj;
             if (_messageProperties == null && other._messageProperties != null )
             {
                 return false;
             }
             return _messageProperties.equals(other._messageProperties);
         }
     }
 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one
	* or more contributor license agreements. See the NOTICE file
	* distributed with this work for additional information
	* regarding copyright ownership. The ASF licenses this file
	* to you under the Apache License, Version 2.0 (the
	* "License"); you may not use this file except in compliance
	* with the License. You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing,
	* software distributed under the License is distributed on an
	* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
	* KIND, either express or implied. See the License for the
	* specific language governing permissions and limitations
	* under the License.
	*
	*/
	package org.apache.qpid.disttest.message;

	import java.util.Arrays;
	import java.util.Collections;
	import java.util.List;
	import java.util.Map;

	import org.junit.Assert;

	import org.apache.qpid.disttest.client.property.ListPropertyValue;
	import org.apache.qpid.disttest.client.property.PropertyValue;
	import org.apache.qpid.disttest.json.JsonHandler;

	import org.junit.Assert;
	import org.junit.Before;
	import org.junit.After;
	import org.junit.Test;

	import org.apache.qpid.test.utils.UnitTestBase;
	import static org.junit.Assert.assertSame;
	import static org.junit.Assert.assertNotSame;
	import static org.junit.Assert.fail;
	import static org.junit.Assert.assertNotNull;

	import static org.junit.Assert.assertFalse;
	import static org.junit.Assert.assertTrue;

	import static org.junit.Assert.assertEquals;
	import static org.junit.Assert.assertNull;

	public class JsonHandlerTest extends UnitTestBase
	{
	private JsonHandler _jsonHandler = null;
	private SendChristmasCards _testCommand = null;

	@Before
	public void setUp() throws Exception
	{

	_jsonHandler = new JsonHandler();

	_testCommand = new SendChristmasCards(CommandType.START_TEST, Collections.singletonMap(SendChristmasCards.CardType.FUNNY, 5));
	_testCommand._persons = Arrays.asList(new Person("Phil"), new Person("Andrew"));
	}

	@Test
	public void testMarshallUnmarshall() throws Exception
	{
	final String jsonString = _jsonHandler.marshall(_testCommand);

	final SendChristmasCards unmarshalledCommand = _jsonHandler.unmarshall(jsonString, SendChristmasCards.class);

	assertEquals("Unmarshalled command should be equal to the original object",
	_testCommand,
	unmarshalledCommand);

	}

	@Test
	public void testSimplePropertyValueMarshallUnmarshall() throws Exception
	{
	String json = "{'_messageProperties': {'test': 1}}";
	final TestCommand unmarshalledCommand = _jsonHandler.unmarshall(json, TestCommand.class);

	Map<String, PropertyValue> properties = unmarshalledCommand.getMessageProperties();
	assertNotNull("Properties should not be null", properties);
	assertFalse("Properties should not be empty", properties.isEmpty());
	assertEquals("Unexpected properties size", (long) 1, (long) properties.size());
	PropertyValue testProperty = properties.get("test");
	assertNotNull("Unexpected property test", testProperty);
	final boolean condition = testProperty.getValue() instanceof Number;
	assertTrue("Unexpected property test", condition);
	assertEquals("Unexpected property value", (long) 1, (long) ((Number) testProperty.getValue()).intValue());

	String newJson =_jsonHandler.marshall(unmarshalledCommand);
	final TestCommand newUnmarshalledCommand = _jsonHandler.unmarshall(newJson, TestCommand.class);
	assertEquals("Unmarshalled command should be equal to the original object",
	unmarshalledCommand,
	newUnmarshalledCommand);
	}

	@Test
	public void testGeneratorDesrialization() throws Exception
	{
	String json = "{'_messageProperties': {'test': 1, 'generator': {'@def': 'list', '_cyclic': false, '_items': ['first', " +
	"{'@def': 'range', '_upper':10, '_type':'int'}]}}}";
	final TestCommand unmarshalledCommand = _jsonHandler.unmarshall(json, TestCommand.class);

	Map<String, PropertyValue> properties = unmarshalledCommand.getMessageProperties();
	assertNotNull("Properties should not be null", properties);
	assertFalse("Properties should not be empty", properties.isEmpty());
	assertEquals("Unexpected properties size", (long) 2, (long) properties.size());
	PropertyValue testProperty = properties.get("test");
	assertNotNull("Unexpected property test", testProperty);
	final boolean condition1 = testProperty.getValue() instanceof Number;
	assertTrue("Unexpected property test", condition1);
	assertEquals("Unexpected property value", (long) 1, (long) ((Number) testProperty.getValue()).intValue());
	Object generatorObject = properties.get("generator");

	final boolean condition = generatorObject instanceof ListPropertyValue;
	assertTrue("Unexpected generator object : " + generatorObject, condition);

	PropertyValue generator = (PropertyValue)generatorObject;
	assertEquals("Unexpected generator value", "first", generator.getValue());
	for (int i = 0; i < 10; i++)
	{
	assertEquals("Unexpected generator value", Integer.valueOf(i), generator.getValue());
	}

	String newJson =_jsonHandler.marshall(unmarshalledCommand);
	final TestCommand newUnmarshalledCommand = _jsonHandler.unmarshall(newJson, TestCommand.class);
	assertEquals("Unmarshalled command should be equal to the original object",
	unmarshalledCommand,
	newUnmarshalledCommand);
	}

	/**
	* A {@link Command} designed to exercise {@link JsonHandler}, e.g does it handle a map of enums?.
	*
	* This class is non-private to avoid auto-deletion of "unused" fields/methods
	*/
	static class SendChristmasCards extends Command
	{
	enum CardType {FUNNY, TRADITIONAL}

	private Map<CardType, Integer> _cardTypes;
	private List<Person> _persons;

	public SendChristmasCards()
	{
	this(null, null);
	}

	public SendChristmasCards(final CommandType type, Map<CardType, Integer> cardTypes)
	{
	super(type);
	_cardTypes = cardTypes;
	}

	public Map<CardType, Integer> getCardTypes()
	{
	return _cardTypes;
	}

	public List<Person> getPersons()
	{
	return _persons;
	}

	@Override
	public boolean equals(final Object o)
	{
	if (this == o)
	{
	return true;
	}
	if (o == null \|\| getClass() != o.getClass())
	{
	return false;
	}

	final SendChristmasCards that = (SendChristmasCards) o;

	if (getCardTypes() != null ? !getCardTypes().equals(that.getCardTypes()) : that.getCardTypes() != null)
	{
	return false;
	}
	return !(getPersons() != null ? !getPersons().equals(that.getPersons()) : that.getPersons() != null);

	}

	@Override
	public int hashCode()
	{
	int result = getCardTypes() != null ? getCardTypes().hashCode() : 0;
	result = 31 * result + (getPersons() != null ? getPersons().hashCode() : 0);
	return result;
	}
	}

	/**
	* This class is non-private to avoid auto-deletion of "unused" fields/methods
	*/
	static class Person
	{
	private String _firstName;

	public Person()
	{
	this(null);
	}

	public Person(final String firstName)
	{
	_firstName = firstName;
	}

	public String getFirstName()
	{
	return _firstName;
	}

	@Override
	public boolean equals(final Object o)
	{
	if (this == o)
	{
	return true;
	}
	if (o == null \|\| getClass() != o.getClass())
	{
	return false;
	}

	final Person person = (Person) o;

	return !(getFirstName() != null
	? !getFirstName().equals(person.getFirstName())
	: person.getFirstName() != null);

	}

	@Override
	public int hashCode()
	{
	return getFirstName() != null ? getFirstName().hashCode() : 0;
	}
	}

	/**
	* Yet another test class
	*/
	static class TestCommand extends Command
	{

	private Map<String, PropertyValue> _messageProperties;

	public TestCommand()
	{
	super(null);
	}

	public TestCommand(CommandType type)
	{
	super(type);
	}

	public Map<String, PropertyValue> getMessageProperties()
	{
	return _messageProperties;
	}

	public void setMessageProperties(Map<String, PropertyValue> _messageProperties)
	{
	this._messageProperties = _messageProperties;
	}

	@Override
	public boolean equals(final Object obj)
	{
	if (obj == null \|\| !(obj instanceof TestCommand))
	{
	return false;
	}
	TestCommand other = (TestCommand)obj;
	if (_messageProperties == null && other._messageProperties != null )
	{
	return false;
	}
	return _messageProperties.equals(other._messageProperties);
	}
	}
	}