# coding=utf-8

import unittest
import threading
from driver import browser
import env
import signal
import utils
import dataUtils
import sys
from time import ctime


class BaseUnitTest(unittest.TestCase):
    # --实例属性区--
    """是否开启TestCase级别的单驱动模式,默认False
    开启时, 一个TestCase只会打开一个浏览器,也就是method之间是公用浏览器,
    而不是每个测试Method都打开一个浏览器
    """
    SINGLE_DRIVER = False

    # --类方法区--
    @classmethod
    def setUpClass(cls):
        env.thread_local.SINGLE_DRIVER = cls.SINGLE_DRIVER

    @classmethod
    def tearDownClass(cls, is_thread=False):
        if is_thread and env.thread_local.SINGLE_DRIVER:
            BaseUnitTest.__quit_driver(env.thread_local.driver)

    # --静态方法区--
    @staticmethod
    def __new_driver():
        driver = browser(env.BROWSER, env.DRIVER_PATH)
        driver.maximize_window()
        return driver

    @staticmethod
    def __quit_driver(driver):
        if env.BROWSER.upper() == 'PHANTOMJS':
            # 这里是为了能够真正的退出
            driver.service.process.send_signal(signal.SIGTERM)
        driver.quit()

    # --实例方法区--
    # 1.重写父类区
    def __init__(self, methodName='runTest'):
        ''' overide the TestCase __init__ Method
            so need to recall the TestCase __init__ method
        '''
        super(BaseUnitTest, self).__init__(methodName)
        self.__testMethodName = methodName

    def setUp(self):
        if env.thread_local.SINGLE_DRIVER:
            if env.thread_local.driver is None:
                env.thread_local.driver = BaseUnitTest.__new_driver()
            self.driver = env.thread_local.driver
        else:
            self.driver = BaseUnitTest.__new_driver()

    def tearDown(self):
        if not env.thread_local.SINGLE_DRIVER:
            BaseUnitTest.__quit_driver(self.driver)

    # 2.自定义方法区
    def assertEqualUnicode(self, a, b):
        a = dataUtils.decodeToUnicode(a)
        b = dataUtils.decodeToUnicode(b)
        self.assertEqual(a, b)

    # 3.私有方法区
    def __exc_info(self):
        ''' copy from TestCase ,aim to overide the TestCase __call__ method
        '''
        exctype, excvalue, tb = sys.exc_info()
        if sys.platform[:4] == 'java':  # tracebacks look different in Jython
            return (exctype, excvalue, tb)
        newtb = tb.tb_next
        if newtb is None:
            return (exctype, excvalue, tb)
        return (exctype, excvalue, newtb)

    def __call__(self, result=None):
        ''' overide the TestCase __call__ method
            aim to add insert_img_decorator to the test_case method
        '''
        if result is None:
            result = self.defaultTestResult()
        result.startTest(self)
        testMethod = getattr(self, self.__testMethodName)
        newMethod = utils.insert_img_decorator(
            self.__testMethodName)(testMethod)
        try:
            try:
                self.setUp()
            except KeyboardInterrupt:
                raise
            except:
                result.addError(self, self.__exc_info())
                return

            ok = 0
            try:
                newMethod(self)
                ok = 1
            except self.failureException, e:
                result.addFailure(self, self.__exc_info())
            except KeyboardInterrupt:
                raise
            except:
                result.addError(self, self.__exc_info())

            try:
                self.tearDown()
            except KeyboardInterrupt:
                raise
            except:
                result.addError(self, self.__exc_info())
                ok = 0
            if ok:
                result.addSuccess(self)
        finally:
            result.stopTest(self)