generator/conftest.py at master · EternalSoul/generator · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import shutil
import sys
from subprocess import Popen

import os
from shutil import rmtree, copy
from venv import EnvBuilder

import pytest

from pytest_forked import forked_run_report

from zmei_generator.cli.server import do_generate
from zmei_generator.parser.parser import ZmeiParser

skeleton_dir = os.path.join(os.path.dirname(__file__), 'zmei_generator/cli/skeleton')
samples_dir = os.path.join(os.path.dirname(__file__), 'tests/samples')
work_dir_prefix = os.path.join(os.path.dirname(__file__), 'tests/gen_result')
files_dir = os.path.join(os.path.dirname(__file__), 'tests/files')

base_path = os.path.realpath('.')

if os.environ.get('DEBUG') == '1':
    import pydevd
    pydevd.settrace('localhost', port=12345, stdoutToServer=True, stderrToServer=True, suspend=False)


def pytest_sessionstart(session):
    if os.path.exists(work_dir_prefix):
        rmtree(work_dir_prefix)
    os.mkdir(work_dir_prefix)


def pytest_addoption(parser):
    parser.addoption("--zmei", action="store_true", help="only run tests that are market with col().")


def pytest_configure(config):
    config.addinivalue_line("markers",
                            "zmei(app_name): test requires generated django application from .col file")


def pytest_runtest_teardown(item):
    os.chdir(base_path)


def pytest_runtest_setup(item):
    import sys
    import os

    before = item.get_marker("zmei_before")
    if before:
        commands_before = before.args
    else:
        commands_before = []

    colmarker = item.get_marker("zmei")
    zmei = item.config.getoption("--zmei")

    if zmei and not colmarker:
        pytest.skip("Running only zmei tests")

    if colmarker:
        # prepare workdir
        work_dir = os.path.join(work_dir_prefix, item.name)
        if os.path.exists(work_dir):
            rmtree(work_dir)
        os.mkdir(work_dir)

        application_name = colmarker.args[0]

        if len(colmarker.args) > 1:
            with open(f'{work_dir}/{application_name}.col', 'w') as f:
                f.write(colmarker.args[1])
        else:
            shutil.copy(f'{samples_dir}/{application_name}.col', f'{work_dir}/{application_name}.col')

        copy(f'{files_dir}/dev.db', f'{work_dir}/dev.db')

        do_generate(work_dir, [application_name])

        os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'app.settings')

        add_to_path(work_dir)

        import django
        django.setup()

        for command in commands_before:
            if command == 'migrate':
                call_django_command('makemigrations {}'.format(application_name), work_dir)

            if command == 'install':
                continue
            else:
                call_django_command(command, work_dir)


def add_to_path(work_dir):
    paths_to_delete = []
    for path in sys.path:
        if path.startswith(work_dir_prefix):
            paths_to_delete.append(path)
    for path in paths_to_delete:
        sys.path.remove(path)
    sys.path.append(work_dir)

    print(sys.path)


def call_django_command(django_command, work_dir):
    import sys
    env = os.environ.copy()

    env.update({
        "PYTHONPATH": ":".join([x for x in sys.path if x != ''] + [work_dir]),
    })
    p1 = Popen([sys.executable, 'manage.py'] + django_command.split(' '), env=env, cwd=work_dir)
    p1.wait()


@pytest.mark.tryfirst
def pytest_runtest_protocol(item):
    colmarker = item.get_marker("zmei")

    if colmarker:
        reports = forked_run_report(item)
        for rep in reports:
            item.ihook.pytest_runtest_logreport(report=rep)
        return True

# def pytest_runtest_setup(item):
#     envmarker = item.get_marker("env")
#     if envmarker is not None:
#         envname = envmarker.args[0]
#         if envname != item.config.getoption("-E"):
#             pytest.skip("test requires env %r" % envname)
#