Paul Kehrer | 7e4bc6d | 2013-12-24 22:23:53 -0600 | [diff] [blame] | 1 | import pytest |
| 2 | |
Alex Stapleton | 0d58373 | 2014-01-10 22:39:12 +0000 | [diff] [blame] | 3 | from cryptography.hazmat.backends import _ALL_BACKENDS |
Paul Kehrer | 7e4bc6d | 2013-12-24 22:23:53 -0600 | [diff] [blame] | 4 | from cryptography.hazmat.backends.interfaces import ( |
| 5 | HMACBackend, CipherBackend, HashBackend |
| 6 | ) |
| 7 | |
Paul Kehrer | 34c075e | 2014-01-13 21:52:08 -0500 | [diff] [blame^] | 8 | from .utils import check_for_iface, check_backend_support, modify_backend_list |
| 9 | |
| 10 | |
| 11 | def pytest_generate_tests(metafunc): |
| 12 | name = metafunc.config.getoption("--backend") |
| 13 | modify_backend_list(name, _ALL_BACKENDS) |
Paul Kehrer | 7e4bc6d | 2013-12-24 22:23:53 -0600 | [diff] [blame] | 14 | |
| 15 | |
Alex Stapleton | 0d58373 | 2014-01-10 22:39:12 +0000 | [diff] [blame] | 16 | @pytest.fixture(params=_ALL_BACKENDS) |
| 17 | def backend(request): |
| 18 | return request.param |
Paul Kehrer | 7e4bc6d | 2013-12-24 22:23:53 -0600 | [diff] [blame] | 19 | |
| 20 | |
| 21 | @pytest.mark.trylast |
| 22 | def pytest_runtest_setup(item): |
Alex Gaynor | 2b3f942 | 2013-12-24 21:55:24 -0800 | [diff] [blame] | 23 | check_for_iface("hmac", HMACBackend, item) |
| 24 | check_for_iface("cipher", CipherBackend, item) |
| 25 | check_for_iface("hash", HashBackend, item) |
Paul Kehrer | 60fc8da | 2013-12-26 20:19:34 -0600 | [diff] [blame] | 26 | check_backend_support(item) |
Paul Kehrer | 34c075e | 2014-01-13 21:52:08 -0500 | [diff] [blame^] | 27 | |
| 28 | |
| 29 | def pytest_addoption(parser): |
| 30 | parser.addoption( |
| 31 | "--backend", action="store", metavar="NAME", |
| 32 | help="Only run tests matching the backend NAME." |
| 33 | ) |