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 os
tests/test_ssh_agent_client.py            12) import socket
tests/test_ssh_agent_client.py            13) import subprocess

tests/test_ssh_agent_client.py            14) from typing import TYPE_CHECKING

tests/test_ssh_agent_client.py            15) 
tests/test_ssh_agent_client.py            16) import click
tests/test_ssh_agent_client.py            17) import click.testing

tests/test_ssh_agent_client.py            18) import pytest
tests/test_ssh_agent_client.py            19) from typing_extensions import Any
tests/test_ssh_agent_client.py            20) 

tests/test_ssh_agent_client.py            21) import tests

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

tests/test_ssh_agent_client.py            23) 

tests/test_ssh_agent_client.py            24) if TYPE_CHECKING:
tests/test_ssh_agent_client.py            25)     from collections.abc import Iterator
tests/test_ssh_agent_client.py            26) 

tests/test_ssh_agent_client.py            27) 

tests/test_ssh_agent_client.py            28) class TestStaticFunctionality:

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

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

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

tests/test_derivepassphrase_ssh_agent.py  44)     def test_200_constructor_no_running_agent(
tests/test_derivepassphrase_ssh_agent.py  45)         self, monkeypatch: pytest.MonkeyPatch
tests/test_derivepassphrase_ssh_agent.py  46)     ) -> None:

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

tests/test_ssh_agent_client.py            49)         with pytest.raises(
tests/test_ssh_agent_client.py            50)             KeyError, match='SSH_AUTH_SOCK environment variable'
tests/test_ssh_agent_client.py            51)         ):

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

tests/test_ssh_agent_client.py            53) 

tests/test_ssh_agent_client.py            54)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py            55)         ['input', 'expected'],
tests/test_ssh_agent_client.py            56)         [
tests/test_ssh_agent_client.py            57)             (16777216, b'\x01\x00\x00\x00'),
tests/test_ssh_agent_client.py            58)         ],
tests/test_ssh_agent_client.py            59)     )

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

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

tests/test_ssh_agent_client.py            62)         assert uint32(input) == expected
tests/test_ssh_agent_client.py            63) 

tests/test_ssh_agent_client.py            64)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py            65)         ['input', 'expected'],
tests/test_ssh_agent_client.py            66)         [
tests/test_ssh_agent_client.py            67)             (b'ssh-rsa', b'\x00\x00\x00\x07ssh-rsa'),
tests/test_ssh_agent_client.py            68)             (b'ssh-ed25519', b'\x00\x00\x00\x0bssh-ed25519'),
tests/test_ssh_agent_client.py            69)             (

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

tests/test_ssh_agent_client.py            71)                 b'\x00\x00\x00\x0f\x00\x00\x00\x0bssh-ed25519',
tests/test_ssh_agent_client.py            72)             ),
tests/test_ssh_agent_client.py            73)         ],
tests/test_ssh_agent_client.py            74)     )

tests/test_ssh_agent_client.py            75)     def test_211_string(
tests/test_ssh_agent_client.py            76)         self, input: bytes | bytearray, expected: bytes | bytearray
tests/test_ssh_agent_client.py            77)     ) -> None:

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

tests/test_ssh_agent_client.py            79)         assert bytes(string(input)) == expected
tests/test_ssh_agent_client.py            80) 

tests/test_ssh_agent_client.py            81)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py            82)         ['input', 'expected'],
tests/test_ssh_agent_client.py            83)         [
tests/test_ssh_agent_client.py            84)             (b'\x00\x00\x00\x07ssh-rsa', b'ssh-rsa'),
tests/test_ssh_agent_client.py            85)             (

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

tests/test_ssh_agent_client.py            87)                 b'ssh-ed25519',
tests/test_ssh_agent_client.py            88)             ),
tests/test_ssh_agent_client.py            89)         ],
tests/test_ssh_agent_client.py            90)     )

tests/test_ssh_agent_client.py            91)     def test_212_unstring(
tests/test_ssh_agent_client.py            92)         self, input: bytes | bytearray, expected: bytes | bytearray
tests/test_ssh_agent_client.py            93)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py  94)         unstring = ssh_agent.SSHAgentClient.unstring
tests/test_derivepassphrase_ssh_agent.py  95)         unstring_prefix = ssh_agent.SSHAgentClient.unstring_prefix

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

tests/test_ssh_agent_client.py            97)         assert tuple(bytes(x) for x in unstring_prefix(input)) == (
tests/test_ssh_agent_client.py            98)             expected,
tests/test_ssh_agent_client.py            99)             b'',
tests/test_ssh_agent_client.py           100)         )

tests/test_ssh_agent_client.py           101) 

tests/test_ssh_agent_client.py           102)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           103)         ['value', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           104)         [
tests/test_ssh_agent_client.py           105)             (10000000000000000, OverflowError, 'int too big to convert'),
tests/test_ssh_agent_client.py           106)             (-1, OverflowError, "can't convert negative int to unsigned"),
tests/test_ssh_agent_client.py           107)         ],
tests/test_ssh_agent_client.py           108)     )

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

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

tests/test_ssh_agent_client.py           113)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           114)             uint32(value)
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', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           118)         [
tests/test_ssh_agent_client.py           119)             ('some string', TypeError, 'invalid payload type'),
tests/test_ssh_agent_client.py           120)         ],
tests/test_ssh_agent_client.py           121)     )

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

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

tests/test_ssh_agent_client.py           126)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           127)             string(input)
tests/test_ssh_agent_client.py           128) 
tests/test_ssh_agent_client.py           129)     @pytest.mark.parametrize(

tests/test_ssh_agent_client.py           130)         ['input', 'exc_type', 'exc_pattern', 'has_trailer', 'parts'],
tests/test_ssh_agent_client.py           131)         [

tests/test_ssh_agent_client.py           132)             (b'ssh', ValueError, 'malformed SSH byte string', False, None),
tests/test_ssh_agent_client.py           133)             (
tests/test_ssh_agent_client.py           134)                 b'\x00\x00\x00\x08ssh-rsa',

tests/test_ssh_agent_client.py           135)                 ValueError,
tests/test_ssh_agent_client.py           136)                 'malformed SSH byte string',
tests/test_ssh_agent_client.py           137)                 False,
tests/test_ssh_agent_client.py           138)                 None,

tests/test_ssh_agent_client.py           139)             ),
tests/test_ssh_agent_client.py           140)             (
tests/test_ssh_agent_client.py           141)                 b'\x00\x00\x00\x04XXX trailing text',

tests/test_ssh_agent_client.py           142)                 ValueError,
tests/test_ssh_agent_client.py           143)                 'malformed SSH byte string',
tests/test_ssh_agent_client.py           144)                 True,
tests/test_ssh_agent_client.py           145)                 (b'XXX ', b'trailing text'),

tests/test_ssh_agent_client.py           146)             ),

tests/test_ssh_agent_client.py           147)         ],
tests/test_ssh_agent_client.py           148)     )
tests/test_ssh_agent_client.py           149)     def test_312_unstring_exceptions(

tests/test_ssh_agent_client.py           150)         self,
tests/test_ssh_agent_client.py           151)         input: bytes | bytearray,
tests/test_ssh_agent_client.py           152)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           153)         exc_pattern: str,
tests/test_ssh_agent_client.py           154)         has_trailer: bool,
tests/test_ssh_agent_client.py           155)         parts: tuple[bytes | bytearray, bytes | bytearray] | None,
tests/test_ssh_agent_client.py           156)     ) -> None:

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

tests/test_ssh_agent_client.py           159)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           160)             unstring(input)
tests/test_ssh_agent_client.py           161)         if has_trailer:
tests/test_ssh_agent_client.py           162)             assert tuple(bytes(x) for x in unstring_prefix(input)) == parts
tests/test_ssh_agent_client.py           163)         else:
tests/test_ssh_agent_client.py           164)             with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           165)                 unstring_prefix(input)
tests/test_ssh_agent_client.py           166) 

tests/test_ssh_agent_client.py           167) 

tests/test_ssh_agent_client.py           168) @tests.skip_if_no_agent
tests/test_ssh_agent_client.py           169) class TestAgentInteraction:

tests/test_ssh_agent_client.py           170)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           171)         ['keytype', 'data_dict'], list(tests.SUPPORTED_KEYS.items())
tests/test_ssh_agent_client.py           172)     )

tests/test_ssh_agent_client.py           173)     def test_200_sign_data_via_agent(
tests/test_ssh_agent_client.py           174)         self, keytype: str, data_dict: tests.SSHTestKey
tests/test_ssh_agent_client.py           175)     ) -> None:

tests/test_ssh_agent_client.py           176)         del keytype  # Unused.

tests/test_ssh_agent_client.py           177)         private_key = data_dict['private_key']
tests/test_ssh_agent_client.py           178)         try:

tests/test_ssh_agent_client.py           179)             _ = subprocess.run(
tests/test_ssh_agent_client.py           180)                 ['ssh-add', '-t', '30', '-q', '-'],
tests/test_ssh_agent_client.py           181)                 input=private_key,
tests/test_ssh_agent_client.py           182)                 check=True,
tests/test_ssh_agent_client.py           183)                 capture_output=True,
tests/test_ssh_agent_client.py           184)             )

tests/test_ssh_agent_client.py           185)         except subprocess.CalledProcessError as e:
tests/test_ssh_agent_client.py           186)             pytest.skip(

tests/test_ssh_agent_client.py           187)                 f'uploading test key: {e!r}, stdout={e.stdout!r}, '
tests/test_ssh_agent_client.py           188)                 f'stderr={e.stderr!r}'

tests/test_ssh_agent_client.py           189)             )
tests/test_ssh_agent_client.py           190)         else:
tests/test_ssh_agent_client.py           191)             try:

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

tests/test_ssh_agent_client.py           193)             except OSError:  # pragma: no cover
tests/test_ssh_agent_client.py           194)                 pytest.skip('communication error with the SSH agent')
tests/test_ssh_agent_client.py           195)         with client:

tests/test_ssh_agent_client.py           196)             key_comment_pairs = {
tests/test_ssh_agent_client.py           197)                 bytes(k): bytes(c) for k, c in client.list_keys()
tests/test_ssh_agent_client.py           198)             }

tests/test_ssh_agent_client.py           199)             public_key_data = data_dict['public_key_data']
tests/test_ssh_agent_client.py           200)             expected_signature = data_dict['expected_signature']
tests/test_ssh_agent_client.py           201)             derived_passphrase = data_dict['derived_passphrase']
tests/test_ssh_agent_client.py           202)             if public_key_data not in key_comment_pairs:  # pragma: no cover
tests/test_ssh_agent_client.py           203)                 pytest.skip('prerequisite SSH key not loaded')

tests/test_ssh_agent_client.py           204)             signature = bytes(

tests/test_derivepassphrase_ssh_agent.py 205)                 client.sign(payload=vault.Vault._UUID, key=public_key_data)

tests/test_ssh_agent_client.py           206)             )

tests/test_ssh_agent_client.py           207)             assert signature == expected_signature, 'SSH signature mismatch'

tests/test_ssh_agent_client.py           208)             signature2 = bytes(

tests/test_derivepassphrase_ssh_agent.py 209)                 client.sign(payload=vault.Vault._UUID, key=public_key_data)

tests/test_ssh_agent_client.py           210)             )

tests/test_ssh_agent_client.py           211)             assert signature2 == expected_signature, 'SSH signature mismatch'
tests/test_ssh_agent_client.py           212)             assert (

tests/test_derivepassphrase_ssh_agent.py 213)                 vault.Vault.phrase_from_key(public_key_data)

tests/test_ssh_agent_client.py           214)                 == derived_passphrase

tests/test_ssh_agent_client.py           215)             ), 'SSH signature mismatch'
tests/test_ssh_agent_client.py           216) 

tests/test_ssh_agent_client.py           217)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           218)         ['keytype', 'data_dict'], list(tests.UNSUITABLE_KEYS.items())
tests/test_ssh_agent_client.py           219)     )

tests/test_ssh_agent_client.py           220)     def test_201_sign_data_via_agent_unsupported(
tests/test_ssh_agent_client.py           221)         self, keytype: str, data_dict: tests.SSHTestKey
tests/test_ssh_agent_client.py           222)     ) -> None:

tests/test_ssh_agent_client.py           223)         del keytype  # Unused.

tests/test_ssh_agent_client.py           224)         private_key = data_dict['private_key']
tests/test_ssh_agent_client.py           225)         try:

tests/test_ssh_agent_client.py           226)             _ = subprocess.run(
tests/test_ssh_agent_client.py           227)                 ['ssh-add', '-t', '30', '-q', '-'],
tests/test_ssh_agent_client.py           228)                 input=private_key,
tests/test_ssh_agent_client.py           229)                 check=True,
tests/test_ssh_agent_client.py           230)                 capture_output=True,
tests/test_ssh_agent_client.py           231)             )

tests/test_ssh_agent_client.py           232)         except subprocess.CalledProcessError as e:  # pragma: no cover
tests/test_ssh_agent_client.py           233)             pytest.skip(

tests/test_ssh_agent_client.py           234)                 f'uploading test key: {e!r}, stdout={e.stdout!r}, '
tests/test_ssh_agent_client.py           235)                 f'stderr={e.stderr!r}'

tests/test_ssh_agent_client.py           236)             )
tests/test_ssh_agent_client.py           237)         else:
tests/test_ssh_agent_client.py           238)             try:

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

tests/test_ssh_agent_client.py           240)             except OSError:  # pragma: no cover
tests/test_ssh_agent_client.py           241)                 pytest.skip('communication error with the SSH agent')
tests/test_ssh_agent_client.py           242)         with client:

tests/test_ssh_agent_client.py           243)             key_comment_pairs = {
tests/test_ssh_agent_client.py           244)                 bytes(k): bytes(c) for k, c in client.list_keys()
tests/test_ssh_agent_client.py           245)             }

tests/test_ssh_agent_client.py           246)             public_key_data = data_dict['public_key_data']

tests/test_ssh_agent_client.py           247)             _ = data_dict['expected_signature']

tests/test_ssh_agent_client.py           248)             if public_key_data not in key_comment_pairs:  # pragma: no cover
tests/test_ssh_agent_client.py           249)                 pytest.skip('prerequisite SSH key not loaded')

tests/test_ssh_agent_client.py           250)             signature = bytes(

tests/test_derivepassphrase_ssh_agent.py 251)                 client.sign(payload=vault.Vault._UUID, key=public_key_data)

tests/test_ssh_agent_client.py           252)             )
tests/test_ssh_agent_client.py           253)             signature2 = bytes(

tests/test_derivepassphrase_ssh_agent.py 254)                 client.sign(payload=vault.Vault._UUID, key=public_key_data)

tests/test_ssh_agent_client.py           255)             )

tests/test_ssh_agent_client.py           256)             assert signature != signature2, 'SSH signature repeatable?!'
tests/test_ssh_agent_client.py           257)             with pytest.raises(ValueError, match='unsuitable SSH key'):

tests/test_derivepassphrase_ssh_agent.py 258)                 vault.Vault.phrase_from_key(public_key_data)

tests/test_ssh_agent_client.py           259) 
tests/test_ssh_agent_client.py           260)     @staticmethod

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

tests/test_ssh_agent_client.py           262)         for value in tests.SUPPORTED_KEYS.values():
tests/test_ssh_agent_client.py           263)             key = value['public_key_data']
tests/test_ssh_agent_client.py           264)             yield (key, False)
tests/test_ssh_agent_client.py           265)         singleton_key = tests.list_keys_singleton()[0].key
tests/test_ssh_agent_client.py           266)         for value in tests.SUPPORTED_KEYS.values():
tests/test_ssh_agent_client.py           267)             key = value['public_key_data']
tests/test_ssh_agent_client.py           268)             if key == singleton_key:
tests/test_ssh_agent_client.py           269)                 yield (key, True)
tests/test_ssh_agent_client.py           270) 
tests/test_ssh_agent_client.py           271)     @pytest.mark.parametrize(['key', 'single'], list(_params()))

tests/test_ssh_agent_client.py           272)     def test_210_ssh_key_selector(

tests/test_derivepassphrase_ssh_agent.py 273)         self, monkeypatch: pytest.MonkeyPatch, key: bytes, single: bool

tests/test_ssh_agent_client.py           274)     ) -> None:
tests/test_ssh_agent_client.py           275)         def key_is_suitable(key: bytes) -> bool:

tests/test_ssh_agent_client.py           276)             return key in {
tests/test_ssh_agent_client.py           277)                 v['public_key_data'] for v in tests.SUPPORTED_KEYS.values()
tests/test_ssh_agent_client.py           278)             }
tests/test_ssh_agent_client.py           279) 

tests/test_ssh_agent_client.py           280)         if single:

tests/test_ssh_agent_client.py           281)             monkeypatch.setattr(

tests/test_derivepassphrase_ssh_agent.py 282)                 ssh_agent.SSHAgentClient,

tests/test_ssh_agent_client.py           283)                 'list_keys',
tests/test_ssh_agent_client.py           284)                 tests.list_keys_singleton,
tests/test_ssh_agent_client.py           285)             )
tests/test_ssh_agent_client.py           286)             keys = [
tests/test_ssh_agent_client.py           287)                 pair.key
tests/test_ssh_agent_client.py           288)                 for pair in tests.list_keys_singleton()
tests/test_ssh_agent_client.py           289)                 if key_is_suitable(pair.key)
tests/test_ssh_agent_client.py           290)             ]

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

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

tests/test_ssh_agent_client.py           293)         else:

tests/test_ssh_agent_client.py           294)             monkeypatch.setattr(

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

tests/test_ssh_agent_client.py           296)             )
tests/test_ssh_agent_client.py           297)             keys = [
tests/test_ssh_agent_client.py           298)                 pair.key
tests/test_ssh_agent_client.py           299)                 for pair in tests.list_keys()
tests/test_ssh_agent_client.py           300)                 if key_is_suitable(pair.key)
tests/test_ssh_agent_client.py           301)             ]

tests/test_ssh_agent_client.py           302)             index = str(1 + keys.index(key))
tests/test_ssh_agent_client.py           303)             n = len(keys)
tests/test_ssh_agent_client.py           304)             text = f'Your selection? (1-{n}, leave empty to abort): {index}\n'
tests/test_ssh_agent_client.py           305)         b64_key = base64.standard_b64encode(key).decode('ASCII')
tests/test_ssh_agent_client.py           306) 
tests/test_ssh_agent_client.py           307)         @click.command()

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

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

tests/test_ssh_agent_client.py           310)             click.echo(base64.standard_b64encode(key).decode('ASCII'))
tests/test_ssh_agent_client.py           311) 
tests/test_ssh_agent_client.py           312)         runner = click.testing.CliRunner(mix_stderr=True)

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

tests/test_ssh_agent_client.py           314)             driver,
tests/test_ssh_agent_client.py           315)             [],
tests/test_ssh_agent_client.py           316)             input=('yes\n' if single else f'{index}\n'),
tests/test_ssh_agent_client.py           317)             catch_exceptions=True,

tests/test_ssh_agent_client.py           318)         )

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

tests/test_ssh_agent_client.py           322) 
tests/test_ssh_agent_client.py           323)     del _params
tests/test_ssh_agent_client.py           324) 

tests/test_derivepassphrase_ssh_agent.py 325)     def test_300_constructor_bad_running_agent(
tests/test_derivepassphrase_ssh_agent.py 326)         self, monkeypatch: pytest.MonkeyPatch
tests/test_derivepassphrase_ssh_agent.py 327)     ) -> None:

tests/test_ssh_agent_client.py           328)         monkeypatch.setenv('SSH_AUTH_SOCK', os.environ['SSH_AUTH_SOCK'] + '~')

tests/test_ssh_agent_client.py           329)         sock = socket.socket(family=socket.AF_UNIX)

tests/test_ssh_agent_client.py           330)         with pytest.raises(OSError):  # noqa: PT011

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

tests/test_ssh_agent_client.py           332) 

tests/test_ssh_agent_client.py           333)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           334)         'response',
tests/test_ssh_agent_client.py           335)         [
tests/test_ssh_agent_client.py           336)             b'\x00\x00',
tests/test_ssh_agent_client.py           337)             b'\x00\x00\x00\x1f some bytes missing',
tests/test_ssh_agent_client.py           338)         ],
tests/test_ssh_agent_client.py           339)     )

tests/test_ssh_agent_client.py           340)     def test_310_truncated_server_response(

tests/test_derivepassphrase_ssh_agent.py 341)         self, monkeypatch: pytest.MonkeyPatch, response: bytes

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

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

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

tests/test_ssh_agent_client.py           345) 

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

tests/test_ssh_agent_client.py           348)                 return None

tests/test_ssh_agent_client.py           349) 

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

tests/test_ssh_agent_client.py           352) 

tests/test_ssh_agent_client.py           353)         pseudo_socket = PseudoSocket()
tests/test_ssh_agent_client.py           354)         monkeypatch.setattr(client, '_connection', pseudo_socket)
tests/test_ssh_agent_client.py           355)         with pytest.raises(EOFError):
tests/test_ssh_agent_client.py           356)             client.request(255, b'')
tests/test_ssh_agent_client.py           357) 
tests/test_ssh_agent_client.py           358)     @tests.skip_if_no_agent
tests/test_ssh_agent_client.py           359)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           360)         ['response_code', 'response', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           361)         [

tests/test_ssh_agent_client.py           362)             (

tests/test_derivepassphrase_ssh_agent.py 363)                 _types.SSH_AGENT.FAILURE,
tests/test_derivepassphrase_ssh_agent.py 364)                 b'',
tests/test_derivepassphrase_ssh_agent.py 365)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 366)                 'failed to complete the request',
tests/test_derivepassphrase_ssh_agent.py 367)             ),
tests/test_derivepassphrase_ssh_agent.py 368)             (
tests/test_derivepassphrase_ssh_agent.py 369)                 _types.SSH_AGENT.IDENTITIES_ANSWER,
tests/test_derivepassphrase_ssh_agent.py 370)                 b'\x00\x00\x00\x01',
tests/test_derivepassphrase_ssh_agent.py 371)                 EOFError,
tests/test_derivepassphrase_ssh_agent.py 372)                 'truncated response',
tests/test_derivepassphrase_ssh_agent.py 373)             ),
tests/test_derivepassphrase_ssh_agent.py 374)             (
tests/test_derivepassphrase_ssh_agent.py 375)                 _types.SSH_AGENT.IDENTITIES_ANSWER,

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

tests/test_derivepassphrase_ssh_agent.py 377)                 ssh_agent.TrailingDataError,

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

tests/test_ssh_agent_client.py           379)             ),

tests/test_ssh_agent_client.py           380)         ],

tests/test_ssh_agent_client.py           381)     )

tests/test_ssh_agent_client.py           382)     def test_320_list_keys_error_responses(

tests/test_ssh_agent_client.py           383)         self,

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

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

tests/test_ssh_agent_client.py           386)         response: bytes | bytearray,
tests/test_ssh_agent_client.py           387)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           388)         exc_pattern: str,
tests/test_ssh_agent_client.py           389)     ) -> None:

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

tests/test_ssh_agent_client.py           391)         monkeypatch.setattr(
tests/test_ssh_agent_client.py           392)             client,
tests/test_ssh_agent_client.py           393)             'request',

tests/test_derivepassphrase_ssh_agent.py 394)             lambda *a, **kw: (response_code.value, response),  # noqa: ARG005

tests/test_ssh_agent_client.py           395)         )

tests/test_ssh_agent_client.py           396)         with pytest.raises(exc_type, match=exc_pattern):
tests/test_ssh_agent_client.py           397)             client.list_keys()
tests/test_ssh_agent_client.py           398) 
tests/test_ssh_agent_client.py           399)     @tests.skip_if_no_agent
tests/test_ssh_agent_client.py           400)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           401)         ['key', 'check', 'response', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           402)         [
tests/test_ssh_agent_client.py           403)             (
tests/test_ssh_agent_client.py           404)                 b'invalid-key',
tests/test_ssh_agent_client.py           405)                 True,

tests/test_derivepassphrase_ssh_agent.py 406)                 (_types.SSH_AGENT.FAILURE, b''),

tests/test_ssh_agent_client.py           407)                 KeyError,
tests/test_ssh_agent_client.py           408)                 'target SSH key not loaded into agent',
tests/test_ssh_agent_client.py           409)             ),
tests/test_ssh_agent_client.py           410)             (
tests/test_ssh_agent_client.py           411)                 tests.SUPPORTED_KEYS['ed25519']['public_key_data'],
tests/test_ssh_agent_client.py           412)                 True,

tests/test_derivepassphrase_ssh_agent.py 413)                 (_types.SSH_AGENT.FAILURE, b''),
tests/test_derivepassphrase_ssh_agent.py 414)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 415)                 'failed to complete the request',

tests/test_ssh_agent_client.py           416)             ),
tests/test_ssh_agent_client.py           417)         ],

tests/test_ssh_agent_client.py           418)     )

tests/test_ssh_agent_client.py           419)     def test_330_sign_error_responses(

tests/test_ssh_agent_client.py           420)         self,

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

tests/test_ssh_agent_client.py           422)         key: bytes | bytearray,
tests/test_ssh_agent_client.py           423)         check: bool,

tests/test_derivepassphrase_ssh_agent.py 424)         response: tuple[_types.SSH_AGENT, bytes | bytearray],

tests/test_ssh_agent_client.py           425)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           426)         exc_pattern: str,
tests/test_ssh_agent_client.py           427)     ) -> None:

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

tests/test_derivepassphrase_ssh_agent.py 429)         monkeypatch.setattr(
tests/test_derivepassphrase_ssh_agent.py 430)             client,
tests/test_derivepassphrase_ssh_agent.py 431)             'request',
tests/test_derivepassphrase_ssh_agent.py 432)             lambda a, b: (response[0].value, response[1]),  # noqa: ARG005
tests/test_derivepassphrase_ssh_agent.py 433)         )

tests/test_derivepassphrase_ssh_agent.py 434)         KeyCommentPair = _types.KeyCommentPair  # noqa: N806

tests/test_ssh_agent_client.py           435)         loaded_keys = [
tests/test_ssh_agent_client.py           436)             KeyCommentPair(v['public_key_data'], b'no comment')
tests/test_ssh_agent_client.py           437)             for v in tests.SUPPORTED_KEYS.values()
tests/test_ssh_agent_client.py           438)         ]