#!/usr/bin/env python
# -*- coding:utf-8 -*-
from selenium import webdriver
import time

def search_something():
    driver = webdriver.Chrome(executable_path="D:\\work\\learn_to_old\\driver\\chromedriver.exe")
    driver.get("https://www.baidu.com")
    time.sleep(3)
    driver.find_element_by_id("kw").send_keys("银魂")
    driver.find_element_by_id("su").click()
    time.sleep(3)
    source = driver.page_source
    if "空知英秋" in source:
        print("success")
    else:
        print("fial")
    driver.quit()

search_something()

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from selenium import webdriver
import time

num = 0
fail_num = 0
suc_num = 0

def search_something(keyword, words):
    driver = webdriver.Chrome(executable_path="D:\\work\\learn_to_old\\driver\\chromedriver.exe")
    driver.get("https://www.baidu.com")
    time.sleep(3)
    driver.find_element_by_id("kw").send_keys(keyword)
    driver.find_element_by_id("su").click()
    time.sleep(3)
    global num, fail_num, suc_num
    num += 1
    source = driver.page_source
    if words in source:
        print(f"搜索{keyword}成功")
        suc_num += 1
    else:
        print(f"搜索{keyword}失败")
        fail_num += 1
    driver.quit()

data_list = [["银魂", "空知英秋"],["kali", "Kali Linux"], ["centos", "Red Hat"]]
for i in range(len(data_list)):
    search_something(data_list[i][0], data_list[i][1])
    
print(f"共执行{num}条用例，成功{suc_num}条，失败{fail_num}条")

#!/usr/bin/env python
# -*- coding:utf-8 -*-
from selenium import webdriver
import time
import unittest
from ddt import ddt,data,unpack

@ddt
class my_search(unittest.TestCase):
    def setUp(self):
        global num, fail_num, suc_num
        num = 0
        fail_num = 0
        suc_num = 0
    
    # data_list = [["银魂", "空知英秋"],["kali", "Kali Linux"], ["centos", "Red Hat"]]
    # 注意，这边传的各组参数要单独拿开（如下），如果跟上面这样传list，ddt会认为是一组参数，
    # 比如@data(data_list)
    # 需要注意的是，每次执行一组数据，整个类是重复执行一遍的，相当于循环执行这个类
    @data(["银魂", "空知英秋"],["kali", "Kali Linux"], ["centos", "Red Hat"])
    @unpack
    def test_search(self, keyword, words):
        self.driver = webdriver.Chrome(executable_path="D:\\work\\learn_to_old\\driver\\chromedriver.exe")
        self.driver.get("https://www.baidu.com")
        time.sleep(3)
        self.driver.find_element_by_id("kw").send_keys(keyword)
        self.driver.find_element_by_id("su").click()
        time.sleep(3)
        global num, fail_num, suc_num
        num += 1
        source = self.driver.page_source
        if words in source:
            print(f"搜索{keyword}成功")
            suc_num += 1
        else:
            print(f"搜索{keyword}失败")
            fail_num += 1
            
        def tearDown(self):
            self.driver.quit()
            print(f"本次共执行{num}条用例，成功{suc_num}条，失败{fail_num}条")

if __name__ == "__main__":
    unittest.main()

def read_file_to_line(file_path):
    with open(file_path, "r", encoding="utf-8") as f:
        lines = f.readlines()
    return lines
    
test_datas = read_file_to_line("step5_data.txt") #读取测试数据

for test_data in test_datas:
    test_data = eval(test_data) # 将字符串转换为json，就不需要json.dump啥的了
    test_steps = read_file_to_line("step5_steps.txt") # 读取测试步骤
    
    for test_step in test_steps:
        # 找到关键字“{{”，替换数据
        if "${" in test_step:
            # print("#"*10,type(test_step))
            # 使用正则找到变量，$前面要加\，即便是字符串前用了r
            key = re.search(r"\${(.*?)}", test_step).group(1)
            # print("==== key ======",key)
            # {{key}} value step  替换数据
            test_step = re.sub(r"\${%s}" %key, test_data[key], test_step)
            # print("#"*10,test_step)
            
        # print(type(test_step))
        # 根据line.count(‘|‘)来判断，步骤可以更加简化
        if "|" in test_step:
            eval_str_list = test_step.strip().split("|")
            # print("===========",eval_str_list)
            if len(eval_str_list) == 2:
                eval_str = f"{eval_str_list[0]}(‘{eval_str_list[1]}‘)"
            elif len(eval_str_list) == 3:
                eval_str = f"{eval_str_list[0]}(‘{eval_str_list[1]}‘, ‘{eval_str_list[2]}‘)"
            elif len(eval_str_list) == 4:
                eval_str = f"{eval_str_list[0]}(‘{eval_str_list[1]}‘, ‘{eval_str_list[2]}‘, ‘{eval_str_list[3]}‘)"
            elif len(eval_str_list) == 5:
                eval_str = f"{eval_str_list[0]}(‘{eval_str_list[1]}‘, ‘{eval_str_list[2]}‘, ‘{eval_str_list[3]}‘, ‘{eval_str_list[4]}‘)"
            elif len(eval_str_list) == 6:
                eval_str = f"{eval_str_list[0]}(‘{eval_str_list[1]}‘, ‘{eval_str_list[2]}‘, ‘{eval_str_list[3]}‘, ‘{eval_str_list[4]}‘, ‘{eval_str_list[5]}‘)"
            else:
                print("过长，需要优化程序")
                print(eval_str_list,"*"*50)                    
        else:
            # eval_str = strs.replace("do_something",test_step.strip())
            eval_str = f"{test_step.strip()}()"
        print(eval_str)
        # 异常处理
        try:
            eval(eval_str)
        except Exception as e:
            print(f"{e}")
print(f"共执行{num}条用例，成功{suc_num}条，失败{fail_num}条")