Skip to content

Commit ed9bed3

Browse files
466725466725
committed
Refactor
1 parent bba4ef2 commit ed9bed3

4 files changed

Lines changed: 55 additions & 80 deletions

File tree

‎pytest_demo/ai_generation/demo.py‎

Lines changed: 49 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -12,32 +12,32 @@ def demo_generate_tangerine_homepage_test():
1212
url="https://www.tangerine.ca/en/personal",
1313
title="Tangerine",
1414
dom="""<html>
15-
<head><title>Tangerine</title></head>
16-
<body>
17-
<header>
18-
<nav>
19-
<a id="login" href="/app#/login">Log in</a>
20-
<a id="menu_signup" href="/app#/signup">Sign me up</a>
21-
</nav>
22-
</header>
23-
<main>
24-
<h1>Welcome to Tangerine</h1>
25-
<p>Your trusted online banking partner</p>
26-
</main>
27-
</body>
28-
</html>""",
29-
element_tree="""<body>
30-
<header>
31-
<nav>
32-
<a id="login" href="/app#/login">Log in</a>
33-
<a id="menu_signup" href="/app#/signup">Sign me up</a>
34-
</nav>
35-
</header>
36-
<main>
37-
<h1>Welcome to Tangerine</h1>
38-
<p>Your trusted online banking partner</p>
39-
</main>
40-
</body>""",
15+
<head><title>Tangerine</title></head>
16+
<body>
17+
<header>
18+
<nav>
19+
<a id="login" href="/app#/login">Log in</a>
20+
<a id="menu_signup" href="/app#/signup">Sign me up</a>
21+
</nav>
22+
</header>
23+
<main>
24+
<h1>Welcome to Tangerine</h1>
25+
<p>Your trusted online banking partner</p>
26+
</main>
27+
</body>
28+
</html>""",
29+
element_tree="""<body>
30+
<header>
31+
<nav>
32+
<a id="login" href="/app#/login">Log in</a>
33+
<a id="menu_signup" href="/app#/signup">Sign me up</a>
34+
</nav>
35+
</header>
36+
<main>
37+
<h1>Welcome to Tangerine</h1>
38+
<p>Your trusted online banking partner</p>
39+
</main>
40+
</body>""",
4141
screenshot_base64="iVBORw0KGgoAAAANSUhEUgAAAAUA...",
4242
network_events=[
4343
{"method": "GET", "url": "https://www.tangerine.ca/en/personal", "status": "200"},
@@ -48,29 +48,29 @@ def demo_generate_tangerine_homepage_test():
4848
class MockClient:
4949
def generate(self, *, system_prompt: str, user_prompt: str) -> str:
5050
return """```python
51-
from playwright.sync_api import Page
52-
import pytest
53-
54-
@pytest.mark.ui
55-
def test_tangerine_homepage_generated(page: Page):
56-
page.goto("https://www.tangerine.ca/en/personal", wait_until="domcontentloaded")
57-
58-
# Verify page loads with correct title
59-
assert page.title() == "Tangerine"
60-
61-
# Verify Sign In button is visible
62-
login_button = page.locator("#login")
63-
assert login_button.is_visible()
64-
65-
# Verify Sign Up button is visible
66-
signup_button = page.locator("#menu_signup")
67-
assert signup_button.is_visible()
68-
69-
# Verify main heading is present
70-
heading = page.locator("h1")
71-
assert heading.is_visible()
72-
assert "Welcome to Tangerine" in heading.text_content()
73-
```"""
51+
from playwright.sync_api import Page
52+
import pytest
53+
54+
@pytest.mark.ui
55+
def test_tangerine_homepage_generated(page: Page):
56+
page.goto("https://www.tangerine.ca/en/personal", wait_until="domcontentloaded")
57+
58+
# Verify page loads with correct title
59+
assert page.title() == "Tangerine"
60+
61+
# Verify Sign In button is visible
62+
login_button = page.locator("#login")
63+
assert login_button.is_visible()
64+
65+
# Verify Sign Up button is visible
66+
signup_button = page.locator("#menu_signup")
67+
assert signup_button.is_visible()
68+
69+
# Verify main heading is present
70+
heading = page.locator("h1")
71+
assert heading.is_visible()
72+
assert "Welcome to Tangerine" in heading.text_content()
73+
```"""
7474

7575
client: ScriptClient = MockClient()
7676
generator = PlaywrightTestScriptGenerator(client)

‎pytest_demo/tests/AI/generated_playwright/test_tangerine_get_our_app_links.py‎

Lines changed: 0 additions & 23 deletions
This file was deleted.

‎pytest_demo/tests/AI/generated_playwright/test_tangerine_homepage_generated.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import pytest
33

44
@pytest.mark.ai
5-
@pytest.mark.ui
65
def test_tangerine_homepage_generated(page: Page):
76
page.goto("https://www.tangerine.ca/en/personal", wait_until="domcontentloaded")
87

‎pytest_demo/tests/AI/test_ai_generation.py‎

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -39,12 +39,12 @@ def test_generator_writes_code_from_fenced_response(tmp_path: Path):
3939
snapshot = _snapshot(dom="<html></html>", element_tree="<body></body>")
4040
client = _script_client(
4141
"""```python
42-
from playwright.sync_api import Page
43-
44-
def test_smoke(page: Page):
45-
page.goto(\"https://www.tangerine.ca/en\")
46-
assert \"Tangerine\" in page.title()
47-
```"""
42+
from playwright.sync_api import Page
43+
44+
def test_smoke(page: Page):
45+
page.goto(\"https://www.tangerine.ca/en\")
46+
assert \"Tangerine\" in page.title()
47+
```"""
4848
)
4949
generator = PlaywrightTestScriptGenerator(client)
5050

@@ -59,4 +59,3 @@ def test_smoke(page: Page):
5959
content = result.output_path.read_text(encoding="utf-8")
6060
assert "def test_generated_ui_flow" in content
6161
assert "page.goto" in content
62-

0 commit comments

Comments
 (0)