Add unit test for Sha256Identifier and ObjectCollection

2 files changed, 84 insertions, 1 deletions

diff --git a/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/ObjectCollectionFactoryTest.java b/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/ObjectCollectionFactoryTest.java
new file mode 100644
index 0000000..60a42cf
--- /dev/null
+++ b/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/ObjectCollectionFactoryTest.java
@@ -0,0 +1,81 @@
+package com.keuin.kbackupfabric.util.backup.incremental;
+
+import com.keuin.kbackupfabric.util.backup.incremental.identifier.Sha256Identifier;
+import org.junit.Test;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.*;
+
+import static org.junit.Assert.*;
+
+public class ObjectCollectionFactoryTest {
+
+    private void validate(ObjectCollection collection, List<String> subCollections, Map<String, String> subElements) {
+        assertEquals(subCollections.size(), collection.getSubCollectionMap().size());
+        assertEquals(subElements.size(), collection.getElementSet().size());
+        for (Map.Entry<String, ObjectCollection> c : collection.getSubCollectionMap().entrySet()) {
+            assertEquals(c.getKey(), c.getValue().getName());
+            assertTrue(subCollections.contains(c.getKey()));
+        }
+        for (Map.Entry<String, ObjectElement> entry : collection.getElementMap().entrySet()) {
+//            assertTrue(subElements.contains(e.getIdentification()));
+            assertEquals(subElements.get(entry.getKey()), entry.getValue().getIdentifier().getIdentification());
+        }
+    }
+
+    @Test
+    public void fromDirectory() {
+        try {
+            ObjectCollectionFactory<Sha256Identifier> factory =
+                    new ObjectCollectionFactory<>(Sha256Identifier.getFactory());
+            ObjectCollection collection =
+                    factory.fromDirectory(new File("./testfile/ObjectCollectionFactoryTest"));
+
+            assertEquals("ObjectCollectionFactoryTest", collection.getName());
+            assertEquals(3, collection.getSubCollectionMap().size());
+            assertEquals(2, collection.getElementSet().size());
+
+            final Map<String, String> elements = new HashMap<>();
+
+            // check root dir
+            elements.put("a", "261CA0D59FEE8FD169802BB8030A07CF23E5C1593FA81A16C6D0A8CF27DAA2ED");
+            elements.put("b", "B3FED75012C4969DC63A50EBC4E745FF77E4A06E0B04720EF71EF033032EBAF7");
+            validate(collection, Arrays.asList("1", "2", "3"), elements);
+            elements.clear();
+
+            // check `1`
+            elements.put("a", "E8620F35A5DB33B1257CC51245DDACDA8AF3E0D431A8A38473575E468BCBD0BD");
+            elements.put("b", "19EE41585A674274891DE5A4B365DBAB9C49C576AB6F86CD515B683724D2DBBD");
+            validate(collection.getSubCollectionMap().get("1"), Arrays.asList("11", "12"), elements);
+            elements.clear();
+
+            // check `2`
+            validate(collection.getSubCollectionMap().get("2"), Collections.emptyList(), Collections.emptyMap());
+
+            // check `3`
+            validate(collection.getSubCollectionMap().get("3"), Collections.emptyList(), Collections.emptyMap());
+
+            // check `11`
+            validate(collection.getSubCollectionMap().get("1").getSubCollectionMap().get("11"), Collections.singletonList("111"), Collections.emptyMap());
+
+            // check `111`
+            elements.put("a", "1EDBE882A757E1FAFCA77A9D3BE3FF5D2BB3E2037B238C865F1F957C431F43B4");
+            elements.put("b", "30BA7CD8B4AD93A8B3826CD8D1518790924EEBB930EC04DF7DFB03A50B17D7BC");
+            validate(
+                    collection.getSubCollectionMap().get("1").getSubCollectionMap().get("11").getSubCollectionMap().get("111"),
+                    Collections.emptyList(),
+                    elements
+            );
+            elements.clear();
+
+            // check `12`
+            validate(collection.getSubCollectionMap().get("1").getSubCollectionMap().get("12"), Collections.emptyList(), Collections.emptyMap());
+
+
+        } catch (IOException e) {
+            e.printStackTrace();
+            fail();
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/identifier/Sha256IdentifierTest.java b/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/identifier/Sha256IdentifierTest.java
index 2296e01..53e92e2 100644
--- a/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/identifier/Sha256IdentifierTest.java
+++ b/src/test/java/com/keuin/kbackupfabric/util/backup/incremental/identifier/Sha256IdentifierTest.java
@@ -6,17 +6,19 @@ import java.io.File;
 import java.io.IOException;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
 
 public class Sha256IdentifierTest {
 
     @Test
     public void fromFile() {
         try {
-            Sha256Identifier sha256 = Sha256Identifier.fromFile(new File("./src/test/sha256"));
+            Sha256Identifier sha256 = Sha256Identifier.fromFile(new File("./testfile/Sha256IdentifierTest"));
             String str = sha256.getIdentification().toUpperCase();
             assertEquals("315F5BDB76D078C43B8AC0064E4A0164612B1FCE77C869345BFC94C75894EDD3", str);
         } catch (IOException e) {
             e.printStackTrace();
+            fail();
         }
     }