diff --git a/src/main/java/com/hashbangbash/trie/App.java b/src/main/java/com/hashbangbash/trie/App.java index 66a3c8f..74d681e 100644 --- a/src/main/java/com/hashbangbash/trie/App.java +++ b/src/main/java/com/hashbangbash/trie/App.java @@ -124,6 +124,7 @@ public class App { } catch (IOException ex) { } return null; } + public static byte[] getBytesFromFile(File file) throws IOException { InputStream is = new FileInputStream(file); diff --git a/src/test/java/com/redhat/trie/TestPathTree.java b/src/test/java/com/redhat/trie/TestPathTree.java index f92307f..f261dac 100644 --- a/src/test/java/com/redhat/trie/TestPathTree.java +++ b/src/test/java/com/redhat/trie/TestPathTree.java @@ -110,8 +110,10 @@ public class TestPathTree { PathTree pt1; List contents0; List contents1; + long len0; bytes = TestHelpers.loadBytes(this, "test.bin"); + len0 = bytes.length; pt0 = new PathTree(bytes); contents0 = pt0.toList(); for (String str : contents0) { @@ -126,6 +128,7 @@ public class TestPathTree { assertNotNull(pt1); //printByteArray(pt1.getPayload()); contents1 = pt1.toList(); + assertEquals(len0, pt1.getPayload().length); assertTrue(TestHelpers.cmpStrings(contents0, contents1)); for (String str : contents1) { System.out.println(str);