test: more tests

Author: 15r10nk

src/inline_snapshot/testing/_example.py

@@ -83,7 +83,9 @@ def parse_outcomes(lines):
 
 
 class Example:
-    def __init__(self, files: str | dict[str, str]):
+    files: dict[str, str | bytes]
+
+    def __init__(self, files: str | dict[str, str | bytes]):
         """
         Parameters:
             files: a collection of files where inline-snapshot operates on,
@@ -131,12 +133,20 @@ def normalize_path(path):
             and ".pytest_cache" not in p.parts
         }
 
-    def with_files(self, extra_files):
-        return Example(self.files | extra_files)
+    def with_files(self, extra_files: dict[str, str | bytes]) -> Example:
+        return Example({**self.files, **extra_files})
+
+    def read_text(self, name: str) -> str:
+        text = self.files[name]
+        assert isinstance(text, str)
+        return text
 
-    def change_code(self, func):
+    def change_code(self, func) -> Example:
         return Example({name: func(text) for name, text in self.files.items()})
 
+    def replace(self, text, new_text) -> Example:
+        return self.change_code(lambda code: code.replace(text, new_text))
+
     def run_inline(
         self,
         args: list[str] = [],

tests/test_docs.py

@@ -381,7 +381,7 @@ def test_block(block: Block):
 
             new_code = code
             if flags:
-                new_code = example.files["test_example.py"]
+                new_code = example.read_text("test_example.py")
             new_code.replace("\n\n", "\n")
 
             if "show_error" in options:

tests/test_external_formats.py

@@ -29,6 +29,8 @@ def test_a():
 """,
             }
         ),
+    ).run_inline(
+        ["--inline-snapshot=disable"]
     )
 
 

tests/test_storage.py

@@ -7,22 +7,44 @@ def test_uuid_storage():
         """
 from inline_snapshot import external
 
+s=external("uuid:")
+
 def test_a():
     assert "value" == external("uuid:")
+    assert "blub"==s
     """
     ).run_inline(
         ["--inline-snapshot=create"],
         changed_files=snapshot(
             {
+                "__inline_snapshot__/test_something/__module__/f728b4fa-4248-4e3a-8a5d-2f346baa9455.txt": "blub",
                 "__inline_snapshot__/test_something/test_a/e3e70682-c209-4cac-a29f-6fbed82c07cd.txt": "value",
                 "test_something.py": """\
 
 from inline_snapshot import external
 
+s=external("uuid:f728b4fa-4248-4e3a-8a5d-2f346baa9455.txt")
+
 def test_a():
     assert "value" == external("uuid:e3e70682-c209-4cac-a29f-6fbed82c07cd.txt")
+    assert "blub"==s
     \
 """,
             }
         ),
+    ).replace(
+        '"value"', '"new_value"'
+    ).run_inline(
+        ["--inline-snapshot=create"],
+        changed_files=snapshot({}),
+        raises=snapshot("AssertionError:\n"),
+    ).run_inline(
+        ["--inline-snapshot=fix"],
+        changed_files=snapshot(
+            {
+                "__inline_snapshot__/test_something/test_a/e3e70682-c209-4cac-a29f-6fbed82c07cd.txt": "new_value"
+            }
+        ),
+    ).run_inline(
+        ["--inline-snapshot=disable"], changed_files=snapshot({})
     )