From 3a33b2b48671e3db17a481bc23d27d2d1386eb38 Mon Sep 17 00:00:00 2001
From: Houkime <>
Date: Mon, 23 Sep 2024 10:42:18 +0000
Subject: [PATCH] test(services): utilities for checking and alterning
 testfiles

---
 tests/conftest.py | 34 ++++++++++++++++++++++------------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index b20df92..2cea1f4 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -177,17 +177,25 @@ def get_testfile_bodies(service: DummyService):
     testfiles: List[str] = []
     for folder in service.get_folders():
         files = listdir(folder)
+        files = [path.join(folder, file) for file in files]
         testfiles.extend(files)
-    bodies = []
+    bodies = {}
     for f in testfiles:
-        pass
-
-    with open("/usr/tsr.txt") as file:
-        content = file.read()
-        bodies.append(content)
+        with open(f, "r") as file:
+            bodies[f] = file.read()
     return bodies
 
 
+def is_original_files(service: DummyService):
+    # For use in restoration tests mostly
+
+    paths = testfile_paths(service.get_folders())
+    return get_testfile_bodies(service) == {
+        paths[0]: TESTFILE_BODY,
+        paths[1]: TESTFILE2_BODY,
+    }
+
+
 @pytest.fixture()
 def raw_dummy_service(tmpdir) -> DummyService:
     dirnames = ["test_service", "also_test_service"]
@@ -197,20 +205,22 @@ def raw_dummy_service(tmpdir) -> DummyService:
         makedirs(service_dir)
         service_dirs.append(service_dir)
 
+    paths = testfile_paths(service_dirs)
     bodies = [TESTFILE_BODY, TESTFILE2_BODY]
 
-    for fullpath, body in zip(testfile_paths(service_dirs), bodies):
+    # Just touching first, filling is separate
+    for fullpath in paths:
         with open(fullpath, "w") as file:
-            file.write(TESTFILE2_BODY)
+            file.write("")
 
-    paths = testfile_paths
-
-    # we need this to not change get_folders() much
     class TestDummyService(DummyService, folders=service_dirs):
         pass
 
     service = TestDummyService()
-    # assert pickle.dumps(service) is not None
+    write_testfile_bodies(service, bodies)
+
+    assert is_original_files(service)
+
     return service