blob: 9049922ba51f8eb6271e8e9edd2dcf1d5a373006 (
plain)
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
|
# This file is dual licensed under the terms of the Apache License, Version
# 2.0, and the BSD License. See the LICENSE file in the root of this repository
# for complete details.
import pytest
from cryptography.hazmat.backends.openssl import backend as openssl_backend
from .utils import check_backend_support
def pytest_configure(config):
if config.getoption("--enable-fips"):
openssl_backend._enable_fips()
def pytest_report_header(config):
return "\n".join(
[
"OpenSSL: {}".format(openssl_backend.openssl_version_text()),
"FIPS Enabled: {}".format(openssl_backend._fips_enabled),
]
)
def pytest_addoption(parser):
parser.addoption("--wycheproof-root", default=None)
parser.addoption("--enable-fips", default=False)
def pytest_runtest_setup(item):
if openssl_backend._fips_enabled:
for marker in item.iter_markers(name="skip_fips"):
pytest.skip(marker.kwargs["reason"])
@pytest.fixture()
def backend(request):
check_backend_support(openssl_backend, request)
# Ensure the error stack is clear before the test
errors = openssl_backend._consume_errors_with_text()
assert not errors
yield openssl_backend
# Ensure the error stack is clear after the test
errors = openssl_backend._consume_errors_with_text()
assert not errors
@pytest.fixture
def disable_rsa_checks(backend):
# Use this fixture to skip RSA key checks in tests that need the
# performance.
backend._rsa_skip_check_key = True
yield
backend._rsa_skip_check_key = False
|
