1 files changed, 53 insertions, 0 deletions

diff --git a/tests/integration/test_statefile.py b/tests/integration/test_statefile.py
new file mode 100644
index 0000000..2a2d147
--- /dev/null
+++ b/tests/integration/test_statefile.py
@@ -0,0 +1,53 @@
+"""The state dump and the -l switch that reads it back.
+
+Nothing exercised the load path before, and the dump's second column is now
+how long ago a torrent was last touched rather than when, because the clock
+counts from boot and an absolute stamp would mean nothing to the process
+that reads the file.
+"""
+
+import os
+import unittest
+
+from harness import Tracker, TrackerTestCase
+
+
+class StateFileRoundTrip(TrackerTestCase):
+    def dump(self):
+        return self.tracker.get("/stats?mode=statedump")
+
+    def test_a_dumped_torrent_comes_back_with_its_completed_count(self):
+        info_hash = self.hash_for("state-rt")
+        self.tracker.announce(info_hash, port=6001, left=0, event="completed")
+        self.assertEqual(self.tracker.scrape(info_hash)[b"files"][info_hash][b"downloaded"], 1)
+
+        path = os.path.join(self.tracker.dir, "state")
+        with open(path, "wb") as handle:
+            handle.write(self.dump())
+
+        restarted = Tracker(extra_args=["-l", path])
+        try:
+            files = restarted.scrape(info_hash)[b"files"]
+            self.assertIn(info_hash, files, "the torrent did not survive the restart")
+            entry = files[info_hash]
+            self.assertEqual(entry[b"complete"], 0, "no peers are carried over, only the torrent")
+            self.assertEqual(entry[b"incomplete"], 0)
+            complaints = restarted.stop()
+            self.assertEqual(complaints, [], "the tracker reported:\n" + "\n".join(complaints))
+        finally:
+            restarted._cleanup()
+
+    def test_a_torrent_older_than_the_tracker_would_keep_is_not_loaded(self):
+        """An age past the torrent timeout would be cleaned away on the first
+        sweep, so it is never built in the first place."""
+        info_hash = self.hash_for("state-stale")
+        path = os.path.join(self.tracker.dir, "stale")
+        with open(path, "w") as handle:
+            handle.write("%s:%d:7\n" % (info_hash.hex(), 24 * 60 + 1))
+
+        restarted = Tracker(extra_args=["-l", path])
+        try:
+            self.assertNotIn(info_hash, restarted.scrape(info_hash)[b"files"])
+            restarted.stop()
+        finally:
+            restarted._cleanup()