测试脚本的编写

from time import sleep
from Base.BasePage import *
from Conf.Read_ini import *
from Pages.LeftBlank import *
from Pages.SysSetting import *
from Usual.LoginMethod import *


class TestDistrictMaintain(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = Browser().driver
        cls.url = conf.url
        cls.name = conf.account
        cls.passwd = conf.password
        cls.province = conf.getYaml('DistrictMaintain', 'province')

    #@unittest.skip('must')
    def test_01(self):
        """
    菜单新增
        """
        d = self.driver
        Link(d).open(self.url)
        login_server(d, self.name, self.passwd)
        Blank(d).click_sys_setting()
        Link(d).click_dis_maintenance()
        DisMaintain(d).click_choose_drop()
        DisMaintain(d).choose_dis(self.province)
        DisMaintain(d).click_city_new()
        DisMaintain(d).input_city_code(self.province)
        '''
        path = os.path.join(os.path.dirname(os.path.dirname(__file__)), 'TestReport', 'Screen', 'DistrictError.png')
        print(path)
        d.get_screenshot_as_file(path)
        '''
        sleep(2)

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()


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