wip

Author: ks6088ts

.gitignore

@@ -167,3 +167,5 @@ tests/test_codegen.py
 .streamlit/*
 !.streamlit/config.yaml
 !.streamlit/secrets.toml.example
+playwright/.auth/*
+!playwright/.auth/.gitkeep

Makefile

@@ -2,6 +2,9 @@
 GIT_REVISION ?= $(shell git rev-parse --short HEAD)
 GIT_TAG ?= $(shell git describe --tags --abbrev=0 --always | sed -e s/v//g)
 
+# Project
+SKIP_TEST ?= true
+
 .PHONY: help
 help:
 	@grep -E '^[a-zA-Z_-]+:.*?## .*$$' $(MAKEFILE_LIST) | sort | awk 'BEGIN {FS = ":.*?## "}; {printf "\033[36m%-30s\033[0m %s\n", $$1, $$2}'
@@ -40,7 +43,7 @@ lint: ## lint
 
 .PHONY: test
 test: ## run tests
-	uv run pytest --capture=no -vv
+	SKIP_TEST=$(SKIP_TEST) uv run pytest --capture=no -vv
 
 .PHONY: ci-test
 ci-test: install-deps-dev format-check lint test ## run CI tests

compose.yaml

@@ -0,0 +1,9 @@
+services:
+  pgadmin:
+    image: dpage/pgadmin4
+    ports:
+      - "8888:80"
+    environment:
+      PGADMIN_DEFAULT_EMAIL: [email protected]
+      PGADMIN_DEFAULT_PASSWORD: very-strong-password
+      PGADMIN_CONFIG_ENHANCED_COOKIE_PROTECTION: "False"

docs/index.md

@@ -36,14 +36,26 @@ make codegen
 
 #### Credentials
 
-Run the following command to set up the credentials for the application.
+1. Run the following command to set up the credentials for the application.
 
 ```shell
 # Run the application
 uv run streamlit run workshop_playwright_python/apps/streamlit_authentication.py
 ```
 
-Then, type your credentials described in [.streamlit/config.yaml](../.streamlit/config.yaml) and click the "Login" button. (e.g. `jsmith:abc`, `rbriggs:def`)
+2. Type your credentials described in [.streamlit/config.yaml](../.streamlit/config.yaml) and click the "Login" button. (e.g. `jsmith:abc`, `rbriggs:def`)
+
+3. Run scripts to save the context.
+
+```shell
+uv run scripts/save_context.py
+```
+
+4. Run the following command to load the context.
+
+```shell
+uv run scripts/load_context.py
+```
 
 #### OpenID Connect
 

playwright/.auth/.gitkeep

@@ -0,0 +1,23 @@
+from playwright.sync_api import Playwright, sync_playwright
+
+
+def run(playwright: Playwright) -> None:
+    browser = playwright.chromium.launch(headless=False)
+    context = browser.new_context(
+        storage_state="playwright/.auth/storage.json",
+    )
+    page = context.new_page()
+    page.goto("http://localhost:8888/")
+
+    # sleep for a few seconds to allow the page to load
+    import time
+
+    time.sleep(10)
+
+    # ---------------------
+    # context.close()
+    # browser.close()
+
+
+with sync_playwright() as playwright:
+    run(playwright)

scripts/load_context.py

@@ -0,0 +1,24 @@
+import json
+
+from playwright.sync_api import Playwright, StorageState, sync_playwright
+
+
+def run(playwright: Playwright) -> None:
+    browser = playwright.chromium.launch(headless=False)
+    context = browser.new_context()
+    page = context.new_page()
+    page.goto("http://localhost:8888/")
+    page.get_by_role("textbox", name="Username").fill("jsmith")
+    page.get_by_role("textbox", name="Password").fill("abc")
+    page.get_by_test_id("stBaseButton-secondaryFormSubmit").click()
+
+    storage: StorageState = context.storage_state(path="playwright/.auth/storage.json")
+    print(json.dumps(storage, indent=2))
+
+    # ---------------------
+    context.close()
+    browser.close()
+
+
+with sync_playwright() as playwright:
+    run(playwright)

scripts/save_context.py

@@ -0,0 +1,3 @@
+from os import getenv
+
+SKIP = getenv("SKIP_TEST", "true").lower() == "true"

tests/flags.py

@@ -1,8 +1,12 @@
 import json
 
+import pytest
+
 from playwright.sync_api import Page
+from tests import flags
 
 
+@pytest.mark.skipif(flags.SKIP, reason="This test is skipped because it requires a live server.")
 def test_hatena_bookmark(page: Page):
     page.goto("https://b.hatena.ne.jp/hotentry/all")