完成一个class的测试用例

test_demo.py

@@ -3,44 +3,45 @@ import pytest
 from config import *
 
 
-def setup():
+class Test_Admin():
+    def setup_class(self):
         # setup 步骤
+        print('setup')
         driver = sync_playwright().start()
         browser = driver.chromium.launch(headless=False)
-    context = browser.new_context()
+        self.context = browser.new_context()
-    page = context.new_page()
+        self.page = self.context.new_page()
         
+    def teardown_class(self):
+        print('teardown')
+        # teardown 步骤
+        self.context.close()
+        self.page.close()
 
-def test_admin_login(page):
+    def test_admin_login(self):
         # 访问医馆首页
-    page.goto(BASE_URL)
+        self.page.goto(BASE_URL)
 
         # 输入手机号后失焦触发获取医馆列表事件
-    mobile_input = page.get_by_placeholder('请输入手机号')
+        mobile_input = self.page.get_by_placeholder('请输入手机号')
         mobile_input.fill(ADMIN_ACCOUNT)
         mobile_input.blur()
 
         # 选择厚生堂诊所
-    page.get_by_placeholder('请选择诊所名称').click()
+        self.page.get_by_placeholder('请选择诊所名称').click()
-    page.get_by_text(CLINIC_NAME).click()
+        self.page.get_by_text(CLINIC_NAME).click()
 
         # 输入密码
-    page.get_by_placeholder('请输入密码').fill(ADMIN_PWD)
+        self.page.get_by_placeholder('请输入密码').fill(ADMIN_PWD)
 
         #点击登录
-    page.get_by_role('button', name='登录').click()
+        self.page.get_by_role('button', name='登录').click()
 
+        print(self.page.locator('.max-logo').inner_text())
         # 验证测试结果
-    assert page.locator('.max-logo').inner_text == CLINIC_NAME
+        assert self.page.locator('.max-logo').inner_text() == CLINIC_NAME
 
         # 临时代码
-    page.wait_for_timeout(2000)
+        self.page.wait_for_timeout(2000)
-
-def teardown(context, page):
-    # teardown 步骤
-    context.close()
-    page.close()
 
    
-if __name__ == '__main__':
-    pytest.main('-vs')