From a087347d0f58aaf0fc7baab7a734c7a53a099d86 Mon Sep 17 00:00:00 2001
From: David Brochart <david.brochart@gmail.com>
Date: Fri, 8 Apr 2022 11:37:32 +0200
Subject: [PATCH] Test appending to file

---
 tests/test_watch.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/tests/test_watch.py b/tests/test_watch.py
index 7b409b04..1fcc185b 100644
--- a/tests/test_watch.py
+++ b/tests/test_watch.py
@@ -1,3 +1,4 @@
+import asyncio
 import sys
 import threading
 from contextlib import contextmanager
@@ -115,3 +116,34 @@ async def test_awatch_no_yield(mock_rust_notify: 'MockRustType', caplog):
     assert changes == {(Change.added, 'spam.py')}
     assert mock.watch_count == 2
     assert caplog.text == "watchfiles.main DEBUG: 1 change detected: {(<Change.added: 1>, 'spam.py')}\n"
+
+
+async def test_watch_directory(tmp_path: Path):
+    stop_event = asyncio.Event()
+
+    async def stop_soon():
+        await asyncio.sleep(0.4)
+        stop_event.set()
+
+    async def change_dir():
+        await asyncio.sleep(0.1)
+        (tmp_path / 'file0').write_text('foo')
+        await asyncio.sleep(0.1)
+        with open(tmp_path / 'file0', 'a') as f:
+            f.write(' bar')
+        await asyncio.sleep(0.1)
+        (tmp_path / 'file0').unlink()
+
+    tasks = [asyncio.create_task(stop_soon()), asyncio.create_task(change_dir())]
+
+    changes = []
+    async for change in awatch(tmp_path, stop_event=stop_event, step=1):
+        changes.append(change)
+
+    assert changes == [
+        {(Change.added, str(tmp_path / 'file0'))},
+        {(Change.modified, str(tmp_path / 'file0'))},
+        {(Change.deleted, str(tmp_path / 'file0'))},
+    ]
+
+    await asyncio.gather(*tasks)