1 files changed, 20 insertions, 0 deletions
diff --git a/test/mst_test.exs b/test/mst_test.exs
index 73b4f63..7fe340e 100644
--- a/test/mst_test.exs
+++ b/test/mst_test.exs
@@ -107,5 +107,25 @@ defmodule ShardTest.MST do
     IO.puts "y.root: #{y.root|>Base.encode16}"
     IO.puts "z.root: #{z.root|>Base.encode16}"
     assert y.root == z.root
+
+    MST.last(y, nil, 10)
+  end
+
+  test "merkle search tree 5" do
+    y = Enum.reduce(0..1000, %MST{},
+                    fn i, acc -> MST.insert(acc, i) end)
+
+    assert(MST.last(y, nil, 2) == [{999, true}, {1000, true}])
+    assert(MST.last(y, 42, 2) == [{40, true}, {41, true}])
+
+    stuff = for i <- 100..199, do: {i, true}
+    assert MST.last(y, 200, 100) == stuff
+
+    stuff = for i <- 200..299, do: {i, true}
+    assert MST.last(y, 300, 100) == stuff
+
+    stuff = for i <- 200..499, do: {i, true}
+    assert MST.last(y, 500, 300) == stuff
   end
+
 end