examples/distmap/src/test/java/org/apache/etch/examples/distmap/TestRemoteDistributedHashTableServer.java - etch - Git at Google

 /* $Id$
  *
  * 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.etch.examples.distmap;

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

 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;

 import org.apache.etch.bindings.java.support.ServerFactory;
 import org.apache.etch.examples.distmap.DistributedHashTable.Entry;
 import org.apache.etch.util.Log;
 import org.apache.etch.util.core.io.Transport;
 import org.junit.After;
 import org.junit.AfterClass;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Test;


 /**
  * Tests the RemoteDistributedHashTableServer interface. Start the
  * TestRemoteDistributedHashTableListener first.
  */
 public class TestRemoteDistributedHashTableServer
 {
 	final boolean myBool1 = true;
 	final byte myByte1 = 127;
 	final short myShort1 = 32767;
 	final int myInt1 = 2147483647;
 	final long myLong1 = 9223372036854775807l;
 	final float myFloat1 = 3.4028235e38f;
 	final double myDouble1 = 1.7976931348623157e308d;
 	final String myString1 = "abc";
 	RemoteDistributedHashTableServer server;

 	final Map<String, Object> map =
 		Collections.synchronizedMap( new HashMap<String, Object>() );

 	/** transport */
 	public static ServerFactory listener;

 	/**
 	 * Connect to service
 	 * @throws Exception
 	 */
 	@BeforeClass
 	public static void startRemoteDistributedHashTableListener()
 		throws Exception
 	{
 		Log.addSink(null);
 		Log.report("StartRemoteDistributedHashTableListener");

 		String uri = "tcp://localhost:4007";
 		MainDistributedHashTableListener implFactory = new MainDistributedHashTableListener();
 		listener = DistributedHashTableHelper.newListener( uri, null, implFactory );

 		listener.transportControl( Transport.START_AND_WAIT_UP, 4000 );

 		Log.report( "StartDistributedHashTableRemoteListener",
 				"status", "up",
 				"listener", listener );
 	}

 	/**
 	 * Shut down listener
 	 * @throws Exception
 	 */
 	@AfterClass
 	public static void shutListener()
 		throws Exception
 	{
 		if (listener != null)
 		{
 			listener.transportControl( Transport.STOP_AND_WAIT_DOWN, 4000 );
 			listener = null;
 		}
 	}

 	/**
 	 * Connect client to serve
 	 * @throws Exception
 	 */
 	@Before
 	public void makeConnection()
 		throws Exception
 	{
 		String uri = "tcp://localhost:4007";

 		server = DistributedHashTableHelper.newServer( uri, null, new MainDistributedHashTableClient() );

 		server._startAndWaitUp( 4000 );
 	}

 	/**
 	 * Shutdown connection to server
 	 * @throws Exception
 	 */
 	@After
 	public void shutConnection()
 		throws Exception
 	{
 		if (server != null)
 		{
 			server._stop();
 		}
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a boolean type
 	 * @throws Exception
 	 */
 	@Test
 	public void testBoolean()
 		throws Exception
 	{
 		Object myObj = server.putObject("boolean1", myBool1);
 		assertNull( myObj );

 		myObj = server.getObject("boolean1");
 		assertEquals( myBool1, myObj );

 		server.removeObject("boolean1");
 		myObj = server.getObject("boolean1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a byte type
 	 * @throws Exception
 	 */
 	@Test
 	public void testByte()
 		throws Exception
 	{
 		Object myObj = server.putObject("byte1", myByte1);
 		assertNull( myObj );

 		myObj = server.getObject("byte1");
 		assertEquals( myByte1, myObj );

 		server.removeObject("byte1");
 		myObj = server.getObject("byte1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a short type
 	 * @throws Exception
 	 */
 	@Test
 	public void testShort()
 		throws Exception
 	{
 		Object myObj = server.putObject("short1", myShort1);
 		assertNull( myObj );

 		myObj = server.getObject("short1");
 		assertEquals( myShort1, myObj );

 		server.removeObject("short1");
 		myObj = server.getObject("short1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for an int type
 	 * @throws Exception
 	 */
 	@Test
 	public void testInt()
 		throws Exception
 	{
 		Object myObj = server.putObject("int1", myInt1);
 		assertNull( myObj );

 		myObj = server.getObject("int1");
 		assertEquals( myInt1, myObj );

 		server.removeObject("int1");
 		myObj = server.getObject("int1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a long type
 	 * @throws Exception
 	 */
 	@Test
 	public void testLong()
 		throws Exception
 	{
 		Object myObj = server.putObject("long1", myLong1);
 		assertNull( myObj );

 		myObj = server.getObject("long1");
 		assertEquals( myLong1, myObj );

 		server.removeObject("long1");
 		myObj = server.getObject("long1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a float type
 	 * @throws Exception
 	 */
 	@Test
 	public void testFloat()
 		throws Exception
 	{
 		Object myObj = server.putObject("float1", myFloat1);
 		assertNull( myObj );

 		myObj = server.getObject("float1");
 		assertEquals( myFloat1, myObj );

 		server.removeObject("float1");
 		myObj = server.getObject("float1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a double type
 	 * @throws Exception
 	 */
 	@Test
 	public void testDouble()
 		throws Exception
 	{
 		Object myObj = server.putObject("double1", myDouble1);
 		assertNull( myObj );

 		myObj = server.getObject("double1");
 		assertEquals( myDouble1, myObj );

 		server.removeObject("double1");
 		myObj = server.getObject("double1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify that putObject, getObject and removeObject
 	 * work for a string type
 	 * @throws Exception
 	 */
 	@Test
 	public void testString()
 		throws Exception
 	{
 		Object myObj = server.putObject("string1", myString1);
 		assertNull( myObj );

 		myObj = server.getObject("string1");
 		assertEquals( myString1, myObj );

 		server.removeObject("string1");
 		myObj = server.getObject("string1");
 		assertNull( myObj );
 	}

 	/**
 	 * Verify size
 	 * @throws Exception
 	 */
 	@Test
 	public void testSize()
 		throws Exception
 	{
 		/*
 		 * Verify that size is initially zero
 		 */
 		int size = server.size();
 		assertEquals(size, 0);

 		/*
 		 * Add items and verify size
 		 */
 		server.putObject("string1", "abc");
 		server.putObject("string2", "xyz");
 		server.putObject("int1", 0);
 		server.putObject("int2", 125);

 		size = server.size();
 		assertEquals(size,4);

 		/*
 		 * Remove items and verify size
 		 */
 		server.removeObject("string1");
 		server.removeObject("string2");
 		server.removeObject("int1");
 		server.removeObject("int2");

 		size = server.size();
 		assertEquals(size, 0);
 	}

 	/**
 	 * Verify that getAll returns all entries
 	 * @throws Exception
 	 */
 	@Test
 	public void testGetAll()
 		throws Exception
 	{
 		/*
 		 * Verify that map is empty to start
 		 */
 		int size = server.size();
 		assertEquals(size, 0);

 		/*
 		 * Put items in map
 		 */
 		server.putObject("string1", "abc");
 		server.putObject("string2", "xyz");
 		server.putObject("int1", 0);
 		server.putObject("int2", 125);

 		/*
 		 * Verify that 4 items were inserted into the map
 		 */
 		int counter = 0;
 		for(Entry entry: server.getAll())
 		{
 			if(entry.key != null && entry.value != null)
 			{
 				counter++;
 			}
 		}

 		assertEquals(counter, 4);

 		/*
 		 * Remove all items added to map
 		 */
 		server.removeObject("string1");
 		server.removeObject("string2");
 		server.removeObject("int1");
 		server.removeObject("int2");

 		/*
 		 * Verify that map is empty when all known items are removed
 		 */
 		size = server.size();
 		assertEquals(size, 0);
 	}
 }
	/* $Id$
	*
	* 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.etch.examples.distmap;

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

	import java.util.Collections;
	import java.util.HashMap;
	import java.util.Map;

	import org.apache.etch.bindings.java.support.ServerFactory;
	import org.apache.etch.examples.distmap.DistributedHashTable.Entry;
	import org.apache.etch.util.Log;
	import org.apache.etch.util.core.io.Transport;
	import org.junit.After;
	import org.junit.AfterClass;
	import org.junit.Before;
	import org.junit.BeforeClass;
	import org.junit.Test;


	/**
	* Tests the RemoteDistributedHashTableServer interface. Start the
	* TestRemoteDistributedHashTableListener first.
	*/
	public class TestRemoteDistributedHashTableServer
	{
	final boolean myBool1 = true;
	final byte myByte1 = 127;
	final short myShort1 = 32767;
	final int myInt1 = 2147483647;
	final long myLong1 = 9223372036854775807l;
	final float myFloat1 = 3.4028235e38f;
	final double myDouble1 = 1.7976931348623157e308d;
	final String myString1 = "abc";
	RemoteDistributedHashTableServer server;

	final Map<String, Object> map =
	Collections.synchronizedMap( new HashMap<String, Object>() );

	/** transport */
	public static ServerFactory listener;

	/**
	* Connect to service
	* @throws Exception
	*/
	@BeforeClass
	public static void startRemoteDistributedHashTableListener()
	throws Exception
	{
	Log.addSink(null);
	Log.report("StartRemoteDistributedHashTableListener");

	String uri = "tcp://localhost:4007";
	MainDistributedHashTableListener implFactory = new MainDistributedHashTableListener();
	listener = DistributedHashTableHelper.newListener( uri, null, implFactory );

	listener.transportControl( Transport.START_AND_WAIT_UP, 4000 );

	Log.report( "StartDistributedHashTableRemoteListener",
	"status", "up",
	"listener", listener );
	}

	/**
	* Shut down listener
	* @throws Exception
	*/
	@AfterClass
	public static void shutListener()
	throws Exception
	{
	if (listener != null)
	{
	listener.transportControl( Transport.STOP_AND_WAIT_DOWN, 4000 );
	listener = null;
	}
	}

	/**
	* Connect client to serve
	* @throws Exception
	*/
	@Before
	public void makeConnection()
	throws Exception
	{
	String uri = "tcp://localhost:4007";

	server = DistributedHashTableHelper.newServer( uri, null, new MainDistributedHashTableClient() );

	server._startAndWaitUp( 4000 );
	}

	/**
	* Shutdown connection to server
	* @throws Exception
	*/
	@After
	public void shutConnection()
	throws Exception
	{
	if (server != null)
	{
	server._stop();
	}
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a boolean type
	* @throws Exception
	*/
	@Test
	public void testBoolean()
	throws Exception
	{
	Object myObj = server.putObject("boolean1", myBool1);
	assertNull( myObj );

	myObj = server.getObject("boolean1");
	assertEquals( myBool1, myObj );

	server.removeObject("boolean1");
	myObj = server.getObject("boolean1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a byte type
	* @throws Exception
	*/
	@Test
	public void testByte()
	throws Exception
	{
	Object myObj = server.putObject("byte1", myByte1);
	assertNull( myObj );

	myObj = server.getObject("byte1");
	assertEquals( myByte1, myObj );

	server.removeObject("byte1");
	myObj = server.getObject("byte1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a short type
	* @throws Exception
	*/
	@Test
	public void testShort()
	throws Exception
	{
	Object myObj = server.putObject("short1", myShort1);
	assertNull( myObj );

	myObj = server.getObject("short1");
	assertEquals( myShort1, myObj );

	server.removeObject("short1");
	myObj = server.getObject("short1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for an int type
	* @throws Exception
	*/
	@Test
	public void testInt()
	throws Exception
	{
	Object myObj = server.putObject("int1", myInt1);
	assertNull( myObj );

	myObj = server.getObject("int1");
	assertEquals( myInt1, myObj );

	server.removeObject("int1");
	myObj = server.getObject("int1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a long type
	* @throws Exception
	*/
	@Test
	public void testLong()
	throws Exception
	{
	Object myObj = server.putObject("long1", myLong1);
	assertNull( myObj );

	myObj = server.getObject("long1");
	assertEquals( myLong1, myObj );

	server.removeObject("long1");
	myObj = server.getObject("long1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a float type
	* @throws Exception
	*/
	@Test
	public void testFloat()
	throws Exception
	{
	Object myObj = server.putObject("float1", myFloat1);
	assertNull( myObj );

	myObj = server.getObject("float1");
	assertEquals( myFloat1, myObj );

	server.removeObject("float1");
	myObj = server.getObject("float1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a double type
	* @throws Exception
	*/
	@Test
	public void testDouble()
	throws Exception
	{
	Object myObj = server.putObject("double1", myDouble1);
	assertNull( myObj );

	myObj = server.getObject("double1");
	assertEquals( myDouble1, myObj );

	server.removeObject("double1");
	myObj = server.getObject("double1");
	assertNull( myObj );
	}

	/**
	* Verify that putObject, getObject and removeObject
	* work for a string type
	* @throws Exception
	*/
	@Test
	public void testString()
	throws Exception
	{
	Object myObj = server.putObject("string1", myString1);
	assertNull( myObj );

	myObj = server.getObject("string1");
	assertEquals( myString1, myObj );

	server.removeObject("string1");
	myObj = server.getObject("string1");
	assertNull( myObj );
	}

	/**
	* Verify size
	* @throws Exception
	*/
	@Test
	public void testSize()
	throws Exception
	{
	/*
	* Verify that size is initially zero
	*/
	int size = server.size();
	assertEquals(size, 0);

	/*
	* Add items and verify size
	*/
	server.putObject("string1", "abc");
	server.putObject("string2", "xyz");
	server.putObject("int1", 0);
	server.putObject("int2", 125);

	size = server.size();
	assertEquals(size,4);

	/*
	* Remove items and verify size
	*/
	server.removeObject("string1");
	server.removeObject("string2");
	server.removeObject("int1");
	server.removeObject("int2");

	size = server.size();
	assertEquals(size, 0);
	}

	/**
	* Verify that getAll returns all entries
	* @throws Exception
	*/
	@Test
	public void testGetAll()
	throws Exception
	{
	/*
	* Verify that map is empty to start
	*/
	int size = server.size();
	assertEquals(size, 0);

	/*
	* Put items in map
	*/
	server.putObject("string1", "abc");
	server.putObject("string2", "xyz");
	server.putObject("int1", 0);
	server.putObject("int2", 125);

	/*
	* Verify that 4 items were inserted into the map
	*/
	int counter = 0;
	for(Entry entry: server.getAll())
	{
	if(entry.key != null && entry.value != null)
	{
	counter++;
	}
	}

	assertEquals(counter, 4);

	/*
	* Remove all items added to map
	*/
	server.removeObject("string1");
	server.removeObject("string2");
	server.removeObject("int1");
	server.removeObject("int2");

	/*
	* Verify that map is empty when all known items are removed
	*/
	size = server.size();
	assertEquals(size, 0);
	}
	}