move bmp to bmp aggregator

[bgpcep.git] / bmp / bmp-mock / src / test / java / org / opendaylight / protocol / bmp / mock / BmpMockArgumentsTest.java

diff --git a/bmp/bmp-mock/src/test/java/org/opendaylight/protocol/bmp/mock/BmpMockArgumentsTest.java b/bmp/bmp-mock/src/test/java/org/opendaylight/protocol/bmp/mock/BmpMockArgumentsTest.java
new file mode 100644 (file)
index 0000000..acc7134
--- /dev/null
+++ b/bmp/bmp-mock/src/test/java/org/opendaylight/protocol/bmp/mock/BmpMockArgumentsTest.java
@@ -0,0 +1,79 @@
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.protocol.bmp.mock;
+
+import static org.junit.Assert.assertEquals;
+
+import ch.qos.logback.classic.Level;
+import java.net.InetSocketAddress;
+import net.sourceforge.argparse4j.inf.ArgumentParserException;
+import org.junit.Test;
+
+public class BmpMockArgumentsTest {
+
+    @Test
+    public void testDefaultArguments() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {});
+        assertEquals(1, arguments.getRoutersCount());
+        assertEquals(0, arguments.getPeersCount());
+        assertEquals(0, arguments.getPrePolicyRoutesCount());
+        assertEquals(0, arguments.getPostPolicyRoutesCount());
+        assertEquals(Level.INFO, arguments.getLogLevel());
+        assertEquals(new InetSocketAddress("127.0.0.1", 0), arguments.getLocalAddress());
+        assertEquals(new InetSocketAddress("127.0.0.1", 12345), arguments.getRemoteAddress());
+    }
+
+    @Test(expected=IllegalArgumentException.class)
+    public void testWrongArgument() {
+        BmpMockArguments.parseArguments(new String[] {"--routers_count", "abcd"});
+    }
+
+    @Test
+    public void testGetRoutersCount() throws ArgumentParserException {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--routers_count", "10"});
+        assertEquals(10, arguments.getRoutersCount());
+    }
+
+    @Test
+    public void testGetPeersCount() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--peers_count", "5"});
+        assertEquals(5, arguments.getPeersCount());
+    }
+
+    @Test
+    public void testGetPrePolicyRoutesCount() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--pre_policy_routes", "20"});
+        assertEquals(20, arguments.getPrePolicyRoutesCount());
+    }
+
+    @Test
+    public void testGetPostPolicyRoutesCount() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--post_policy_routes", "100"});
+        assertEquals(100, arguments.getPostPolicyRoutesCount());
+    }
+
+    @Test
+    public void testGetLocalAddress() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--local_address", "1.2.3.4"});
+        assertEquals(new InetSocketAddress("1.2.3.4", 0), arguments.getLocalAddress());
+    }
+
+    @Test
+    public void testGetRemoteAddress() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--remote_address", "4.5.6.7:1025"});
+        assertEquals(new InetSocketAddress("4.5.6.7", 1025), arguments.getRemoteAddress());
+    }
+
+    @Test
+    public void testGetLogLevel() {
+        final BmpMockArguments arguments = BmpMockArguments.parseArguments(new String[] {"--log_level", "TRACE"});
+        assertEquals(Level.TRACE, arguments.getLogLevel());
+    }
+
+}