From 5f01538987681bbf7d92e0b363a7ee02db04a86b Mon Sep 17 00:00:00 2001
From: Adam Johnson <me@adamj.eu>
Date: Mon, 28 Oct 2024 12:35:15 +0000
Subject: [PATCH] Add some proper tests

---
 src/django_watchfiles/__init__.py |  5 ---
 tests/test_django_watchfiles.py   | 68 +++++++++++++++++++++++++++++--
 2 files changed, 64 insertions(+), 9 deletions(-)

diff --git a/src/django_watchfiles/__init__.py b/src/django_watchfiles/__init__.py
index 7437f2c..a0d49cc 100644
--- a/src/django_watchfiles/__init__.py
+++ b/src/django_watchfiles/__init__.py
@@ -57,9 +57,7 @@ def __init__(self) -> None:
 
     def file_filter(self, change: watchfiles.Change, filename: str) -> bool:
         path = Path(filename)
-        # print(f"Path: {path} / {change}")
         if path in set(self.watched_files(include_globs=False)):
-            # print("Path in watched files")
             return True
         for directory, globs in self.directory_globs.items():
             try:
@@ -67,12 +65,9 @@ def file_filter(self, change: watchfiles.Change, filename: str) -> bool:
             except ValueError:
                 pass
             else:
-                # print("Path is sub dir")
                 for glob in globs:
                     if fnmatch.fnmatch(str(relative_path), glob):
-                        # print("Path is glob match")
                         return True
-        # print("file filter", change, path)
         return False
 
     def watched_roots(self, watched_files: list[Path]) -> frozenset[Path]:
diff --git a/tests/test_django_watchfiles.py b/tests/test_django_watchfiles.py
index 1c96d72..d521476 100644
--- a/tests/test_django_watchfiles.py
+++ b/tests/test_django_watchfiles.py
@@ -1,9 +1,69 @@
 from __future__ import annotations
 
+import tempfile
+from pathlib import Path
 
-def test_import():
-    # TODO: figure out tests
+from django.test import SimpleTestCase
+from django.utils import autoreload
 
-    import django_watchfiles
+from django_watchfiles import MutableWatcher
+from django_watchfiles import WatchfilesReloader
 
-    assert django_watchfiles  # type: ignore [truthy-bool]
+
+class MutableWatcherTests(SimpleTestCase):
+    def setUp(self):
+        self.watcher = MutableWatcher(lambda *args: True)
+        self.addCleanup(self.watcher.stop)
+
+        temp_dir = self.enterContext(tempfile.TemporaryDirectory())
+        self.temp_path = Path(temp_dir)
+
+    def test_set_roots_unchanged(self):
+        assert not self.watcher.change_event.is_set()
+        self.watcher.set_roots(set())
+        assert not self.watcher.change_event.is_set()
+
+    def test_set_roots_changed(self):
+        assert not self.watcher.change_event.is_set()
+        self.watcher.set_roots({Path("/tmp")})
+        assert self.watcher.change_event.is_set()
+
+    def test_stop(self):
+        assert not self.watcher.stop_event.is_set()
+        self.watcher.stop()
+        assert self.watcher.stop_event.is_set()
+
+    def test_iter_yields_changes(self):
+        (self.temp_path / "test.txt").touch()
+        self.watcher.set_roots({self.temp_path})
+        iterator = iter(self.watcher)
+        # flush initial events
+        next(iterator)
+
+        (self.temp_path / "test.txt").touch()
+        changes = next(iterator)
+
+        assert isinstance(changes, set)
+        assert len(changes) == 1
+        _, path = changes.pop()
+        assert path == str(self.temp_path.resolve() / "test.txt")
+
+    def test_iter_respects_change_event(self):
+        (self.temp_path / "test.txt").touch()
+        self.watcher.set_roots({self.temp_path})
+        iterator = iter(self.watcher)
+        # flush initial events
+        next(iterator)
+
+        self.watcher.set_roots(set())
+        self.watcher.set_roots({self.temp_path})
+        changes = next(iterator)
+
+        assert isinstance(changes, set)
+        assert len(changes) == 0
+
+
+class ReplacedGetReloaderTests(SimpleTestCase):
+    def test_replaced_get_reloader(self):
+        reloader = autoreload.get_reloader()
+        assert isinstance(reloader, WatchfilesReloader)