MappingServiceManager: pass VPN info to intents

[vpnservice.git] / vpnintent / impl / src / test / java / org / opendaylight / vpnservice / impl / MappingServiceManagerTests.java

diff --git a/vpnintent/impl/src/test/java/org/opendaylight/vpnservice/impl/MappingServiceManagerTests.java b/vpnintent/impl/src/test/java/org/opendaylight/vpnservice/impl/MappingServiceManagerTests.java
new file mode 100644 (file)
index 0000000..2fa5cb3
--- /dev/null
+++ b/vpnintent/impl/src/test/java/org/opendaylight/vpnservice/impl/MappingServiceManagerTests.java
@@ -0,0 +1,81 @@
+/*
+ * Copyright (c) 2016 Inocybe Technologies 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.vpnservice.impl;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import org.junit.Test;
+import org.opendaylight.nic.mapping.api.IntentMappingService;
+
+public class MappingServiceManagerTests {
+
+    @Test
+    public void addInfo() {
+        // Arrange
+        String siteName = "UoR";
+        String ipPrefix = "16.101.233.2/8";
+        String switchPortId = "openflow:1:3";
+        long mplsLabel = 10L;
+        String nextHop = "16.101.233.1/8";
+
+        Map<String, String> map = new HashMap<>();
+        map.put("ip_prefix", ipPrefix);
+        map.put("switch_port", switchPortId);
+        map.put("mpls_label", String.valueOf(mplsLabel));
+        map.put("next_hop", nextHop);
+
+        IntentMappingService mapSvc = mock(IntentMappingService.class);
+        when(mapSvc.get(any(String.class))).thenReturn(map);
+
+        MappingServiceManager manager = new MappingServiceManager(mapSvc);
+
+        // Act
+        manager.add(siteName, ipPrefix, switchPortId, mplsLabel, nextHop);
+
+        Map<String, String> returnedObjs = manager.get(siteName);
+
+        // Assert
+        assertEquals(ipPrefix, returnedObjs.get("ip_prefix"));
+        assertEquals(switchPortId, returnedObjs.get("switch_port"));
+        assertEquals(mplsLabel, Long.parseLong(returnedObjs.get("mpls_label")));
+        assertEquals(nextHop, returnedObjs.get("next_hop"));
+    }
+
+    @Test
+    public void removeInfo() {
+        // Arrange
+        String siteName = "UoR";
+        String ipPrefix = "16.101.233.2/8";
+        String switchPortId = "openflow:1:3";
+
+        Map<String, String> map = new HashMap<>();
+        map.put("ip_prefix", ipPrefix);
+        map.put("switch_port", switchPortId);
+
+        IntentMappingService mapSvc = mock(IntentMappingService.class);
+
+        MappingServiceManager manager = new MappingServiceManager(mapSvc);
+
+        // Add first to delete next
+        manager.add(siteName, ipPrefix, switchPortId, null, null);
+
+        // Act
+        boolean result = manager.delete(siteName);
+
+        // Assert
+        assertTrue(result);
+    }
+}