from datetime import datetime
import time
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys

import pytest
from selenium import webdriver
from po.login_page import LoginPage
from po.home_page import HomePage
from po.test_package_list_page import TestPackageListPage
from po.share_add_page import ShareAddPage
from po.maq_answer_page import MAQAnswerPage
from comm.my_random import *


class TestLogin:
    driver = None

    @pytest.mark.parametrize('data_read', ["login"], indirect=True)
    def test_login(self, data_read: dict, driver):
        """
        打开页面并进行登录测试
        :param data_read: 执行数据读取的函数并获取到返回值
        :return:
        """
        login = LoginPage(driver)

        # 打开链接
        login.goto(data_read["url"])
        # 浏览器窗口最大化
        driver.maximize_window()
        # 通过标题判断是否访问正确
        while True:
            if driver.title == "OTS":
                # 标题等于OTS退出循环
                break
            else:
                # 标题不等于OTS休眠
                time.sleep(5)
        # 登录
        login.login(data_read["username"], data_read["password"], data_read["code"])
        # 断言登录是否成功
        assert login.login_assert()

    # def test_logout(self):
    #     driver = TestLogin.driver
    #     home = HomePage(driver)
    #     # 登出
    #     home.logout()


if __name__ == '__main__':
    pytest.main(["-s", __file__])