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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
#!/usr/bin/env python
#############################################################################
# Copyright (c) 2015-2018 Balabit
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 2 as published
# by the Free Software Foundation, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
#
# As an additional exemption you are allowed to compile & link against the
# OpenSSL libraries as published by the OpenSSL project. See the file
# COPYING for details.
#
#############################################################################
import argparse
import logging
import subprocess
from datetime import datetime
from pathlib import Path
import pytest
import src.testcase_parameters.testcase_parameters as tc_parameters
from src.helpers.loggen.loggen import Loggen
from src.message_builder.bsd_format import BSDFormat
from src.message_builder.log_message import LogMessage
from src.syslog_ng.syslog_ng import SyslogNg
from src.syslog_ng.syslog_ng_paths import SyslogNgPaths
from src.syslog_ng_config.syslog_ng_config import SyslogNgConfig
from src.syslog_ng_ctl.syslog_ng_ctl import SyslogNgCtl
from src.testcase_parameters.testcase_parameters import TestcaseParameters
logger = logging.getLogger(__name__)
class InstallDirAction(argparse.Action):
def __call__(self, parser, namespace, path, option_string=None):
install_dir = Path(path)
if not install_dir.is_dir():
raise argparse.ArgumentTypeError("{0} is not a valid directory".format(path))
binary = Path(install_dir, "sbin/syslog-ng")
if not binary.exists():
raise argparse.ArgumentTypeError("{0} not exist".format(binary))
setattr(namespace, self.dest, path)
# Command line options
def pytest_addoption(parser):
parser.addoption("--runslow", action="store_true", default=False, help="run slow tests")
parser.addoption("--run-under", help="Run syslog-ng under selected tool, example tools: [valgrind, strace]")
parser.addoption(
"--installdir",
action=InstallDirAction,
help="Set installdir for installed syslog-ng. Used when installmode is: custom. Example path: '/home/user/syslog-ng/installdir/'",
)
parser.addoption(
"--reports",
action="store",
default=get_relative_report_dir(),
help="Path for report files folder. Default form: 'reports/<current_date>'",
)
def get_relative_report_dir():
return str(Path("reports/", get_current_date()))
def get_current_date():
return datetime.now().strftime("%Y-%m-%d-%H-%M-%S-%f")
# Pytest Hooks
def pytest_collection_modifyitems(config, items):
if config.getoption("--runslow"):
return
skip_slow = pytest.mark.skip(reason="need --runslow option to run")
for item in items:
if "slow" in item.keywords:
item.add_marker(skip_slow)
def pytest_runtest_logreport(report):
if report.outcome == "failed":
logger.error("\n{}".format(report.longrepr))
# Pytest Fixtures
@pytest.fixture
def testcase_parameters(request):
return TestcaseParameters(request)
@pytest.fixture
def config(request, teardown):
return SyslogNgConfig(request.getfixturevalue("version"), teardown)
@pytest.fixture
def syslog_ng(request, testcase_parameters, teardown):
tc_parameters.INSTANCE_PATH = SyslogNgPaths(testcase_parameters).set_syslog_ng_paths("server")
syslog_ng = SyslogNg(tc_parameters.INSTANCE_PATH, testcase_parameters, teardown)
teardown.register(syslog_ng.stop)
return syslog_ng
class TeardownRegistry:
teardown_callbacks = []
def register(self, teardown_callback):
TeardownRegistry.teardown_callbacks.append(teardown_callback)
def execute_teardown_callbacks(self):
for teardown_callback in TeardownRegistry.teardown_callbacks:
teardown_callback()
@pytest.fixture(scope="function")
def teardown():
teardown_registry = TeardownRegistry()
yield teardown_registry
teardown_registry.execute_teardown_callbacks()
@pytest.fixture
def syslog_ng_ctl(syslog_ng):
return SyslogNgCtl(syslog_ng.instance_paths)
@pytest.fixture
def bsd_formatter():
return BSDFormat()
@pytest.fixture
def log_message():
return LogMessage()
@pytest.fixture(scope="session")
def version(request):
installdir = request.config.getoption("--installdir")
binary_path = str(Path(installdir, "sbin", "syslog-ng"))
version_output = subprocess.check_output([binary_path, "--version"]).decode()
return version_output.splitlines()[1].split()[2]
@pytest.fixture
def loggen(testcase_parameters):
server = Loggen()
yield server
server.stop()
|
