fix cassette hashing

docs/tests/unit_tests/test_cassette_hash.py

@@ -0,0 +1,117 @@
+import re
+from textwrap import dedent
+
+import pytest
+from _scripts.hook_state import hook_state
+from _scripts.notebook_hooks import handle_vcr_setup
+from _scripts.setup_vcr import extract_code_blocks_for_session, get_hash_for_session
+
+INITIAL_DOCUMENT_CONTENT = dedent(
+    """
+    Blah blah blah
+
+    ```python exec="on" source="above" session="1" result="ansi"
+    print("FIRST_CODE_BLOCK")
+    ```
+
+    Blah blah blah!
+
+    ```python exec="on" source="above" session="1" result="ansi"
+    print("SECOND_CODE_BLOCK")
+    ```
+    
+    more blah blah blah
+    """
+)
+
+
+@pytest.mark.parametrize(
+    "replace_string",
+    [
+        "FIRST_CODE_BLOCK",
+        "SECOND_CODE_BLOCK",
+    ],
+)
+def test_changing_block_in_session_invalidates_hash(replace_string: str):
+    hook_state['document_filename'] = 'test.md'
+    hook_state['document_content'] = INITIAL_DOCUMENT_CONTENT
+
+    code = "print('Hello, world!')"
+
+    result1 = handle_vcr_setup(
+        formatter=lambda **kwargs: None,
+        language="python",
+        session="1",
+        id="test",
+        code=code,
+        md=None,
+        extra={},
+    )
+
+    cassette_init_expr = re.compile(r"^_cassette = HashedCassette\('[^']+', '(?P<hash>[^']+)'\)$")
+    assert result1['transform_source']
+    execute_source, display_source = result1['transform_source'](code)
+    assert display_source == code
+    cassette_init_line = [line for line in execute_source.splitlines() if line.startswith("_cassette = HashedCassette(")][0]
+    assert cassette_init_line
+    match = cassette_init_expr.match(cassette_init_line)
+    assert match
+    hash_ = str(match.group('hash'))
+
+    # change the content of the second block of code
+    hook_state['document_content'] = INITIAL_DOCUMENT_CONTENT.replace(replace_string, "world")
+    assert hook_state['document_content'] != INITIAL_DOCUMENT_CONTENT
+
+    result2 = handle_vcr_setup(
+        formatter=lambda **kwargs: None,
+        language="python",
+        session="1",
+        id="test",
+        code=code,
+        md=None,
+        extra={},
+    )
+
+    assert result2['transform_source']
+    execute_source, display_source = result2['transform_source'](code)
+    assert display_source == code
+    cassette_init_line = [line for line in execute_source.splitlines() if line.startswith("_cassette = HashedCassette(")][0]
+    assert cassette_init_line
+    match = cassette_init_expr.match(cassette_init_line)
+    assert match
+
+    # this is the important part
+    assert str(match.group('hash')) != hash_
+
+@pytest.mark.parametrize(
+    "replace_string",
+    [
+        "FIRST_CODE_BLOCK",
+        "SECOND_CODE_BLOCK",
+    ],
+)
+def test_get_hash_for_session(replace_string: str):
+    hash_ = get_hash_for_session(
+        language="python",
+        session="1",
+        content=INITIAL_DOCUMENT_CONTENT,
+    )
+    
+    content = INITIAL_DOCUMENT_CONTENT.replace(replace_string, "world")
+    assert content != INITIAL_DOCUMENT_CONTENT
+    assert get_hash_for_session(
+        language="python",
+        session="1",
+        content=INITIAL_DOCUMENT_CONTENT.replace(replace_string, "world"),
+    ) != hash_
+
+def test_get_code_blocks_for_session():
+    code_blocks = extract_code_blocks_for_session(
+        language="python",
+        session="1",
+        content=INITIAL_DOCUMENT_CONTENT,
+    )
+    assert code_blocks
+    assert len(code_blocks) == 2
+    assert code_blocks[0] == 'print("FIRST_CODE_BLOCK")'
+    assert code_blocks[1] == 'print("SECOND_CODE_BLOCK")'

docs/tests/unit_tests/test_highlights.py

@@ -111,7 +111,7 @@ print("Hello, World!")
 
 
 END_TO_END_INPUT_HIGHLIGHT_1_EXPECT = """\
-```python exec="on" source="below" hl_lines="2" path="dummy.md"
+```python exec="on" source="below" hl_lines="2"
 print("Hello, World!")
 print("Hello, World!")
 ```