From 316951efbaaba3bb64c3a4d81611987de5443686 Mon Sep 17 00:00:00 2001 From: Rob Austin Date: Thu, 13 Nov 2014 15:58:31 +0000 Subject: [PATCH] Improved the stateless client test cases --- .../chronicle/map/StatelessClientTest.java | 344 ++++++++++++++++++ 1 file changed, 344 insertions(+) create mode 100644 src/test/java/net/openhft/chronicle/map/StatelessClientTest.java diff --git a/src/test/java/net/openhft/chronicle/map/StatelessClientTest.java b/src/test/java/net/openhft/chronicle/map/StatelessClientTest.java new file mode 100644 index 000000000..507c94ff0 --- /dev/null +++ b/src/test/java/net/openhft/chronicle/map/StatelessClientTest.java @@ -0,0 +1,344 @@ +/* + * Copyright 2014 Higher Frequency Trading + * + * http://www.higherfrequencytrading.com + * + * Licensed 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 net.openhft.chronicle.map; + +import net.openhft.chronicle.hash.replication.TcpTransportAndNetworkConfig; +import org.junit.Assert; +import org.junit.Test; + +import java.io.IOException; +import java.net.InetSocketAddress; +import java.util.Collection; +import java.util.HashMap; +import java.util.Map; +import java.util.Set; + + +/** + * @author Rob Austin. + */ +public class StatelessClientTest { + + + public static final int SIZE = 100000; + + + @Test(timeout = 5000) + public void testBufferOverFlowPutAllAndEntrySet() throws IOException, InterruptedException { + + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + Map payload = new HashMap(); + + for (int i = 0; i < SIZE; i++) { + payload.put(i, "some value=" + i); + } + + + statelessMap.putAll(payload); + + + Set> entries = statelessMap.entrySet(); + + Map.Entry next = entries.iterator().next(); + Assert.assertEquals("some value=" + next.getKey(), next.getValue()); + + Assert.assertEquals(entries.size(), SIZE); + + } + } + } + + + @Test + public void testBufferOverFlowPutAllAndValues() throws IOException, InterruptedException { + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + Map payload = new HashMap(); + + for (int i = 0; i < SIZE; i++) { + payload.put(i, "some value=" + i); + } + + + statelessMap.putAll(payload); + + + Collection values = statelessMap.values(); + + + Assert.assertEquals(values.size(), SIZE); + + } + } + } + + + @Test + public void testBufferOverFlowPutAllAndKeySet() throws IOException, InterruptedException { + + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + Map payload = new HashMap(); + + for (int i = 0; i < SIZE; i++) { + payload.put(i, "some value=" + i); + } + + + statelessMap.putAll(payload); + + + final Set keys = statelessMap.keySet(); + + Assert.assertEquals(keys.size(), payload.size()); + + + } + } + } + + + @Test + public void test() throws IOException, InterruptedException { + + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + serverMap.put(10, "EXAMPLE-10"); + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + Assert.assertEquals("EXAMPLE-10", statelessMap.get(10)); + Assert.assertEquals(1, statelessMap.size()); + } + } + } + + + @Test + public void testClientCreatedBeforeServer() throws IOException, InterruptedException { + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + + serverMap.put(10, "EXAMPLE-10"); + + Assert.assertEquals("EXAMPLE-10", statelessMap.get(10)); + Assert.assertEquals(1, statelessMap.size()); + + } + + + } + + + } + + + @Test + public void testBufferOverFlowPutAll() throws IOException, InterruptedException { + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + Map payload = new HashMap(); + + for (int i = 0; i < SIZE; i++) { + payload.put(i, "some value=" + i); + } + + + statelessMap.putAll(payload); + + int value = SIZE - 10; + + Assert.assertEquals("some value=" + value, statelessMap.get(value)); + Assert.assertEquals(SIZE, statelessMap.size()); + + } + } + } + + + @Test + public void testBufferOverFlowPutAllWherePutReturnsNull() throws IOException, + InterruptedException { + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + Map payload = new HashMap(); + + for (int i = 0; i < SIZE; i++) { + payload.put(i, "some value=" + i); + } + + + statelessMap.putAll(payload); + + int value = SIZE - 10; + + Assert.assertEquals("some value=" + value, statelessMap.get(value)); + Assert.assertEquals(SIZE, statelessMap.size()); + + } + } + + } + + + @Test + public void testPutWherePutReturnsNull() throws IOException, + InterruptedException { + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + statelessMap.put(1, "some value"); + + Assert.assertEquals("some value", statelessMap.get(1)); + Assert.assertEquals(1, statelessMap.size()); + + + } + } + } + + + @Test + public void testRemoveWhereRemoveReturnsNull() throws IOException, + InterruptedException { + + try (ChronicleMap serverMap = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create()) { + + try (ChronicleMap statelessMap = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create()) { + + + statelessMap.put(1, "some value"); + + Assert.assertEquals("some value", statelessMap.get(1)); + Assert.assertEquals(1, statelessMap.size()); + + statelessMap.remove(1); + + Assert.assertEquals(null, statelessMap.get(1)); + Assert.assertEquals(0, statelessMap.size()); + + } + } + + } + + @Test + public void testEquals() throws IOException, InterruptedException { + + final ChronicleMap serverMap1; + final ChronicleMap serverMap2; + final ChronicleMap statelessMap1; + final ChronicleMap statelessMap2; + + + + // server + { + serverMap1 = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8056)).create(); + serverMap2 = ChronicleMapBuilder.of(Integer.class, CharSequence.class) + .replication((byte) 2, TcpTransportAndNetworkConfig.of(8077)).create(); + + } + + + // stateless client + { + statelessMap1 = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8056)).create(); + statelessMap2 = ChronicleMapBuilder.of(Integer + .class, CharSequence.class) + .statelessClient(new InetSocketAddress("localhost", 8077)).create(); + } + + + Map payload = new HashMap(); + for (int i = 0; i < 1000; i++) { + payload.put(i, "some value=" + i); + } + + statelessMap1.putAll(payload); + Assert.assertTrue(statelessMap1.equals(payload)); + + + statelessMap1.close(); + statelessMap2.close(); + + serverMap1.close(); + serverMap2.close(); + + } + + +} \ No newline at end of file