Add test for document dirty attribute (jupyterlab#251)

* Add test for document dirty attribute

* Add rtc_document_save_delay fixture

Author: davidbrochart

pyproject.toml

@@ -49,7 +49,8 @@ test = [
     "jupyter_server_fileid[test]",
     "pytest>=7.0",
     "pytest-cov",
-    "websockets"
+    "websockets",
+    "importlib_metadata >=3.6; python_version<'3.10'",
 ]
 docs = [
     "jupyterlab>=4.0.0",

tests/conftest.py

@@ -24,7 +24,12 @@
 
 
 @pytest.fixture
-def jp_server_config(jp_root_dir, jp_server_config):
+def rtc_document_save_delay():
+    return 1
+
+
+@pytest.fixture
+def jp_server_config(jp_root_dir, jp_server_config, rtc_document_save_delay):
     return {
         "ServerApp": {
             "jpserver_extensions": {"jupyter_collaboration": True, "jupyter_server_fileid": True},
@@ -38,6 +43,7 @@ def jp_server_config(jp_root_dir, jp_server_config):
             "db_path": str(jp_root_dir.joinpath(".fid_test.db")),
             "db_journal_mode": "OFF",
         },
+        "YDocExtension": {"document_save_delay": rtc_document_save_delay},
     }
 
 

tests/test_documents.py

@@ -0,0 +1,39 @@
+# Copyright (c) Jupyter Development Team.
+# Distributed under the terms of the Modified BSD License.
+
+import sys
+
+if sys.version_info < (3, 10):
+    from importlib_metadata import entry_points
+else:
+    from importlib.metadata import entry_points
+
+import pytest
+from anyio import sleep
+from pycrdt_websocket import WebsocketProvider
+
+jupyter_ydocs = {ep.name: ep.load() for ep in entry_points(group="jupyter_ydoc")}
+
+
+@pytest.fixture
+def rtc_document_save_delay():
+    return 0.5
+
+
+async def test_dirty(
+    rtc_create_file,
+    rtc_connect_doc_client,
+    rtc_document_save_delay,
+):
+    file_format = "text"
+    file_type = "file"
+    file_path = "dummy.txt"
+    await rtc_create_file(file_path)
+    jupyter_ydoc = jupyter_ydocs[file_type]()
+
+    async with await rtc_connect_doc_client(file_format, file_type, file_path) as ws:
+        async with WebsocketProvider(jupyter_ydoc.ydoc, ws):
+            for _ in range(2):
+                jupyter_ydoc.dirty = True
+                await sleep(rtc_document_save_delay * 1.5)
+                assert not jupyter_ydoc.dirty