Add a test for partial user state reset (#106)

test/saxy/partial_test.exs

@@ -27,7 +27,7 @@ defmodule Saxy.PartialTest do
       "</foo>"
     ]
 
-    assert {:ok, events} = parse_partial(chunks)
+    assert {:ok, events} = parse_partial(chunks) |> Partial.terminate()
 
     events = Enum.reverse(events)
 
@@ -40,23 +40,40 @@ defmodule Saxy.PartialTest do
            ]
   end
 
+  test "fetch user state from a partial" do
+    chunks = ["<foo>", "sdf</foo>", ""]
+    partial = parse_partial(chunks)
+
+    assert Partial.get_state(partial) == [
+             {:end_element, "foo"},
+             {:characters, "sdf"},
+             {:start_element, {"foo", []}},
+             {:start_document, []}
+           ]
+  end
+
+  test "resets user state" do
+    chunks = ["<some>c", "hun", "k</some>"]
+    partial = parse_partial(chunks)
+    {:cont, partial} = Partial.parse(partial, "", [:made_up_thing])
+    {:ok, state} = Partial.terminate(partial)
+    assert state == [{:end_document, {}}, :made_up_thing]
+  end
+
   defp parse_partial(chunks) do
     assert {:ok, partial} = Partial.new(StackHandler, [])
 
-    partial =
-      Enum.reduce(chunks, partial, fn chunk, acc ->
-        assert {:cont, partial} = Partial.parse(acc, chunk)
-        partial
-      end)
-
-    Partial.terminate(partial)
+    Enum.reduce(chunks, partial, fn chunk, acc ->
+      assert {:cont, partial} = Partial.parse(acc, chunk)
+      partial
+    end)
   end
 
   test "parses XML document partially line by line" do
     for fixture <- @fixtures do
       data_chunks = stream_fixture(fixture)
 
-      assert {:ok, _state} = parse_partial(data_chunks)
+      assert {:ok, _state} = parse_partial(data_chunks) |> Partial.terminate()
     end
   end