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
|
# -*- coding: utf-8 -*-
"""
Pytest configuration.
"""
import decimal
import os
from configparser import ConfigParser
import pytest
import tests.helpers as th
from .helpers import clear_db, get_app_lock, release_app_lock
cdir = os.path.dirname(__file__)
cfgpath = os.path.join(cdir, 'tests.cfg')
@pytest.fixture(scope="module")
def mssql_conn():
return th.mssqlconn()
_parser = ConfigParser({
'server': 'localhost',
'username': 'sa',
'password': 'sqlServerPassw0rd',
'database': 'tempdb',
'port': '1433',
'ipaddress': '127.0.0.1',
'instance': '',
})
optional_markers = {
"slow": {"help": "Skip long tests",
"marker-descr": "Mark tests that run longer than ~3 seconds",
"skip-reason": "Test runs too long."},
"mssql_server_required": {"help": "Skip tests that require MSSQL server",
"marker-descr": "Mark tests that require MSSQL server",
"skip-reason": "Test only runs if MSSQL server is available."},
# add further markers here
}
def pytest_addoption(parser):
parser.addoption(
"--pymssql-section",
type=str,
default=os.environ.get('PYMSSQL_TEST_CONFIG', None),
help="The name of the section to use from tests.cfg"
)
for marker, info in optional_markers.items():
parser.addoption("--skip-{}".format(marker.replace('_','-')), action="store_true",
default=False, help=info['help'])
def pytest_configure(config):
section = config.getoption('--pymssql-section')
if section is not None:
if not os.path.exists(cfgpath):
raise ValueError(f"Config file '{cfgpath}' dos not exist.")
_parser.read(cfgpath)
if not _parser.has_section(section):
raise ValueError('the tests.cfg file does not have section: %s' % section)
else:
section = 'DEFAULT'
th.config.server = os.getenv('PYMSSQL_TEST_SERVER') or _parser.get(section, 'server')
th.config.user = os.getenv('PYMSSQL_TEST_USERNAME') or _parser.get(section, 'username')
th.config.password = os.getenv('PYMSSQL_TEST_PASSWORD') or _parser.get(section, 'password')
th.config.database = os.getenv('PYMSSQL_TEST_DATABASE') or _parser.get(section, 'database')
th.config.port = os.getenv('PYMSSQL_TEST_PORT') or _parser.get(section, 'port')
th.config.ipaddress = os.getenv('PYMSSQL_TEST_IPADDRESS') or _parser.get(section, 'ipaddress')
th.config.instance = os.getenv('PYMSSQL_TEST_INSTANCE') or _parser.get(section, 'instance')
th.config.orig_decimal_prec = decimal.getcontext().prec
for marker, info in optional_markers.items():
config.addinivalue_line("markers",
"{}: {}".format(marker, info['marker-descr']))
if get_app_lock():
clear_db()
def pytest_unconfigure(config):
release_app_lock()
def pytest_collection_modifyitems(config, items):
marker = "mssql_server_required"
info = optional_markers[marker]
if th.global_mssqlconn is None or config.getoption("--skip-{}".format(marker.replace('_','-'))):
skip = pytest.mark.skip(reason=info['skip-reason'])
for item in items:
if marker in item.keywords:
item.add_marker(skip)
marker = "slow"
info = optional_markers[marker]
if th.global_mssqlconn is None or config.getoption("--skip-{}".format(marker)):
skip = pytest.mark.skip(reason=info['skip-reason'])
for item in items:
if marker in item.keywords:
item.add_marker(skip)
|
