predicate-runtime-python/tests/test_generator.py at 954e18d4df33585b5367cbed3ec0b857c9c7db8d · PredicateSystems/predicate-runtime-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
"""
Tests for script generator functionality
"""

import pytest
import tempfile
import os
from sentience import SentienceBrowser, record
from sentience.recorder import Trace, TraceStep
from sentience.generator import ScriptGenerator, generate


def test_generator_python():
    """Test Python script generation"""
    with SentienceBrowser() as browser:
        browser.page.goto("https://example.com")
        browser.page.wait_for_load_state("networkidle")

        with record(browser) as rec:
            rec.record_navigation("https://example.com")
            rec.record_click(1, "role=button text~'Click'")
            rec.record_type(2, "hello", "role=textbox")
            rec.record_press("Enter")

        generator = ScriptGenerator(rec.trace)
        code = generator.generate_python()

        # Verify code contains expected elements
        assert "from sentience import" in code
        assert "def main():" in code
        assert "SentienceBrowser" in code
        assert "role=button text~'Click'" in code
        assert "click(browser" in code
        assert "type_text(browser" in code
        assert "press(browser" in code


def test_generator_typescript():
    """Test TypeScript script generation"""
    with SentienceBrowser() as browser:
        browser.page.goto("https://example.com")
        browser.page.wait_for_load_state("networkidle")

        with record(browser) as rec:
            rec.record_navigation("https://example.com")
            rec.record_click(1, "role=button")

        generator = ScriptGenerator(rec.trace)
        code = generator.generate_typescript()

        # Verify code contains expected elements
        assert "import" in code
        assert "async function main()" in code
        assert "SentienceBrowser" in code
        assert "await click" in code


def test_generator_save_python():
    """Test saving generated Python script"""
    with SentienceBrowser() as browser:
        browser.page.goto("https://example.com")
        browser.page.wait_for_load_state("networkidle")

        with record(browser) as rec:
            rec.record_click(1)

        generator = ScriptGenerator(rec.trace)

        with tempfile.NamedTemporaryFile(mode='w', suffix='.py', delete=False) as f:
            temp_path = f.name

        try:
            generator.save_python(temp_path)
            assert os.path.exists(temp_path)

            with open(temp_path) as f:
                code = f.read()
                assert "from sentience import" in code
        finally:
            os.unlink(temp_path)


def test_generator_save_typescript():
    """Test saving generated TypeScript script"""
    with SentienceBrowser() as browser:
        browser.page.goto("https://example.com")
        browser.page.wait_for_load_state("networkidle")

        with record(browser) as rec:
            rec.record_click(1)

        generator = ScriptGenerator(rec.trace)

        with tempfile.NamedTemporaryFile(mode='w', suffix='.ts', delete=False) as f:
            temp_path = f.name

        try:
            generator.save_typescript(temp_path)
            assert os.path.exists(temp_path)

            with open(temp_path) as f:
                code = f.read()
                assert "import" in code
        finally:
            os.unlink(temp_path)


def test_generator_without_selector():
    """Test generator handles steps without selectors"""
    with SentienceBrowser() as browser:
        browser.page.goto("https://example.com")
        browser.page.wait_for_load_state("networkidle")

        # Create a trace manually with a step that has no selector
        # (The recorder automatically infers selectors, so we create the step directly)
        trace = Trace("https://example.com")
        step = TraceStep(
            ts=0,
            type="click",
            element_id=1,
            selector=None  # Explicitly no selector
        )
        trace.add_step(step)

        generator = ScriptGenerator(trace)
        code = generator.generate_python()

        # Should include TODO comment for missing selector
        assert "TODO: replace with semantic selector" in code
        assert "click(browser, 1)" in code


def test_generate_helper():
    """Test generate() helper function"""
    with SentienceBrowser() as browser:
        browser.page.goto("https://example.com")
        browser.page.wait_for_load_state("networkidle")

        with record(browser) as rec:
            rec.record_click(1)

        # Test Python generation
        py_code = generate(rec.trace, 'py')
        assert "from sentience import" in py_code

        # Test TypeScript generation
        ts_code = generate(rec.trace, 'ts')
        assert "import" in ts_code