更新 app/test/test_bug551_DingZeYu.py

我这边好像没有helper文件所以我直接判断有没有这个账号，如果没有就注册一个来进行测试。这样子可以吗

app/test/test_bug551_DingZeYu.py

@@ -3,6 +3,9 @@ import pytest
 from selenium import webdriver
 from selenium.webdriver import ActionChains
 from selenium.webdriver.common.by import By
+from selenium.webdriver.common.alert import Alert
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.support.wait import WebDriverWait
 
 
 def test_bug551():
@@ -11,10 +14,24 @@ def test_bug551():
     driver.get("http://127.0.0.1:5000/")
 
     driver.find_element(By.CSS_SELECTOR, 'a[href="/login"]').click()
-    driver.find_element(By.CLASS_NAME, 'username').send_keys('dzy')
+    driver.find_element(By.ID, 'username').send_keys('dzy')
-    driver.find_element(By.CLASS_NAME, 'password').send_keys('123456')
+    driver.find_element(By.ID, 'password').send_keys('123456')
     driver.find_element(By.CLASS_NAME, 'btn').click()
 
+    try:
+        alert = WebDriverWait(driver, 10).until(EC.alert_is_present())
+        alert.accept()
+        driver.find_element(By.CSS_SELECTOR, 'a[href="/signup"]').click()
+        driver.find_element(By.ID, 'username').send_keys('dzy')
+        driver.find_element(By.ID, 'password').send_keys('123456')
+        driver.find_element(By.ID, 'password2').send_keys('123456')
+        driver.find_element(By.CLASS_NAME, 'btn').click()
+        alert = WebDriverWait(driver, 10).until(EC.alert_is_present())
+        alert.accept()
+
+    except:
+        pass
+
     article = driver.find_element(By.ID, 'article')
     actions = ActionChains(driver)