mirror of
https://github.com/ansible-collections/community.crypto.git
synced 2026-04-13 20:31:04 +00:00
124 lines
4.1 KiB
Python
124 lines
4.1 KiB
Python
# Copyright (c) Ansible Project
|
|
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
__metaclass__ = type
|
|
|
|
|
|
import pytest
|
|
|
|
from ansible_collections.community.crypto.tests.unit.compat.mock import MagicMock
|
|
|
|
|
|
from ansible_collections.community.crypto.plugins.module_utils.acme.backend_cryptography import (
|
|
HAS_CURRENT_CRYPTOGRAPHY,
|
|
CryptographyBackend,
|
|
)
|
|
|
|
from ansible_collections.community.crypto.plugins.module_utils.crypto.support import (
|
|
ensure_utc_timezone,
|
|
)
|
|
|
|
from ansible_collections.community.crypto.plugins.module_utils.crypto.cryptography_support import (
|
|
CRYPTOGRAPHY_TIMEZONE,
|
|
)
|
|
|
|
from .backend_data import (
|
|
TEST_KEYS,
|
|
TEST_CSRS,
|
|
TEST_CERT,
|
|
TEST_CERT_DAYS,
|
|
TEST_CERT_INFO,
|
|
TEST_PARSE_ACME_TIMESTAMP,
|
|
TEST_INTERPOLATE_TIMESTAMP,
|
|
)
|
|
|
|
|
|
if not HAS_CURRENT_CRYPTOGRAPHY:
|
|
pytest.skip('cryptography not found')
|
|
|
|
|
|
@pytest.mark.parametrize("pem, result, dummy", TEST_KEYS)
|
|
def test_eckeyparse_cryptography(pem, result, dummy, tmpdir):
|
|
fn = tmpdir / 'test.pem'
|
|
fn.write(pem)
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
key = backend.parse_key(key_file=str(fn))
|
|
key.pop('key_obj')
|
|
assert key == result
|
|
key = backend.parse_key(key_content=pem)
|
|
key.pop('key_obj')
|
|
assert key == result
|
|
|
|
|
|
@pytest.mark.parametrize("csr, result, openssl_output", TEST_CSRS)
|
|
def test_csridentifiers_cryptography(csr, result, openssl_output, tmpdir):
|
|
fn = tmpdir / 'test.csr'
|
|
fn.write(csr)
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
identifiers = backend.get_csr_identifiers(csr_filename=str(fn))
|
|
assert identifiers == result
|
|
identifiers = backend.get_csr_identifiers(csr_content=csr)
|
|
assert identifiers == result
|
|
|
|
|
|
@pytest.mark.parametrize("now, expected_days", TEST_CERT_DAYS)
|
|
def test_certdays_cryptography(now, expected_days, tmpdir):
|
|
fn = tmpdir / 'test-cert.pem'
|
|
fn.write(TEST_CERT)
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
days = backend.get_cert_days(cert_filename=str(fn), now=now)
|
|
assert days == expected_days
|
|
days = backend.get_cert_days(cert_content=TEST_CERT, now=now)
|
|
assert days == expected_days
|
|
|
|
|
|
@pytest.mark.parametrize("cert_content, expected_cert_info, openssl_output", TEST_CERT_INFO)
|
|
def test_get_cert_information(cert_content, expected_cert_info, openssl_output, tmpdir):
|
|
fn = tmpdir / 'test-cert.pem'
|
|
fn.write(cert_content)
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
|
|
if CRYPTOGRAPHY_TIMEZONE:
|
|
expected_cert_info = expected_cert_info._replace(
|
|
not_valid_after=ensure_utc_timezone(expected_cert_info.not_valid_after),
|
|
not_valid_before=ensure_utc_timezone(expected_cert_info.not_valid_before),
|
|
)
|
|
|
|
cert_info = backend.get_cert_information(cert_filename=str(fn))
|
|
assert cert_info == expected_cert_info
|
|
cert_info = backend.get_cert_information(cert_content=cert_content)
|
|
assert cert_info == expected_cert_info
|
|
|
|
|
|
def test_now():
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
now = backend.get_now()
|
|
assert CRYPTOGRAPHY_TIMEZONE == (now.tzinfo is not None)
|
|
|
|
|
|
@pytest.mark.parametrize("input, expected", TEST_PARSE_ACME_TIMESTAMP)
|
|
def test_parse_acme_timestamp(input, expected):
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
ts_expected = backend.get_utc_datetime(**expected)
|
|
timestamp = backend.parse_acme_timestamp(input)
|
|
assert ts_expected == timestamp
|
|
|
|
|
|
@pytest.mark.parametrize("start, end, percentage, expected", TEST_INTERPOLATE_TIMESTAMP)
|
|
def test_interpolate_timestamp(start, end, percentage, expected):
|
|
module = MagicMock()
|
|
backend = CryptographyBackend(module)
|
|
ts_start = backend.get_utc_datetime(**start)
|
|
ts_end = backend.get_utc_datetime(**end)
|
|
ts_expected = backend.get_utc_datetime(**expected)
|
|
timestamp = backend.interpolate_timestamp(ts_start, ts_end, percentage)
|
|
assert ts_expected == timestamp
|