from contextlib import nullcontext as does_not_raise from datetime import datetime from datetime import timedelta from datetime import timezone from pathlib import Path from tempfile import TemporaryDirectory from typing import ContextManager from typing import Dict from typing import Optional from unittest.mock import Mock from unittest.mock import patch from pytest import mark from pytest import raises from libkeyringctl import sequoia from libkeyringctl.types import Fingerprint from libkeyringctl.types import Uid from libkeyringctl.types import Username @mark.parametrize( "create_subdir, preserve_filename", [ (False, True), (False, False), (True, True), (True, False), ], ) @patch("libkeyringctl.sequoia.system") @patch("libkeyringctl.sequoia.mkdtemp") def test_keyring_split(mkdtemp_mock: Mock, system_mock: Mock, create_subdir: bool, preserve_filename: bool) -> None: with TemporaryDirectory() as tmp_dir_name: tmp_dir = Path(tmp_dir_name) keyring_tmp_dir = tmp_dir / "keyring" keyring_tmp_dir.mkdir() mkdtemp_mock.return_value = keyring_tmp_dir.absolute() if create_subdir: keyring_sub_dir = keyring_tmp_dir / "foo" keyring_sub_dir.mkdir() returned = sequoia.keyring_split( working_dir=tmp_dir, keyring=Path("foo"), preserve_filename=preserve_filename, ) if create_subdir: assert returned == [keyring_sub_dir] else: assert returned == [] @mark.parametrize( "output", [ None, Path("output"), ], ) @patch("libkeyringctl.sequoia.system") def test_keyring_merge(system_mock: Mock, output: Optional[Path]) -> None: certificates = [Path("foo"), Path("bar")] system_mock.return_value = "return" assert sequoia.keyring_merge(certificates=certificates, output=output) == "return" name, args, kwargs = system_mock.mock_calls[0] for cert in certificates: assert str(cert) in args[0] if output: assert "--output" in args[0] and str(output) in args[0] @patch("libkeyringctl.sequoia.system") @patch("libkeyringctl.sequoia.mkdtemp") def test_packet_split(mkdtemp_mock: Mock, system_mock: Mock) -> None: certificate = Path("certificate") with TemporaryDirectory() as tmp_dir_name: tmp_dir = Path(tmp_dir_name) keyring_tmp_dir = tmp_dir / "keyring" keyring_tmp_dir.mkdir() mkdtemp_mock.return_value = keyring_tmp_dir.absolute() keyring_sub_dir = keyring_tmp_dir / "foo" keyring_sub_dir.mkdir() assert sequoia.packet_split(working_dir=tmp_dir, certificate=certificate) == [keyring_sub_dir] name, args, kwargs = system_mock.mock_calls[0] assert str(certificate) == args[0][-1] @mark.parametrize("output, force", [(None, True), (None, False), (Path("output"), True), (Path("output"), False)]) @patch("libkeyringctl.sequoia.system") def test_packet_join(system_mock: Mock, output: Optional[Path], force: bool) -> None: packets = [Path("packet1"), Path("packet2")] system_return = "return" system_mock.return_value = system_return assert sequoia.packet_join(packets, output=output, force=force) == system_return name, args, kwargs = system_mock.mock_calls[0] for packet in packets: assert str(packet) in args[0] if force: assert "--force" == args[0][1] if output: assert "--output" in args[0] and str(output) in args[0] @mark.parametrize( "certifications_in_result, certifications, fingerprints", [ ("something: 0123456789123456789012345678901234567890\n", True, None), ("something: 0123456789123456789012345678901234567890\n", False, None), ( "something: 0123456789123456789012345678901234567890\n", True, {Fingerprint("0123456789123456789012345678901234567890"): Username("foo")}, ), ( "something: 0123456789123456789012345678901234567890\n", False, {Fingerprint("0123456789123456789012345678901234567890"): Username("foo")}, ), ( "something: 5678901234567890\n", True, {Fingerprint("0123456789123456789012345678901234567890"): Username("foo")}, ), ( "something: 5678901234567890\n", False, {Fingerprint("0123456789123456789012345678901234567890"): Username("foo")}, ), ], ) @patch("libkeyringctl.sequoia.system") def test_inspect( system_mock: Mock, certifications_in_result: str, certifications: bool, fingerprints: Optional[Dict[Fingerprint, Username]], ) -> None: packet = Path("packet") result_header = "result\n" if certifications: system_mock.return_value = result_header + "\n" + certifications_in_result else: system_mock.return_value = result_header returned = sequoia.inspect(packet=packet, certifications=certifications, fingerprints=fingerprints) if fingerprints and certifications: for fingerprint, username in fingerprints.items(): assert f"{fingerprint[24:]} {username}" in returned assert result_header in returned @patch("libkeyringctl.sequoia.system") def test_packet_dump(system_mock: Mock) -> None: system_mock.return_value = "return" assert sequoia.packet_dump(packet=Path("packet")) == "return" system_mock.called_once_with(["sq", "packet", "dump", "packet"]) @mark.parametrize( "packet_dump_return, field, expectation", [ ( "foo: bar", "foo", does_not_raise(), ), ( "foo: bar", "baz", raises(Exception), ), ], ) @patch("libkeyringctl.sequoia.packet_dump") def test_packet_dump_field( packet_dump_mock: Mock, packet_dump_return: str, field: str, expectation: ContextManager[str], ) -> None: packet_dump_mock.return_value = packet_dump_return with expectation: sequoia.packet_dump_field(packet=Path("packet"), field=field) @patch("libkeyringctl.sequoia.packet_dump_field") def test_packet_signature_creation_time(packet_dump_field_mock: Mock) -> None: creation_time = "2021-10-31 00:48:09 UTC" packet_dump_field_mock.return_value = creation_time assert sequoia.packet_signature_creation_time(packet=Path("packet")) == datetime.strptime( creation_time, "%Y-%m-%d %H:%M:%S %Z" ) @patch("libkeyringctl.sequoia.packet_signature_creation_time") def test_latest_certification(packet_signature_creation_time_mock: Mock) -> None: now = datetime.now(tz=timezone.utc) later = now + timedelta(days=1) early_cert = Path("cert1") later_cert = Path("cert2") packet_signature_creation_time_mock.side_effect = [now, later] assert sequoia.latest_certification(certifications=[early_cert, later_cert]) == later_cert packet_signature_creation_time_mock.side_effect = [later, now] assert sequoia.latest_certification(certifications=[later_cert, early_cert]) == later_cert @mark.parametrize("output", [(None), (Path("output"))]) @patch("libkeyringctl.sequoia.system") def test_key_extract_certificate(system_mock: Mock, output: Optional[Path]) -> None: system_mock.return_value = "return" assert sequoia.key_extract_certificate(key=Path("key"), output=output) == "return" name, args, kwargs = system_mock.mock_calls[0] if output: assert str(output) == args[0][-1] @mark.parametrize("output", [(None), (Path("output"))]) @patch("libkeyringctl.sequoia.system") def test_certify(system_mock: Mock, output: Optional[Path]) -> None: system_mock.return_value = "return" assert sequoia.certify(key=Path("key"), certificate=Path("cert"), uid=Uid("uid"), output=output) == "return" name, args, kwargs = system_mock.mock_calls[0] if output: assert str(output) == args[0][-1]