open-bounty/test/end-to-end/tests/basetestcase.py

import pytest, sys, os
from selenium import webdriver
from selenium.common.exceptions import WebDriverException
from tests.postconditions import remove_application, remove_installation
from os import environ
from tests import test_data
from pages.thirdparty.github import GithubPage
from pages.openbounty.landing import LandingPage


class BaseTestCase:

    def print_sauce_lab_info(self, driver):
        sys.stdout = sys.stderr
        print("SauceOnDemandSessionID=%s job-name=%s" % (driver.session_id,
                                                         pytest.config.getoption('build')))

    def get_remote_caps(self):
        sauce_lab_cap = dict()
        sauce_lab_cap['name'] = test_data.test_name
        sauce_lab_cap['build'] = pytest.config.getoption('build')
        sauce_lab_cap['idleTimeout'] = 900
        sauce_lab_cap['commandTimeout'] = 500
        sauce_lab_cap['platform'] = "MAC"
        sauce_lab_cap['browserName'] = 'Chrome'
        sauce_lab_cap['screenResolution'] = '2048x1536'
        sauce_lab_cap['captureHtml'] = False
        return sauce_lab_cap

    def verify_no_errors(self):
        if self.errors:
            msg = ''
            for error in self.errors:
                msg += (error + '\n')
            pytest.fail(msg, pytrace=False)

    @classmethod
    def setup_class(cls):
        cls.errors = []
        cls.environment = pytest.config.getoption('env')

        ################################################################################################################
        ######### Drivers setup
        ################################################################################################################

        #
        # Dev Chrome options
        #
        cls.capabilities_dev = webdriver.ChromeOptions()
        cls.capabilities_dev.add_argument('--start-fullscreen')

        #
        # Org Chrome options
        #
        cls.capabilities_org = webdriver.ChromeOptions()
        cls.capabilities_org.add_extension(
            os.path.join(test_data.tests_path, os.pardir, 'resources', 'metamask3_12_0.crx'))

        #
        # SauceLab capabilities
        #
        cls.executor_sauce_lab = 'http://%s:%s@ondemand.saucelabs.com:80/wd/hub' % (
            environ.get('SAUCE_USERNAME'), environ.get('SAUCE_ACCESS_KEY'))
        cls.drivers = []

        if cls.environment == 'local':
            for caps in cls.capabilities_dev, cls.capabilities_org:
                driver = webdriver.Chrome(chrome_options=caps)
                cls.drivers.append(driver)

        if cls.environment == 'sauce':
            for caps in cls.capabilities_dev, cls.capabilities_org:
                remote = cls.get_remote_caps(cls)
                new_caps = caps.to_capabilities()
                new_caps.update(remote)
                driver = webdriver.Remote(cls.executor_sauce_lab,
                                          desired_capabilities=new_caps)
                cls.drivers.append(driver)

        cls.driver_dev = cls.drivers[0]
        cls.driver_org = cls.drivers[1]

        for driver in cls.drivers:
            driver.implicitly_wait(10)

        ################################################################################################################
        ######### Actions for each driver before class
        ################################################################################################################

        ######ORG
        landing = LandingPage(cls.driver_org)
        landing.get_landing_page()

        # Sign Up to SOB
        cls.github_org = landing.login_button.click()
        cls.github_org.sign_in(test_data.config['ORG']['gh_login'],
                               test_data.config['ORG']['gh_password'])
        assert cls.github_org.permission_type.text == 'Personal user data'
        bounties_page = cls.github_org.authorize_sob.click()

        # SOB Plugin installation and navigate to "Open bounties"
        cls.github_org.install_sob_plugin()
        assert bounties_page.bounties_header.text == 'Bounties'
        assert bounties_page.top_hunters_header.text == 'Top 5 hunters'

        ######DEV
        cls.github_dev = GithubPage(cls.driver_dev)
        # Sign In to GH as Developer
        cls.github_dev.get_login_page()
        cls.github_dev.sign_in(test_data.config['DEV']['gh_login'],
                               test_data.config['DEV']['gh_password'])

        # Fork repo as Developer from Organization
        cls.github_dev.fork_repo(test_data.config['ORG']['gh_repo'])

        # Cloning repo to local git as Developer and set upstream to Organization (via HTTPS)
        cls.github_dev.clone_repo(test_data.config['ORG']['gh_repo'],
                                  test_data.config['DEV']['gh_username'],
                                  test_data.config['ORG']['gh_repo_name'])
        cls.verify_no_errors(cls)

    @classmethod
    def teardown_class(cls):

        ######ORG

        # SOB Plugin remove installation
        remove_application(cls.driver_org)
        remove_installation(cls.driver_org)

        ######DEV
        cls.github_dev.clean_repo_local_folder()
        cls.github_dev.delete_fork()

        try:
            for driver in cls.drivers:
                cls.print_sauce_lab_info(cls, driver)
                driver.quit()

        except WebDriverException:
            pass