test_env.py

# coding: utf-8

import urllib.request as request
import os
import subprocess
import shutil
import sys
import zipfile
import argparse

"""
Setup environment for run test
"""

TMP_FOLDER = os.sep.join([os.getcwd(), 'install'])


def head():
    """Print head"""

    title = """\n
    #####################################################
    # Setup environment for run tests.                  #
    # Install:                                          #
    # - chromedriver                                    #
    # - selenium-standalone-server                      #
    # - test                                            #
    # - Page Objects                                    #
    # - lib selenium                                    #
    #####################################################
    """

    print(title)


def svn_checkout(url, path, user, password):
    """Call svn checkout"""

    exit_code = subprocess.Popen(['svn', 'co', url, path, '--username', user, '--password', password]).wait()
    if exit_code != 0:
        raise Exception('Don\'t get data from repository')


def download_file(url, file_name):
    """Download file from remote host"""

    request.urlretrieve(url=url, filename=os.path.join(TMP_FOLDER, file_name))


def get_chromedriver(ver_chromedriver):
    """Get chromedriver from internet"""

    win_filename = 'chromedriver_win32.zip'
    url_chromedriver = 'http://chromedriver.storage.googleapis.com/{}/{}'.format(ver_chromedriver, win_filename)
    download_file(url_chromedriver, win_filename)


def get_standalone_server(ver_standalone_server):
    """Get selenium-standalone-server.jar"""

    patch = '1'
    file_standalone_server = 'selenium-server-standalone-{ver}.{p}.jar'.format(ver=ver_standalone_server,
                                                                               p=patch or '0')
    url_selenium_server = 'http://selenium-release.storage.googleapis.com/{}/{}'.format(ver_standalone_server,
                                                                                        file_standalone_server)
    download_file(url_selenium_server, file_standalone_server)


def get_atf(ver_atf, user, password):
    """Get automated testing framework"""

    url = 'https://svn.sbis.ru/svn/sbis3-tests/atf/release/{}/'.format(ver_atf)
    path = os.path.join(TMP_FOLDER, 'atf')
    if os.path.exists(path):
        shutil.rmtree(path)
    svn_checkout(url, path, user, password)


def get_genie_tests(ver_genie, user, password):
    """Get test from svn"""

    url = 'https://svn.sbis.ru/svn/sbis3-tests/genie/{}/'.format(ver_genie)
    path = os.path.join(TMP_FOLDER, 'tests')
    if os.path.exists(path):
        shutil.rmtree(path)
    svn_checkout(url, path, user, password)


def install_lib(user, password):
    """Install lib's through pip"""

    os.environ['HTTP_PROXY'] = 'http://{}:{}@ias.corp.tensor.ru:8080'.format(user, password)
    p = subprocess.Popen(['pip', 'install', '-r', 'requirement.txt', '-U', '--disable-pip-version-check'])
    exit_code = p.wait()
    if exit_code != 0:
        raise Exception('Don\'t install packet from pip')


def copy_into_site_packages():
    """Copy atf, PO into folder site-packages"""

    site_packages = [path for path in sys.path if 'site-packages' in path][0]
    try:
        shutil.copytree(os.path.join(TMP_FOLDER, 'atf'), os.path.join(site_packages, 'atf'))
        shutil.copytree(os.path.join(TMP_FOLDER, 'tests', 'module_genie'), os.path.join(site_packages, 'module_genie'))
        shutil.copytree(os.path.join(TMP_FOLDER, 'tests', 'pages'), os.path.join(site_packages, 'pages'))
    except FileExistsError:
        pass


def copy_chromedriver():
    """Copy chromedriver in execute path"""

    exec_dir = sys.prefix
    zipfile.ZipFile(os.path.join(TMP_FOLDER, 'chromedriver_win32.zip')).extractall(path=os.path.join(exec_dir, 'bin'))


def add_arguments():
    """Add arguments"""

    parg = argparse.ArgumentParser()
    parg.add_argument('-cd', help='Version chromedriver')
    parg.add_argument('-ss', help='Version standalone-selenium-server')
    parg.add_argument('-u', help='SVN username')
    parg.add_argument('-p', help='SVN password')
    parg.add_argument('-wg', help='Version genie')
    parg.add_argument('-atf', help='Version atf')

    return parg


def main():
    """Common algorithm"""
    args = add_arguments()
    p = args.parse_args()
    if not os.path.exists(TMP_FOLDER):
        print('[*] create tmp folder: ', TMP_FOLDER)
        os.mkdir(TMP_FOLDER)

    print('[*] get chromedriver..')
    get_chromedriver(p.cd)

    print('[*] get selenium-standalone-server..')
    get_standalone_server(p.ss)

    print('[*] get atf..')
    get_atf(p.atf, p.u, p.p)

    print('[*] get tests..')
    get_genie_tests(p.wg, p.u, p.p)

    print('[*] install lib..')
    install_lib(p.u, p.p)

    print('[*] copy data into site-packages..')
    copy_into_site_packages()

    print('[*] copy chromedriver..')
    copy_chromedriver()


if __name__ == '__main__':
    head()
    main()