Add simple address retrieval to BCAddressService

maltemoeser · Mar 23, 2016 · af3ebbc · af3ebbc
1 parent a36be78
commit af3ebbc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/main/java/de/maltemoeser/bcgraph/entities/BCAddressService.java b/src/main/java/de/maltemoeser/bcgraph/entities/BCAddressService.java
@@ -21,4 +21,15 @@ public BCAddress getOrCreateAddress(String addressHash) {
         return myAddress;
     }
 
+    /**
+     * Returns the address with the given hash, or {@code null} if it does not exist.
+     */
+    public BCAddress getAddress(String addressHash) {
+        Node node = graphDatabaseService.findNode(LabelType.Address, NodeProperty.ADDRESS_HASH, addressHash);
+        if(node == null) {
+            return null;
+        } else {
+            return new BCAddress(node);
+        }
+    }
 }
diff --git a/src/test/java/de/maltemoeser/bcgraph/entities/BCServiceTest.java b/src/test/java/de/maltemoeser/bcgraph/entities/BCServiceTest.java
@@ -93,6 +93,9 @@ public void testBCAddressService() {
             BCAddress address2 = addressService.getOrCreateAddress(ADDRESSHASH);
             assertNotEquals(address0, address1);
             assertEquals(address0, address2);
+
+            assertNotNull(addressService.getAddress(ADDRESSHASH));
+            assertNull(addressService.getAddress(ADDRESSHASH + "-IS-NULL"));
         }
     }
 }