Add findTopology method to the API

khalidzahra · Sep 29, 2021 · 9a5a445 · 9a5a445
1 parent 9bf33ef
commit 9a5a445
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/main/java/com/github/khalidzahra/TopologyAPI.java b/src/main/java/com/github/khalidzahra/TopologyAPI.java
@@ -58,6 +58,19 @@ public List<Topology> queryTopologies() {
         return topologyRegistry.getTopologyList();
     }
 
+    /**
+     * Finds topology with specified topology ID
+     * @param topologyID String variable containing the topology ID
+     * @return Returns Topology object with specified topology ID
+     */
+    public Topology findTopology(String topologyID) {
+        return topologyRegistry.getTopologyList()
+                .stream()
+                .filter(topology -> topology.getId().equals(topologyID))
+                .findAny()
+                .orElse(null);
+    }
+
     /**
      * Deletes the specified topology from memory
      *

diff --git a/src/test/java/com/github/khalidzahra/TopologyAPITest.java b/src/test/java/com/github/khalidzahra/TopologyAPITest.java
@@ -100,4 +100,20 @@ public void testQueryDevicesWithNetlistNode() {
         assertEquals(topologyAPI.queryDevicesWithNetlistNode("top2", "vss").get(0).getDeviceType(), DeviceType.PMOS);
 
     }
+
+    public void testFindTopology() {
+        TopologyAPI topologyAPI = new TopologyAPI();
+        topologyAPI.readJSON(resourceDirectory.toAbsolutePath() + "/topology1.json");
+        assertNotNull(topologyAPI.findTopology("top1"));
+        assertEquals(topologyAPI.findTopology("top1").getId(), "top1");
+        topologyAPI.readJSON(resourceDirectory.toAbsolutePath() + "/topology2.json");
+        assertNotNull(topologyAPI.findTopology("top2"));
+        assertEquals(topologyAPI.findTopology("top2").getId(), "top2");
+        assertNull(topologyAPI.findTopology("top3"));
+        assertNull(topologyAPI.findTopology("top4"));
+        topologyAPI.deleteTopology("top1");
+        assertNull(topologyAPI.findTopology("top1"));
+        topologyAPI.deleteTopology("top2");
+        assertNull(topologyAPI.findTopology("top2"));
+    }
 }