derivepassphrase.git

tests/test_derivepassphrase_ssh_agent.py   1) # SPDX-FileCopyrightText: 2024 Marco Ricci <software@the13thletter.info>

tests/test_ssh_agent_client.py             2) #
tests/test_ssh_agent_client.py             3) # SPDX-License-Identifier: MIT
tests/test_ssh_agent_client.py             4) 
tests/test_ssh_agent_client.py             5) """Test OpenSSH key loading and signing."""
tests/test_ssh_agent_client.py             6) 
tests/test_ssh_agent_client.py             7) from __future__ import annotations
tests/test_ssh_agent_client.py             8) 
tests/test_ssh_agent_client.py             9) import base64
tests/test_ssh_agent_client.py            10) import io
tests/test_ssh_agent_client.py            11) import socket

tests/test_ssh_agent_client.py            12) from typing import TYPE_CHECKING

tests/test_ssh_agent_client.py            13) 
tests/test_ssh_agent_client.py            14) import click
tests/test_ssh_agent_client.py            15) import click.testing

tests/test_derivepassphrase_ssh_agent.py  16) import hypothesis

tests/test_ssh_agent_client.py            17) import pytest

tests/test_derivepassphrase_ssh_agent.py  18) from hypothesis import strategies

tests/test_ssh_agent_client.py            19) 

tests/test_ssh_agent_client.py            20) import tests

tests/test_derivepassphrase_ssh_agent.py  21) from derivepassphrase import _types, cli, ssh_agent, vault

tests/test_ssh_agent_client.py            22) 

tests/test_ssh_agent_client.py            23) if TYPE_CHECKING:

tests/test_derivepassphrase_ssh_agent.py  24)     from collections.abc import Iterable, Iterator

tests/test_ssh_agent_client.py            25) 

tests/test_derivepassphrase_ssh_agent.py  26)     from typing_extensions import Any
tests/test_derivepassphrase_ssh_agent.py  27) 

tests/test_ssh_agent_client.py            28) 

tests/test_ssh_agent_client.py            29) class TestStaticFunctionality:

tests/test_ssh_agent_client.py            30)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py            31)         ['public_key', 'public_key_data'],
tests/test_ssh_agent_client.py            32)         [
tests/test_ssh_agent_client.py            33)             (val['public_key'], val['public_key_data'])
tests/test_ssh_agent_client.py            34)             for val in tests.SUPPORTED_KEYS.values()
tests/test_ssh_agent_client.py            35)         ],
tests/test_ssh_agent_client.py            36)     )

tests/test_ssh_agent_client.py            37)     def test_100_key_decoding(
tests/test_ssh_agent_client.py            38)         self, public_key: bytes, public_key_data: bytes
tests/test_ssh_agent_client.py            39)     ) -> None:

tests/test_ssh_agent_client.py            40)         keydata = base64.b64decode(public_key.split(None, 2)[1])
tests/test_ssh_agent_client.py            41)         assert (
tests/test_ssh_agent_client.py            42)             keydata == public_key_data
tests/test_ssh_agent_client.py            43)         ), "recorded public key data doesn't match"
tests/test_ssh_agent_client.py            44) 

tests/test_derivepassphrase_ssh_agent.py  45)     @pytest.mark.parametrize(
tests/test_derivepassphrase_ssh_agent.py  46)         ['line', 'env_name', 'value'],
tests/test_derivepassphrase_ssh_agent.py  47)         [
tests/test_derivepassphrase_ssh_agent.py  48)             (
tests/test_derivepassphrase_ssh_agent.py  49)                 'SSH_AUTH_SOCK=/tmp/pageant.user/pageant.27170; export SSH_AUTH_SOCK;',  # noqa: E501
tests/test_derivepassphrase_ssh_agent.py  50)                 'SSH_AUTH_SOCK',
tests/test_derivepassphrase_ssh_agent.py  51)                 '/tmp/pageant.user/pageant.27170',
tests/test_derivepassphrase_ssh_agent.py  52)             ),
tests/test_derivepassphrase_ssh_agent.py  53)             (
tests/test_derivepassphrase_ssh_agent.py  54)                 'SSH_AUTH_SOCK=/tmp/ssh-3CSTC1W5M22A/agent.27270; export SSH_AUTH_SOCK;',  # noqa: E501
tests/test_derivepassphrase_ssh_agent.py  55)                 'SSH_AUTH_SOCK',
tests/test_derivepassphrase_ssh_agent.py  56)                 '/tmp/ssh-3CSTC1W5M22A/agent.27270',
tests/test_derivepassphrase_ssh_agent.py  57)             ),
tests/test_derivepassphrase_ssh_agent.py  58)             (
tests/test_derivepassphrase_ssh_agent.py  59)                 'SSH_AUTH_SOCK=/tmp/pageant.user/pageant.27170; export SSH_AUTH_SOCK',  # noqa: E501
tests/test_derivepassphrase_ssh_agent.py  60)                 'SSH_AUTH_SOCK',
tests/test_derivepassphrase_ssh_agent.py  61)                 '/tmp/pageant.user/pageant.27170',
tests/test_derivepassphrase_ssh_agent.py  62)             ),
tests/test_derivepassphrase_ssh_agent.py  63)             (
tests/test_derivepassphrase_ssh_agent.py  64)                 'export SSH_AUTH_SOCK=/tmp/ssh-3CSTC1W5M22A/agent.27270;',
tests/test_derivepassphrase_ssh_agent.py  65)                 'SSH_AUTH_SOCK',
tests/test_derivepassphrase_ssh_agent.py  66)                 '/tmp/ssh-3CSTC1W5M22A/agent.27270',
tests/test_derivepassphrase_ssh_agent.py  67)             ),
tests/test_derivepassphrase_ssh_agent.py  68)             (
tests/test_derivepassphrase_ssh_agent.py  69)                 'export SSH_AUTH_SOCK=/tmp/pageant.user/pageant.27170',
tests/test_derivepassphrase_ssh_agent.py  70)                 'SSH_AUTH_SOCK',
tests/test_derivepassphrase_ssh_agent.py  71)                 '/tmp/pageant.user/pageant.27170',
tests/test_derivepassphrase_ssh_agent.py  72)             ),
tests/test_derivepassphrase_ssh_agent.py  73)             (
tests/test_derivepassphrase_ssh_agent.py  74)                 'SSH_AGENT_PID=27170; export SSH_AGENT_PID;',
tests/test_derivepassphrase_ssh_agent.py  75)                 'SSH_AGENT_PID',
tests/test_derivepassphrase_ssh_agent.py  76)                 '27170',
tests/test_derivepassphrase_ssh_agent.py  77)             ),
tests/test_derivepassphrase_ssh_agent.py  78)             (
tests/test_derivepassphrase_ssh_agent.py  79)                 'SSH_AGENT_PID=27170; export SSH_AGENT_PID',
tests/test_derivepassphrase_ssh_agent.py  80)                 'SSH_AGENT_PID',
tests/test_derivepassphrase_ssh_agent.py  81)                 '27170',
tests/test_derivepassphrase_ssh_agent.py  82)             ),
tests/test_derivepassphrase_ssh_agent.py  83)             ('export SSH_AGENT_PID=27170;', 'SSH_AGENT_PID', '27170'),
tests/test_derivepassphrase_ssh_agent.py  84)             ('export SSH_AGENT_PID=27170', 'SSH_AGENT_PID', '27170'),
tests/test_derivepassphrase_ssh_agent.py  85)             (
tests/test_derivepassphrase_ssh_agent.py  86)                 'export VARIABLE=value; export OTHER_VARIABLE=other_value;',
tests/test_derivepassphrase_ssh_agent.py  87)                 'VARIABLE',
tests/test_derivepassphrase_ssh_agent.py  88)                 None,
tests/test_derivepassphrase_ssh_agent.py  89)             ),
tests/test_derivepassphrase_ssh_agent.py  90)             (
tests/test_derivepassphrase_ssh_agent.py  91)                 'VARIABLE=value',
tests/test_derivepassphrase_ssh_agent.py  92)                 'VARIABLE',
tests/test_derivepassphrase_ssh_agent.py  93)                 None,
tests/test_derivepassphrase_ssh_agent.py  94)             ),
tests/test_derivepassphrase_ssh_agent.py  95)         ],
tests/test_derivepassphrase_ssh_agent.py  96)     )
tests/test_derivepassphrase_ssh_agent.py  97)     def test_190_sh_export_line_parsing(
tests/test_derivepassphrase_ssh_agent.py  98)         self, line: str, env_name: str, value: str | None
tests/test_derivepassphrase_ssh_agent.py  99)     ) -> None:
tests/test_derivepassphrase_ssh_agent.py 100)         if value is not None:
tests/test_derivepassphrase_ssh_agent.py 101)             assert tests.parse_sh_export_line(line, env_name=env_name) == value
tests/test_derivepassphrase_ssh_agent.py 102)         else:
tests/test_derivepassphrase_ssh_agent.py 103)             with pytest.raises(ValueError, match='Cannot parse sh line:'):
tests/test_derivepassphrase_ssh_agent.py 104)                 tests.parse_sh_export_line(line, env_name=env_name)
tests/test_derivepassphrase_ssh_agent.py 105) 

tests/test_derivepassphrase_ssh_agent.py 106)     def test_200_constructor_no_running_agent(
tests/test_derivepassphrase_ssh_agent.py 107)         self, monkeypatch: pytest.MonkeyPatch
tests/test_derivepassphrase_ssh_agent.py 108)     ) -> None:

tests/test_ssh_agent_client.py           109)         monkeypatch.delenv('SSH_AUTH_SOCK', raising=False)
tests/test_ssh_agent_client.py           110)         sock = socket.socket(family=socket.AF_UNIX)

tests/test_ssh_agent_client.py           111)         with pytest.raises(
tests/test_ssh_agent_client.py           112)             KeyError, match='SSH_AUTH_SOCK environment variable'
tests/test_ssh_agent_client.py           113)         ):

tests/test_derivepassphrase_ssh_agent.py 114)             ssh_agent.SSHAgentClient(socket=sock)

tests/test_ssh_agent_client.py           115) 

tests/test_ssh_agent_client.py           116)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           117)         ['input', 'expected'],
tests/test_ssh_agent_client.py           118)         [
tests/test_ssh_agent_client.py           119)             (16777216, b'\x01\x00\x00\x00'),
tests/test_ssh_agent_client.py           120)         ],
tests/test_ssh_agent_client.py           121)     )

tests/test_ssh_agent_client.py           122)     def test_210_uint32(self, input: int, expected: bytes | bytearray) -> None:

tests/test_derivepassphrase_ssh_agent.py 123)         uint32 = ssh_agent.SSHAgentClient.uint32

tests/test_ssh_agent_client.py           124)         assert uint32(input) == expected
tests/test_ssh_agent_client.py           125) 

tests/test_ssh_agent_client.py           126)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           127)         ['input', 'expected'],
tests/test_ssh_agent_client.py           128)         [
tests/test_ssh_agent_client.py           129)             (b'ssh-rsa', b'\x00\x00\x00\x07ssh-rsa'),
tests/test_ssh_agent_client.py           130)             (b'ssh-ed25519', b'\x00\x00\x00\x0bssh-ed25519'),
tests/test_ssh_agent_client.py           131)             (

tests/test_derivepassphrase_ssh_agent.py 132)                 ssh_agent.SSHAgentClient.string(b'ssh-ed25519'),

tests/test_ssh_agent_client.py           133)                 b'\x00\x00\x00\x0f\x00\x00\x00\x0bssh-ed25519',
tests/test_ssh_agent_client.py           134)             ),
tests/test_ssh_agent_client.py           135)         ],
tests/test_ssh_agent_client.py           136)     )

tests/test_ssh_agent_client.py           137)     def test_211_string(
tests/test_ssh_agent_client.py           138)         self, input: bytes | bytearray, expected: bytes | bytearray
tests/test_ssh_agent_client.py           139)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 140)         string = ssh_agent.SSHAgentClient.string

tests/test_ssh_agent_client.py           141)         assert bytes(string(input)) == expected
tests/test_ssh_agent_client.py           142) 

tests/test_ssh_agent_client.py           143)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           144)         ['input', 'expected'],
tests/test_ssh_agent_client.py           145)         [
tests/test_ssh_agent_client.py           146)             (b'\x00\x00\x00\x07ssh-rsa', b'ssh-rsa'),
tests/test_ssh_agent_client.py           147)             (

tests/test_derivepassphrase_ssh_agent.py 148)                 ssh_agent.SSHAgentClient.string(b'ssh-ed25519'),

tests/test_ssh_agent_client.py           149)                 b'ssh-ed25519',
tests/test_ssh_agent_client.py           150)             ),
tests/test_ssh_agent_client.py           151)         ],
tests/test_ssh_agent_client.py           152)     )

tests/test_ssh_agent_client.py           153)     def test_212_unstring(
tests/test_ssh_agent_client.py           154)         self, input: bytes | bytearray, expected: bytes | bytearray
tests/test_ssh_agent_client.py           155)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 156)         unstring = ssh_agent.SSHAgentClient.unstring
tests/test_derivepassphrase_ssh_agent.py 157)         unstring_prefix = ssh_agent.SSHAgentClient.unstring_prefix

tests/test_ssh_agent_client.py           158)         assert bytes(unstring(input)) == expected

tests/test_ssh_agent_client.py           159)         assert tuple(bytes(x) for x in unstring_prefix(input)) == (
tests/test_ssh_agent_client.py           160)             expected,
tests/test_ssh_agent_client.py           161)             b'',
tests/test_ssh_agent_client.py           162)         )

tests/test_ssh_agent_client.py           163) 

tests/test_ssh_agent_client.py           164)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           165)         ['value', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           166)         [
tests/test_ssh_agent_client.py           167)             (10000000000000000, OverflowError, 'int too big to convert'),
tests/test_ssh_agent_client.py           168)             (-1, OverflowError, "can't convert negative int to unsigned"),
tests/test_ssh_agent_client.py           169)         ],
tests/test_ssh_agent_client.py           170)     )

tests/test_ssh_agent_client.py           171)     def test_310_uint32_exceptions(
tests/test_ssh_agent_client.py           172)         self, value: int, exc_type: type[Exception], exc_pattern: str
tests/test_ssh_agent_client.py           173)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 174)         uint32 = ssh_agent.SSHAgentClient.uint32

tests/test_ssh_agent_client.py           175)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           176)             uint32(value)
tests/test_ssh_agent_client.py           177) 

tests/test_ssh_agent_client.py           178)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           179)         ['input', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           180)         [
tests/test_ssh_agent_client.py           181)             ('some string', TypeError, 'invalid payload type'),
tests/test_ssh_agent_client.py           182)         ],
tests/test_ssh_agent_client.py           183)     )

tests/test_ssh_agent_client.py           184)     def test_311_string_exceptions(
tests/test_ssh_agent_client.py           185)         self, input: Any, exc_type: type[Exception], exc_pattern: str
tests/test_ssh_agent_client.py           186)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 187)         string = ssh_agent.SSHAgentClient.string

tests/test_ssh_agent_client.py           188)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           189)             string(input)
tests/test_ssh_agent_client.py           190) 
tests/test_ssh_agent_client.py           191)     @pytest.mark.parametrize(

tests/test_ssh_agent_client.py           192)         ['input', 'exc_type', 'exc_pattern', 'has_trailer', 'parts'],
tests/test_ssh_agent_client.py           193)         [

tests/test_ssh_agent_client.py           194)             (b'ssh', ValueError, 'malformed SSH byte string', False, None),
tests/test_ssh_agent_client.py           195)             (
tests/test_ssh_agent_client.py           196)                 b'\x00\x00\x00\x08ssh-rsa',

tests/test_ssh_agent_client.py           197)                 ValueError,
tests/test_ssh_agent_client.py           198)                 'malformed SSH byte string',
tests/test_ssh_agent_client.py           199)                 False,
tests/test_ssh_agent_client.py           200)                 None,

tests/test_ssh_agent_client.py           201)             ),
tests/test_ssh_agent_client.py           202)             (
tests/test_ssh_agent_client.py           203)                 b'\x00\x00\x00\x04XXX trailing text',

tests/test_ssh_agent_client.py           204)                 ValueError,
tests/test_ssh_agent_client.py           205)                 'malformed SSH byte string',
tests/test_ssh_agent_client.py           206)                 True,
tests/test_ssh_agent_client.py           207)                 (b'XXX ', b'trailing text'),

tests/test_ssh_agent_client.py           208)             ),

tests/test_ssh_agent_client.py           209)         ],
tests/test_ssh_agent_client.py           210)     )
tests/test_ssh_agent_client.py           211)     def test_312_unstring_exceptions(

tests/test_ssh_agent_client.py           212)         self,
tests/test_ssh_agent_client.py           213)         input: bytes | bytearray,
tests/test_ssh_agent_client.py           214)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           215)         exc_pattern: str,
tests/test_ssh_agent_client.py           216)         has_trailer: bool,
tests/test_ssh_agent_client.py           217)         parts: tuple[bytes | bytearray, bytes | bytearray] | None,
tests/test_ssh_agent_client.py           218)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 219)         unstring = ssh_agent.SSHAgentClient.unstring
tests/test_derivepassphrase_ssh_agent.py 220)         unstring_prefix = ssh_agent.SSHAgentClient.unstring_prefix

tests/test_ssh_agent_client.py           221)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           222)             unstring(input)
tests/test_ssh_agent_client.py           223)         if has_trailer:
tests/test_ssh_agent_client.py           224)             assert tuple(bytes(x) for x in unstring_prefix(input)) == parts
tests/test_ssh_agent_client.py           225)         else:
tests/test_ssh_agent_client.py           226)             with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           227)                 unstring_prefix(input)
tests/test_ssh_agent_client.py           228) 

tests/test_ssh_agent_client.py           229) 

tests/test_ssh_agent_client.py           230) class TestAgentInteraction:

tests/test_ssh_agent_client.py           231)     @pytest.mark.parametrize(

tests/test_derivepassphrase_ssh_agent.py 232)         'data_dict',
tests/test_derivepassphrase_ssh_agent.py 233)         list(tests.SUPPORTED_KEYS.values()),
tests/test_derivepassphrase_ssh_agent.py 234)         ids=tests.SUPPORTED_KEYS.keys(),

tests/test_ssh_agent_client.py           235)     )

tests/test_ssh_agent_client.py           236)     def test_200_sign_data_via_agent(

tests/test_derivepassphrase_ssh_agent.py 237)         self,
tests/test_derivepassphrase_ssh_agent.py 238)         ssh_agent_client_with_test_keys_loaded: ssh_agent.SSHAgentClient,
tests/test_derivepassphrase_ssh_agent.py 239)         data_dict: tests.SSHTestKey,

tests/test_ssh_agent_client.py           240)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 241)         client = ssh_agent_client_with_test_keys_loaded
tests/test_derivepassphrase_ssh_agent.py 242)         key_comment_pairs = {bytes(k): bytes(c) for k, c in client.list_keys()}
tests/test_derivepassphrase_ssh_agent.py 243)         public_key_data = data_dict['public_key_data']
tests/test_derivepassphrase_ssh_agent.py 244)         expected_signature = data_dict['expected_signature']
tests/test_derivepassphrase_ssh_agent.py 245)         derived_passphrase = data_dict['derived_passphrase']
tests/test_derivepassphrase_ssh_agent.py 246)         if public_key_data not in key_comment_pairs:  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 247)             pytest.skip('prerequisite SSH key not loaded')
tests/test_derivepassphrase_ssh_agent.py 248)         signature = bytes(
tests/test_derivepassphrase_ssh_agent.py 249)             client.sign(payload=vault.Vault._UUID, key=public_key_data)
tests/test_derivepassphrase_ssh_agent.py 250)         )
tests/test_derivepassphrase_ssh_agent.py 251)         assert signature == expected_signature, 'SSH signature mismatch'
tests/test_derivepassphrase_ssh_agent.py 252)         signature2 = bytes(
tests/test_derivepassphrase_ssh_agent.py 253)             client.sign(payload=vault.Vault._UUID, key=public_key_data)
tests/test_derivepassphrase_ssh_agent.py 254)         )
tests/test_derivepassphrase_ssh_agent.py 255)         assert signature2 == expected_signature, 'SSH signature mismatch'
tests/test_derivepassphrase_ssh_agent.py 256)         assert (
tests/test_derivepassphrase_ssh_agent.py 257)             vault.Vault.phrase_from_key(public_key_data) == derived_passphrase
tests/test_derivepassphrase_ssh_agent.py 258)         ), 'SSH signature mismatch'

tests/test_ssh_agent_client.py           259) 

tests/test_ssh_agent_client.py           260)     @pytest.mark.parametrize(

tests/test_derivepassphrase_ssh_agent.py 261)         'data_dict',
tests/test_derivepassphrase_ssh_agent.py 262)         list(tests.UNSUITABLE_KEYS.values()),
tests/test_derivepassphrase_ssh_agent.py 263)         ids=tests.UNSUITABLE_KEYS.keys(),

tests/test_ssh_agent_client.py           264)     )

tests/test_ssh_agent_client.py           265)     def test_201_sign_data_via_agent_unsupported(

tests/test_derivepassphrase_ssh_agent.py 266)         self,
tests/test_derivepassphrase_ssh_agent.py 267)         ssh_agent_client_with_test_keys_loaded: ssh_agent.SSHAgentClient,
tests/test_derivepassphrase_ssh_agent.py 268)         data_dict: tests.SSHTestKey,

tests/test_ssh_agent_client.py           269)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 270)         client = ssh_agent_client_with_test_keys_loaded
tests/test_derivepassphrase_ssh_agent.py 271)         key_comment_pairs = {bytes(k): bytes(c) for k, c in client.list_keys()}
tests/test_derivepassphrase_ssh_agent.py 272)         public_key_data = data_dict['public_key_data']
tests/test_derivepassphrase_ssh_agent.py 273)         _ = data_dict['expected_signature']
tests/test_derivepassphrase_ssh_agent.py 274)         if public_key_data not in key_comment_pairs:  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 275)             pytest.skip('prerequisite SSH key not loaded')
tests/test_derivepassphrase_ssh_agent.py 276)         with pytest.raises(ValueError, match='unsuitable SSH key'):
tests/test_derivepassphrase_ssh_agent.py 277)             vault.Vault.phrase_from_key(public_key_data)

tests/test_ssh_agent_client.py           278) 
tests/test_ssh_agent_client.py           279)     @staticmethod

tests/test_ssh_agent_client.py           280)     def _params() -> Iterator[tuple[bytes, bool]]:

tests/test_ssh_agent_client.py           281)         for value in tests.SUPPORTED_KEYS.values():
tests/test_ssh_agent_client.py           282)             key = value['public_key_data']
tests/test_ssh_agent_client.py           283)             yield (key, False)
tests/test_ssh_agent_client.py           284)         singleton_key = tests.list_keys_singleton()[0].key
tests/test_ssh_agent_client.py           285)         for value in tests.SUPPORTED_KEYS.values():
tests/test_ssh_agent_client.py           286)             key = value['public_key_data']
tests/test_ssh_agent_client.py           287)             if key == singleton_key:
tests/test_ssh_agent_client.py           288)                 yield (key, True)
tests/test_ssh_agent_client.py           289) 
tests/test_ssh_agent_client.py           290)     @pytest.mark.parametrize(['key', 'single'], list(_params()))

tests/test_ssh_agent_client.py           291)     def test_210_ssh_key_selector(

tests/test_derivepassphrase_ssh_agent.py 292)         self,
tests/test_derivepassphrase_ssh_agent.py 293)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 294)         running_ssh_agent: str,
tests/test_derivepassphrase_ssh_agent.py 295)         key: bytes,
tests/test_derivepassphrase_ssh_agent.py 296)         single: bool,

tests/test_ssh_agent_client.py           297)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 298)         del running_ssh_agent
tests/test_derivepassphrase_ssh_agent.py 299) 

tests/test_ssh_agent_client.py           300)         def key_is_suitable(key: bytes) -> bool:

tests/test_ssh_agent_client.py           301)             return key in {
tests/test_ssh_agent_client.py           302)                 v['public_key_data'] for v in tests.SUPPORTED_KEYS.values()
tests/test_ssh_agent_client.py           303)             }
tests/test_ssh_agent_client.py           304) 

tests/test_ssh_agent_client.py           305)         if single:

tests/test_ssh_agent_client.py           306)             monkeypatch.setattr(

tests/test_derivepassphrase_ssh_agent.py 307)                 ssh_agent.SSHAgentClient,

tests/test_ssh_agent_client.py           308)                 'list_keys',
tests/test_ssh_agent_client.py           309)                 tests.list_keys_singleton,
tests/test_ssh_agent_client.py           310)             )
tests/test_ssh_agent_client.py           311)             keys = [
tests/test_ssh_agent_client.py           312)                 pair.key
tests/test_ssh_agent_client.py           313)                 for pair in tests.list_keys_singleton()
tests/test_ssh_agent_client.py           314)                 if key_is_suitable(pair.key)
tests/test_ssh_agent_client.py           315)             ]

tests/test_ssh_agent_client.py           316)             index = '1'

tests/test_ssh_agent_client.py           317)             text = 'Use this key? yes\n'

tests/test_ssh_agent_client.py           318)         else:

tests/test_ssh_agent_client.py           319)             monkeypatch.setattr(

tests/test_derivepassphrase_ssh_agent.py 320)                 ssh_agent.SSHAgentClient, 'list_keys', tests.list_keys

tests/test_ssh_agent_client.py           321)             )
tests/test_ssh_agent_client.py           322)             keys = [
tests/test_ssh_agent_client.py           323)                 pair.key
tests/test_ssh_agent_client.py           324)                 for pair in tests.list_keys()
tests/test_ssh_agent_client.py           325)                 if key_is_suitable(pair.key)
tests/test_ssh_agent_client.py           326)             ]

tests/test_ssh_agent_client.py           327)             index = str(1 + keys.index(key))
tests/test_ssh_agent_client.py           328)             n = len(keys)
tests/test_ssh_agent_client.py           329)             text = f'Your selection? (1-{n}, leave empty to abort): {index}\n'
tests/test_ssh_agent_client.py           330)         b64_key = base64.standard_b64encode(key).decode('ASCII')
tests/test_ssh_agent_client.py           331) 
tests/test_ssh_agent_client.py           332)         @click.command()

tests/test_ssh_agent_client.py           333)         def driver() -> None:

tests/test_derivepassphrase_ssh_agent.py 334)             key = cli._select_ssh_key()

tests/test_ssh_agent_client.py           335)             click.echo(base64.standard_b64encode(key).decode('ASCII'))
tests/test_ssh_agent_client.py           336) 
tests/test_ssh_agent_client.py           337)         runner = click.testing.CliRunner(mix_stderr=True)

tests/test_derivepassphrase_ssh_agent.py 338)         _result = runner.invoke(

tests/test_ssh_agent_client.py           339)             driver,
tests/test_ssh_agent_client.py           340)             [],
tests/test_ssh_agent_client.py           341)             input=('yes\n' if single else f'{index}\n'),
tests/test_ssh_agent_client.py           342)             catch_exceptions=True,

tests/test_ssh_agent_client.py           343)         )

tests/test_derivepassphrase_ssh_agent.py 344)         result = tests.ReadableResult.parse(_result)
tests/test_derivepassphrase_ssh_agent.py 345)         for snippet in ('Suitable SSH keys:\n', text, f'\n{b64_key}\n'):
tests/test_derivepassphrase_ssh_agent.py 346)             assert result.clean_exit(output=snippet), 'expected clean exit'

tests/test_ssh_agent_client.py           347) 
tests/test_ssh_agent_client.py           348)     del _params
tests/test_ssh_agent_client.py           349) 

tests/test_derivepassphrase_ssh_agent.py 350)     def test_300_constructor_bad_running_agent(

tests/test_derivepassphrase_ssh_agent.py 351)         self,
tests/test_derivepassphrase_ssh_agent.py 352)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 353)         running_ssh_agent: str,

tests/test_derivepassphrase_ssh_agent.py 354)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 355)         with monkeypatch.context() as monkeypatch2:
tests/test_derivepassphrase_ssh_agent.py 356)             monkeypatch2.setenv('SSH_AUTH_SOCK', running_ssh_agent + '~')
tests/test_derivepassphrase_ssh_agent.py 357)             sock = socket.socket(family=socket.AF_UNIX)
tests/test_derivepassphrase_ssh_agent.py 358)             with pytest.raises(OSError):  # noqa: PT011
tests/test_derivepassphrase_ssh_agent.py 359)                 ssh_agent.SSHAgentClient(socket=sock)

tests/test_ssh_agent_client.py           360) 

tests/test_ssh_agent_client.py           361)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           362)         'response',
tests/test_ssh_agent_client.py           363)         [
tests/test_ssh_agent_client.py           364)             b'\x00\x00',
tests/test_ssh_agent_client.py           365)             b'\x00\x00\x00\x1f some bytes missing',
tests/test_ssh_agent_client.py           366)         ],
tests/test_ssh_agent_client.py           367)     )

tests/test_ssh_agent_client.py           368)     def test_310_truncated_server_response(

tests/test_derivepassphrase_ssh_agent.py 369)         self,
tests/test_derivepassphrase_ssh_agent.py 370)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 371)         running_ssh_agent: str,
tests/test_derivepassphrase_ssh_agent.py 372)         response: bytes,

tests/test_ssh_agent_client.py           373)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 374)         del running_ssh_agent

tests/test_derivepassphrase_ssh_agent.py 375)         client = ssh_agent.SSHAgentClient()

tests/test_ssh_agent_client.py           376)         response_stream = io.BytesIO(response)

tests/test_ssh_agent_client.py           377) 

tests/test_ssh_agent_client.py           378)         class PseudoSocket:
tests/test_ssh_agent_client.py           379)             def sendall(self, *args: Any, **kwargs: Any) -> Any:  # noqa: ARG002

tests/test_ssh_agent_client.py           380)                 return None

tests/test_ssh_agent_client.py           381) 

tests/test_ssh_agent_client.py           382)             def recv(self, *args: Any, **kwargs: Any) -> Any:
tests/test_ssh_agent_client.py           383)                 return response_stream.read(*args, **kwargs)

tests/test_ssh_agent_client.py           384) 

tests/test_ssh_agent_client.py           385)         pseudo_socket = PseudoSocket()
tests/test_ssh_agent_client.py           386)         monkeypatch.setattr(client, '_connection', pseudo_socket)
tests/test_ssh_agent_client.py           387)         with pytest.raises(EOFError):
tests/test_ssh_agent_client.py           388)             client.request(255, b'')
tests/test_ssh_agent_client.py           389) 
tests/test_ssh_agent_client.py           390)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           391)         ['response_code', 'response', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           392)         [

tests/test_ssh_agent_client.py           393)             (

tests/test_derivepassphrase_ssh_agent.py 394)                 _types.SSH_AGENT.FAILURE,
tests/test_derivepassphrase_ssh_agent.py 395)                 b'',
tests/test_derivepassphrase_ssh_agent.py 396)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 397)                 'failed to complete the request',
tests/test_derivepassphrase_ssh_agent.py 398)             ),
tests/test_derivepassphrase_ssh_agent.py 399)             (
tests/test_derivepassphrase_ssh_agent.py 400)                 _types.SSH_AGENT.IDENTITIES_ANSWER,
tests/test_derivepassphrase_ssh_agent.py 401)                 b'\x00\x00\x00\x01',
tests/test_derivepassphrase_ssh_agent.py 402)                 EOFError,
tests/test_derivepassphrase_ssh_agent.py 403)                 'truncated response',
tests/test_derivepassphrase_ssh_agent.py 404)             ),
tests/test_derivepassphrase_ssh_agent.py 405)             (
tests/test_derivepassphrase_ssh_agent.py 406)                 _types.SSH_AGENT.IDENTITIES_ANSWER,

tests/test_ssh_agent_client.py           407)                 b'\x00\x00\x00\x00abc',

tests/test_derivepassphrase_ssh_agent.py 408)                 ssh_agent.TrailingDataError,

tests/test_ssh_agent_client.py           409)                 'Overlong response',

tests/test_ssh_agent_client.py           410)             ),

tests/test_ssh_agent_client.py           411)         ],

tests/test_ssh_agent_client.py           412)     )

tests/test_ssh_agent_client.py           413)     def test_320_list_keys_error_responses(

tests/test_ssh_agent_client.py           414)         self,

tests/test_derivepassphrase_ssh_agent.py 415)         monkeypatch: pytest.MonkeyPatch,

tests/test_derivepassphrase_ssh_agent.py 416)         running_ssh_agent: str,

tests/test_derivepassphrase_ssh_agent.py 417)         response_code: _types.SSH_AGENT,

tests/test_ssh_agent_client.py           418)         response: bytes | bytearray,
tests/test_ssh_agent_client.py           419)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           420)         exc_pattern: str,
tests/test_ssh_agent_client.py           421)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 422)         del running_ssh_agent
tests/test_derivepassphrase_ssh_agent.py 423) 

tests/test_derivepassphrase_ssh_agent.py 424)         passed_response_code = response_code
tests/test_derivepassphrase_ssh_agent.py 425) 
tests/test_derivepassphrase_ssh_agent.py 426)         def request(
tests/test_derivepassphrase_ssh_agent.py 427)             request_code: int | _types.SSH_AGENTC,
tests/test_derivepassphrase_ssh_agent.py 428)             payload: bytes | bytearray,
tests/test_derivepassphrase_ssh_agent.py 429)             /,
tests/test_derivepassphrase_ssh_agent.py 430)             *,
tests/test_derivepassphrase_ssh_agent.py 431)             response_code: Iterable[int | _types.SSH_AGENT]
tests/test_derivepassphrase_ssh_agent.py 432)             | int
tests/test_derivepassphrase_ssh_agent.py 433)             | _types.SSH_AGENT
tests/test_derivepassphrase_ssh_agent.py 434)             | None = None,
tests/test_derivepassphrase_ssh_agent.py 435)         ) -> tuple[int, bytes | bytearray] | bytes | bytearray:
tests/test_derivepassphrase_ssh_agent.py 436)             del request_code
tests/test_derivepassphrase_ssh_agent.py 437)             del payload
tests/test_derivepassphrase_ssh_agent.py 438)             if isinstance(  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 439)                 response_code, int | _types.SSH_AGENT
tests/test_derivepassphrase_ssh_agent.py 440)             ):
tests/test_derivepassphrase_ssh_agent.py 441)                 response_code = frozenset({response_code})
tests/test_derivepassphrase_ssh_agent.py 442)             if response_code is not None:  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 443)                 response_code = frozenset({
tests/test_derivepassphrase_ssh_agent.py 444)                     c if isinstance(c, int) else c.value for c in response_code
tests/test_derivepassphrase_ssh_agent.py 445)                 })
tests/test_derivepassphrase_ssh_agent.py 446) 
tests/test_derivepassphrase_ssh_agent.py 447)             if not response_code:  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 448)                 return (passed_response_code.value, response)
tests/test_derivepassphrase_ssh_agent.py 449)             if passed_response_code.value not in response_code:
tests/test_derivepassphrase_ssh_agent.py 450)                 raise ssh_agent.SSHAgentFailedError(
tests/test_derivepassphrase_ssh_agent.py 451)                     passed_response_code.value, response
tests/test_derivepassphrase_ssh_agent.py 452)                 )
tests/test_derivepassphrase_ssh_agent.py 453)             return response
tests/test_derivepassphrase_ssh_agent.py 454) 

tests/test_derivepassphrase_ssh_agent.py 455)         with monkeypatch.context() as monkeypatch2:
tests/test_derivepassphrase_ssh_agent.py 456)             client = ssh_agent.SSHAgentClient()
tests/test_derivepassphrase_ssh_agent.py 457)             monkeypatch2.setattr(client, 'request', request)
tests/test_derivepassphrase_ssh_agent.py 458)             with pytest.raises(exc_type, match=exc_pattern):
tests/test_derivepassphrase_ssh_agent.py 459)                 client.list_keys()

tests/test_ssh_agent_client.py           460) 
tests/test_ssh_agent_client.py           461)     @pytest.mark.parametrize(

tests/test_derivepassphrase_ssh_agent.py 462)         [
tests/test_derivepassphrase_ssh_agent.py 463)             'key',
tests/test_derivepassphrase_ssh_agent.py 464)             'check',
tests/test_derivepassphrase_ssh_agent.py 465)             'response_code',
tests/test_derivepassphrase_ssh_agent.py 466)             'response',
tests/test_derivepassphrase_ssh_agent.py 467)             'exc_type',
tests/test_derivepassphrase_ssh_agent.py 468)             'exc_pattern',
tests/test_derivepassphrase_ssh_agent.py 469)         ],

tests/test_ssh_agent_client.py           470)         [
tests/test_ssh_agent_client.py           471)             (
tests/test_ssh_agent_client.py           472)                 b'invalid-key',
tests/test_ssh_agent_client.py           473)                 True,

tests/test_derivepassphrase_ssh_agent.py 474)                 _types.SSH_AGENT.FAILURE,
tests/test_derivepassphrase_ssh_agent.py 475)                 b'',

tests/test_ssh_agent_client.py           476)                 KeyError,
tests/test_ssh_agent_client.py           477)                 'target SSH key not loaded into agent',
tests/test_ssh_agent_client.py           478)             ),
tests/test_ssh_agent_client.py           479)             (
tests/test_ssh_agent_client.py           480)                 tests.SUPPORTED_KEYS['ed25519']['public_key_data'],
tests/test_ssh_agent_client.py           481)                 True,

tests/test_derivepassphrase_ssh_agent.py 482)                 _types.SSH_AGENT.FAILURE,
tests/test_derivepassphrase_ssh_agent.py 483)                 b'',

tests/test_derivepassphrase_ssh_agent.py 484)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 485)                 'failed to complete the request',

tests/test_ssh_agent_client.py           486)             ),
tests/test_ssh_agent_client.py           487)         ],

tests/test_ssh_agent_client.py           488)     )

tests/test_ssh_agent_client.py           489)     def test_330_sign_error_responses(

tests/test_ssh_agent_client.py           490)         self,

tests/test_derivepassphrase_ssh_agent.py 491)         monkeypatch: pytest.MonkeyPatch,

tests/test_derivepassphrase_ssh_agent.py 492)         running_ssh_agent: str,

tests/test_ssh_agent_client.py           493)         key: bytes | bytearray,
tests/test_ssh_agent_client.py           494)         check: bool,

tests/test_derivepassphrase_ssh_agent.py 495)         response_code: _types.SSH_AGENT,
tests/test_derivepassphrase_ssh_agent.py 496)         response: bytes | bytearray,

tests/test_ssh_agent_client.py           497)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           498)         exc_pattern: str,
tests/test_ssh_agent_client.py           499)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 500)         del running_ssh_agent

tests/test_derivepassphrase_ssh_agent.py 501)         passed_response_code = response_code
tests/test_derivepassphrase_ssh_agent.py 502) 
tests/test_derivepassphrase_ssh_agent.py 503)         def request(
tests/test_derivepassphrase_ssh_agent.py 504)             request_code: int | _types.SSH_AGENTC,
tests/test_derivepassphrase_ssh_agent.py 505)             payload: bytes | bytearray,
tests/test_derivepassphrase_ssh_agent.py 506)             /,
tests/test_derivepassphrase_ssh_agent.py 507)             *,
tests/test_derivepassphrase_ssh_agent.py 508)             response_code: Iterable[int | _types.SSH_AGENT]
tests/test_derivepassphrase_ssh_agent.py 509)             | int
tests/test_derivepassphrase_ssh_agent.py 510)             | _types.SSH_AGENT
tests/test_derivepassphrase_ssh_agent.py 511)             | None = None,
tests/test_derivepassphrase_ssh_agent.py 512)         ) -> tuple[int, bytes | bytearray] | bytes | bytearray:
tests/test_derivepassphrase_ssh_agent.py 513)             del request_code
tests/test_derivepassphrase_ssh_agent.py 514)             del payload
tests/test_derivepassphrase_ssh_agent.py 515)             if isinstance(  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 516)                 response_code, int | _types.SSH_AGENT
tests/test_derivepassphrase_ssh_agent.py 517)             ):
tests/test_derivepassphrase_ssh_agent.py 518)                 response_code = frozenset({response_code})
tests/test_derivepassphrase_ssh_agent.py 519)             if response_code is not None:  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 520)                 response_code = frozenset({
tests/test_derivepassphrase_ssh_agent.py 521)                     c if isinstance(c, int) else c.value for c in response_code
tests/test_derivepassphrase_ssh_agent.py 522)                 })
tests/test_derivepassphrase_ssh_agent.py 523) 
tests/test_derivepassphrase_ssh_agent.py 524)             if not response_code:  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 525)                 return (passed_response_code.value, response)
tests/test_derivepassphrase_ssh_agent.py 526)             if (
tests/test_derivepassphrase_ssh_agent.py 527)                 passed_response_code.value not in response_code
tests/test_derivepassphrase_ssh_agent.py 528)             ):  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 529)                 raise ssh_agent.SSHAgentFailedError(
tests/test_derivepassphrase_ssh_agent.py 530)                     passed_response_code.value, response
tests/test_derivepassphrase_ssh_agent.py 531)                 )
tests/test_derivepassphrase_ssh_agent.py 532)             return response  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 533) 

tests/test_derivepassphrase_ssh_agent.py 534)         with monkeypatch.context() as monkeypatch2:
tests/test_derivepassphrase_ssh_agent.py 535)             client = ssh_agent.SSHAgentClient()
tests/test_derivepassphrase_ssh_agent.py 536)             monkeypatch2.setattr(client, 'request', request)
tests/test_derivepassphrase_ssh_agent.py 537)             KeyCommentPair = _types.KeyCommentPair  # noqa: N806
tests/test_derivepassphrase_ssh_agent.py 538)             loaded_keys = [
tests/test_derivepassphrase_ssh_agent.py 539)                 KeyCommentPair(v['public_key_data'], b'no comment')
tests/test_derivepassphrase_ssh_agent.py 540)                 for v in tests.SUPPORTED_KEYS.values()
tests/test_derivepassphrase_ssh_agent.py 541)             ]
tests/test_derivepassphrase_ssh_agent.py 542)             monkeypatch2.setattr(client, 'list_keys', lambda: loaded_keys)
tests/test_derivepassphrase_ssh_agent.py 543)             with pytest.raises(exc_type, match=exc_pattern):
tests/test_derivepassphrase_ssh_agent.py 544)                 client.sign(key, b'abc', check_if_key_loaded=check)

tests/test_derivepassphrase_ssh_agent.py 545) 
tests/test_derivepassphrase_ssh_agent.py 546)     @pytest.mark.parametrize(
tests/test_derivepassphrase_ssh_agent.py 547)         ['request_code', 'response_code', 'exc_type', 'exc_pattern'],
tests/test_derivepassphrase_ssh_agent.py 548)         [
tests/test_derivepassphrase_ssh_agent.py 549)             (
tests/test_derivepassphrase_ssh_agent.py 550)                 _types.SSH_AGENTC.REQUEST_IDENTITIES,
tests/test_derivepassphrase_ssh_agent.py 551)                 _types.SSH_AGENT.SUCCESS,
tests/test_derivepassphrase_ssh_agent.py 552)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 553)                 f'[Code {_types.SSH_AGENT.IDENTITIES_ANSWER.value}]',
tests/test_derivepassphrase_ssh_agent.py 554)             ),
tests/test_derivepassphrase_ssh_agent.py 555)         ],
tests/test_derivepassphrase_ssh_agent.py 556)     )
tests/test_derivepassphrase_ssh_agent.py 557)     def test_340_request_error_responses(
tests/test_derivepassphrase_ssh_agent.py 558)         self,

tests/test_derivepassphrase_ssh_agent.py 559)         running_ssh_agent: str,

tests/test_derivepassphrase_ssh_agent.py 560)         request_code: _types.SSH_AGENTC,
tests/test_derivepassphrase_ssh_agent.py 561)         response_code: _types.SSH_AGENT,
tests/test_derivepassphrase_ssh_agent.py 562)         exc_type: type[Exception],
tests/test_derivepassphrase_ssh_agent.py 563)         exc_pattern: str,
tests/test_derivepassphrase_ssh_agent.py 564)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 565)         del running_ssh_agent
tests/test_derivepassphrase_ssh_agent.py 566) 

tests/test_derivepassphrase_ssh_agent.py 567)         with (
tests/test_derivepassphrase_ssh_agent.py 568)             pytest.raises(exc_type, match=exc_pattern),
tests/test_derivepassphrase_ssh_agent.py 569)             ssh_agent.SSHAgentClient() as client,
tests/test_derivepassphrase_ssh_agent.py 570)         ):
tests/test_derivepassphrase_ssh_agent.py 571)             client.request(request_code, b'', response_code=response_code)