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

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,
tests/test_derivepassphrase_ssh_agent.py 108)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 109)         skip_if_no_af_unix_support: None,

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

tests/test_derivepassphrase_ssh_agent.py 111)         del skip_if_no_af_unix_support

tests/test_ssh_agent_client.py           112)         monkeypatch.delenv('SSH_AUTH_SOCK', raising=False)

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

tests/test_derivepassphrase_ssh_agent.py 116)             ssh_agent.SSHAgentClient()

tests/test_ssh_agent_client.py           117) 

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

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

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

tests/test_ssh_agent_client.py           126)         assert uint32(input) == expected
tests/test_ssh_agent_client.py           127) 

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

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

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

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

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

tests/test_ssh_agent_client.py           143)         assert bytes(string(input)) == expected
tests/test_ssh_agent_client.py           144) 

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

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

tests/test_ssh_agent_client.py           151)                 b'ssh-ed25519',
tests/test_ssh_agent_client.py           152)             ),
tests/test_ssh_agent_client.py           153)         ],
tests/test_ssh_agent_client.py           154)     )

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

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

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

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

tests/test_ssh_agent_client.py           165) 

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

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

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

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

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

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

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

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

tests/test_ssh_agent_client.py           194)         ['input', 'exc_type', 'exc_pattern', 'has_trailer', 'parts'],
tests/test_ssh_agent_client.py           195)         [

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

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

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

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

tests/test_ssh_agent_client.py           210)             ),

tests/test_ssh_agent_client.py           211)         ],
tests/test_ssh_agent_client.py           212)     )
tests/test_ssh_agent_client.py           213)     def test_312_unstring_exceptions(

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

tests/test_derivepassphrase_ssh_agent.py 221)         unstring = ssh_agent.SSHAgentClient.unstring
tests/test_derivepassphrase_ssh_agent.py 222)         unstring_prefix = ssh_agent.SSHAgentClient.unstring_prefix

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

tests/test_ssh_agent_client.py           231) 

tests/test_ssh_agent_client.py           232) class TestAgentInteraction:

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

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

tests/test_ssh_agent_client.py           237)     )

tests/test_ssh_agent_client.py           238)     def test_200_sign_data_via_agent(

tests/test_derivepassphrase_ssh_agent.py 239)         self,
tests/test_derivepassphrase_ssh_agent.py 240)         ssh_agent_client_with_test_keys_loaded: ssh_agent.SSHAgentClient,
tests/test_derivepassphrase_ssh_agent.py 241)         data_dict: tests.SSHTestKey,

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

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

tests/test_ssh_agent_client.py           261) 

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

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

tests/test_ssh_agent_client.py           266)     )

tests/test_ssh_agent_client.py           267)     def test_201_sign_data_via_agent_unsupported(

tests/test_derivepassphrase_ssh_agent.py 268)         self,
tests/test_derivepassphrase_ssh_agent.py 269)         ssh_agent_client_with_test_keys_loaded: ssh_agent.SSHAgentClient,
tests/test_derivepassphrase_ssh_agent.py 270)         data_dict: tests.SSHTestKey,

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

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

tests/test_ssh_agent_client.py           280) 

tests/test_derivepassphrase_ssh_agent.py 281)     @pytest.mark.parametrize(
tests/test_derivepassphrase_ssh_agent.py 282)         ['key', 'single'],
tests/test_derivepassphrase_ssh_agent.py 283)         [
tests/test_derivepassphrase_ssh_agent.py 284)             (value['public_key_data'], False)
tests/test_derivepassphrase_ssh_agent.py 285)             for value in tests.SUPPORTED_KEYS.values()
tests/test_derivepassphrase_ssh_agent.py 286)         ]
tests/test_derivepassphrase_ssh_agent.py 287)         + [(tests.list_keys_singleton()[0].key, True)],
tests/test_derivepassphrase_ssh_agent.py 288)     )

tests/test_ssh_agent_client.py           289)     def test_210_ssh_key_selector(

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

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

tests/test_derivepassphrase_ssh_agent.py 296)         del running_ssh_agent
tests/test_derivepassphrase_ssh_agent.py 297) 

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

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

tests/test_ssh_agent_client.py           303)         if single:

tests/test_ssh_agent_client.py           304)             monkeypatch.setattr(

tests/test_derivepassphrase_ssh_agent.py 305)                 ssh_agent.SSHAgentClient,

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

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

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

tests/test_ssh_agent_client.py           316)         else:

tests/test_ssh_agent_client.py           317)             monkeypatch.setattr(

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

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

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

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

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

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

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

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

tests/test_ssh_agent_client.py           341)         )

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

tests/test_ssh_agent_client.py           345) 

tests/test_derivepassphrase_ssh_agent.py 346)     def test_300_constructor_bad_running_agent(

tests/test_derivepassphrase_ssh_agent.py 347)         self,
tests/test_derivepassphrase_ssh_agent.py 348)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 349)         running_ssh_agent: str,

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

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

tests/test_ssh_agent_client.py           356) 

tests/test_derivepassphrase_ssh_agent.py 357)     def test_301_constructor_no_af_unix_support(
tests/test_derivepassphrase_ssh_agent.py 358)         self,
tests/test_derivepassphrase_ssh_agent.py 359)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 360)     ) -> None:
tests/test_derivepassphrase_ssh_agent.py 361)         with monkeypatch.context() as monkeypatch2:
tests/test_derivepassphrase_ssh_agent.py 362)             monkeypatch2.setenv('SSH_AUTH_SOCK', "the value doesn't matter")
tests/test_derivepassphrase_ssh_agent.py 363)             monkeypatch2.delattr(socket, 'AF_UNIX', raising=False)
tests/test_derivepassphrase_ssh_agent.py 364)             with pytest.raises(
tests/test_derivepassphrase_ssh_agent.py 365)                 NotImplementedError,
tests/test_derivepassphrase_ssh_agent.py 366)                 match='UNIX domain sockets',
tests/test_derivepassphrase_ssh_agent.py 367)             ):
tests/test_derivepassphrase_ssh_agent.py 368)                 ssh_agent.SSHAgentClient()
tests/test_derivepassphrase_ssh_agent.py 369) 

tests/test_ssh_agent_client.py           370)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           371)         'response',
tests/test_ssh_agent_client.py           372)         [
tests/test_ssh_agent_client.py           373)             b'\x00\x00',
tests/test_ssh_agent_client.py           374)             b'\x00\x00\x00\x1f some bytes missing',
tests/test_ssh_agent_client.py           375)         ],
tests/test_ssh_agent_client.py           376)     )

tests/test_ssh_agent_client.py           377)     def test_310_truncated_server_response(

tests/test_derivepassphrase_ssh_agent.py 378)         self,
tests/test_derivepassphrase_ssh_agent.py 379)         monkeypatch: pytest.MonkeyPatch,
tests/test_derivepassphrase_ssh_agent.py 380)         running_ssh_agent: str,
tests/test_derivepassphrase_ssh_agent.py 381)         response: bytes,

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

tests/test_derivepassphrase_ssh_agent.py 383)         del running_ssh_agent

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

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

tests/test_ssh_agent_client.py           386) 

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

tests/test_ssh_agent_client.py           389)                 return None

tests/test_ssh_agent_client.py           390) 

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

tests/test_ssh_agent_client.py           393) 

tests/test_ssh_agent_client.py           394)         pseudo_socket = PseudoSocket()
tests/test_ssh_agent_client.py           395)         monkeypatch.setattr(client, '_connection', pseudo_socket)
tests/test_ssh_agent_client.py           396)         with pytest.raises(EOFError):
tests/test_ssh_agent_client.py           397)             client.request(255, b'')
tests/test_ssh_agent_client.py           398) 
tests/test_ssh_agent_client.py           399)     @pytest.mark.parametrize(
tests/test_ssh_agent_client.py           400)         ['response_code', 'response', 'exc_type', 'exc_pattern'],
tests/test_ssh_agent_client.py           401)         [

tests/test_ssh_agent_client.py           402)             (

tests/test_derivepassphrase_ssh_agent.py 403)                 _types.SSH_AGENT.FAILURE,
tests/test_derivepassphrase_ssh_agent.py 404)                 b'',
tests/test_derivepassphrase_ssh_agent.py 405)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 406)                 'failed to complete the request',
tests/test_derivepassphrase_ssh_agent.py 407)             ),
tests/test_derivepassphrase_ssh_agent.py 408)             (
tests/test_derivepassphrase_ssh_agent.py 409)                 _types.SSH_AGENT.IDENTITIES_ANSWER,
tests/test_derivepassphrase_ssh_agent.py 410)                 b'\x00\x00\x00\x01',
tests/test_derivepassphrase_ssh_agent.py 411)                 EOFError,
tests/test_derivepassphrase_ssh_agent.py 412)                 'truncated response',
tests/test_derivepassphrase_ssh_agent.py 413)             ),
tests/test_derivepassphrase_ssh_agent.py 414)             (
tests/test_derivepassphrase_ssh_agent.py 415)                 _types.SSH_AGENT.IDENTITIES_ANSWER,

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

tests/test_derivepassphrase_ssh_agent.py 417)                 ssh_agent.TrailingDataError,

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

tests/test_ssh_agent_client.py           419)             ),

tests/test_ssh_agent_client.py           420)         ],

tests/test_ssh_agent_client.py           421)     )

tests/test_ssh_agent_client.py           422)     def test_320_list_keys_error_responses(

tests/test_ssh_agent_client.py           423)         self,

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

tests/test_derivepassphrase_ssh_agent.py 425)         running_ssh_agent: str,

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

tests/test_ssh_agent_client.py           427)         response: bytes | bytearray,
tests/test_ssh_agent_client.py           428)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           429)         exc_pattern: str,
tests/test_ssh_agent_client.py           430)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 431)         del running_ssh_agent
tests/test_derivepassphrase_ssh_agent.py 432) 

tests/test_derivepassphrase_ssh_agent.py 433)         passed_response_code = response_code
tests/test_derivepassphrase_ssh_agent.py 434) 
tests/test_derivepassphrase_ssh_agent.py 435)         def request(
tests/test_derivepassphrase_ssh_agent.py 436)             request_code: int | _types.SSH_AGENTC,
tests/test_derivepassphrase_ssh_agent.py 437)             payload: bytes | bytearray,
tests/test_derivepassphrase_ssh_agent.py 438)             /,
tests/test_derivepassphrase_ssh_agent.py 439)             *,
tests/test_derivepassphrase_ssh_agent.py 440)             response_code: Iterable[int | _types.SSH_AGENT]
tests/test_derivepassphrase_ssh_agent.py 441)             | int
tests/test_derivepassphrase_ssh_agent.py 442)             | _types.SSH_AGENT
tests/test_derivepassphrase_ssh_agent.py 443)             | None = None,
tests/test_derivepassphrase_ssh_agent.py 444)         ) -> tuple[int, bytes | bytearray] | bytes | bytearray:
tests/test_derivepassphrase_ssh_agent.py 445)             del request_code
tests/test_derivepassphrase_ssh_agent.py 446)             del payload
tests/test_derivepassphrase_ssh_agent.py 447)             if isinstance(  # pragma: no branch

tests/test_derivepassphrase_ssh_agent.py 448)                 response_code, (int, _types.SSH_AGENT)

tests/test_derivepassphrase_ssh_agent.py 449)             ):
tests/test_derivepassphrase_ssh_agent.py 450)                 response_code = frozenset({response_code})
tests/test_derivepassphrase_ssh_agent.py 451)             if response_code is not None:  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 452)                 response_code = frozenset({
tests/test_derivepassphrase_ssh_agent.py 453)                     c if isinstance(c, int) else c.value for c in response_code
tests/test_derivepassphrase_ssh_agent.py 454)                 })
tests/test_derivepassphrase_ssh_agent.py 455) 
tests/test_derivepassphrase_ssh_agent.py 456)             if not response_code:  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 457)                 return (passed_response_code.value, response)
tests/test_derivepassphrase_ssh_agent.py 458)             if passed_response_code.value not in response_code:
tests/test_derivepassphrase_ssh_agent.py 459)                 raise ssh_agent.SSHAgentFailedError(
tests/test_derivepassphrase_ssh_agent.py 460)                     passed_response_code.value, response
tests/test_derivepassphrase_ssh_agent.py 461)                 )
tests/test_derivepassphrase_ssh_agent.py 462)             return response
tests/test_derivepassphrase_ssh_agent.py 463) 

tests/test_derivepassphrase_ssh_agent.py 464)         with monkeypatch.context() as monkeypatch2:
tests/test_derivepassphrase_ssh_agent.py 465)             client = ssh_agent.SSHAgentClient()
tests/test_derivepassphrase_ssh_agent.py 466)             monkeypatch2.setattr(client, 'request', request)
tests/test_derivepassphrase_ssh_agent.py 467)             with pytest.raises(exc_type, match=exc_pattern):
tests/test_derivepassphrase_ssh_agent.py 468)                 client.list_keys()

tests/test_ssh_agent_client.py           469) 
tests/test_ssh_agent_client.py           470)     @pytest.mark.parametrize(

tests/test_derivepassphrase_ssh_agent.py 471)         [
tests/test_derivepassphrase_ssh_agent.py 472)             'key',
tests/test_derivepassphrase_ssh_agent.py 473)             'check',
tests/test_derivepassphrase_ssh_agent.py 474)             'response_code',
tests/test_derivepassphrase_ssh_agent.py 475)             'response',
tests/test_derivepassphrase_ssh_agent.py 476)             'exc_type',
tests/test_derivepassphrase_ssh_agent.py 477)             'exc_pattern',
tests/test_derivepassphrase_ssh_agent.py 478)         ],

tests/test_ssh_agent_client.py           479)         [
tests/test_ssh_agent_client.py           480)             (
tests/test_ssh_agent_client.py           481)                 b'invalid-key',
tests/test_ssh_agent_client.py           482)                 True,

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

tests/test_ssh_agent_client.py           485)                 KeyError,
tests/test_ssh_agent_client.py           486)                 'target SSH key not loaded into agent',
tests/test_ssh_agent_client.py           487)             ),
tests/test_ssh_agent_client.py           488)             (
tests/test_ssh_agent_client.py           489)                 tests.SUPPORTED_KEYS['ed25519']['public_key_data'],
tests/test_ssh_agent_client.py           490)                 True,

tests/test_derivepassphrase_ssh_agent.py 491)                 _types.SSH_AGENT.FAILURE,
tests/test_derivepassphrase_ssh_agent.py 492)                 b'',

tests/test_derivepassphrase_ssh_agent.py 493)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 494)                 'failed to complete the request',

tests/test_ssh_agent_client.py           495)             ),
tests/test_ssh_agent_client.py           496)         ],

tests/test_ssh_agent_client.py           497)     )

tests/test_ssh_agent_client.py           498)     def test_330_sign_error_responses(

tests/test_ssh_agent_client.py           499)         self,

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

tests/test_derivepassphrase_ssh_agent.py 501)         running_ssh_agent: str,

tests/test_ssh_agent_client.py           502)         key: bytes | bytearray,
tests/test_ssh_agent_client.py           503)         check: bool,

tests/test_derivepassphrase_ssh_agent.py 504)         response_code: _types.SSH_AGENT,
tests/test_derivepassphrase_ssh_agent.py 505)         response: bytes | bytearray,

tests/test_ssh_agent_client.py           506)         exc_type: type[Exception],
tests/test_ssh_agent_client.py           507)         exc_pattern: str,
tests/test_ssh_agent_client.py           508)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 509)         del running_ssh_agent

tests/test_derivepassphrase_ssh_agent.py 510)         passed_response_code = response_code
tests/test_derivepassphrase_ssh_agent.py 511) 
tests/test_derivepassphrase_ssh_agent.py 512)         def request(
tests/test_derivepassphrase_ssh_agent.py 513)             request_code: int | _types.SSH_AGENTC,
tests/test_derivepassphrase_ssh_agent.py 514)             payload: bytes | bytearray,
tests/test_derivepassphrase_ssh_agent.py 515)             /,
tests/test_derivepassphrase_ssh_agent.py 516)             *,
tests/test_derivepassphrase_ssh_agent.py 517)             response_code: Iterable[int | _types.SSH_AGENT]
tests/test_derivepassphrase_ssh_agent.py 518)             | int
tests/test_derivepassphrase_ssh_agent.py 519)             | _types.SSH_AGENT
tests/test_derivepassphrase_ssh_agent.py 520)             | None = None,
tests/test_derivepassphrase_ssh_agent.py 521)         ) -> tuple[int, bytes | bytearray] | bytes | bytearray:
tests/test_derivepassphrase_ssh_agent.py 522)             del request_code
tests/test_derivepassphrase_ssh_agent.py 523)             del payload
tests/test_derivepassphrase_ssh_agent.py 524)             if isinstance(  # pragma: no branch

tests/test_derivepassphrase_ssh_agent.py 525)                 response_code, (int, _types.SSH_AGENT)

tests/test_derivepassphrase_ssh_agent.py 526)             ):
tests/test_derivepassphrase_ssh_agent.py 527)                 response_code = frozenset({response_code})
tests/test_derivepassphrase_ssh_agent.py 528)             if response_code is not None:  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 529)                 response_code = frozenset({
tests/test_derivepassphrase_ssh_agent.py 530)                     c if isinstance(c, int) else c.value for c in response_code
tests/test_derivepassphrase_ssh_agent.py 531)                 })
tests/test_derivepassphrase_ssh_agent.py 532) 
tests/test_derivepassphrase_ssh_agent.py 533)             if not response_code:  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 534)                 return (passed_response_code.value, response)
tests/test_derivepassphrase_ssh_agent.py 535)             if (
tests/test_derivepassphrase_ssh_agent.py 536)                 passed_response_code.value not in response_code
tests/test_derivepassphrase_ssh_agent.py 537)             ):  # pragma: no branch
tests/test_derivepassphrase_ssh_agent.py 538)                 raise ssh_agent.SSHAgentFailedError(
tests/test_derivepassphrase_ssh_agent.py 539)                     passed_response_code.value, response
tests/test_derivepassphrase_ssh_agent.py 540)                 )
tests/test_derivepassphrase_ssh_agent.py 541)             return response  # pragma: no cover
tests/test_derivepassphrase_ssh_agent.py 542) 

tests/test_derivepassphrase_ssh_agent.py 543)         with monkeypatch.context() as monkeypatch2:
tests/test_derivepassphrase_ssh_agent.py 544)             client = ssh_agent.SSHAgentClient()
tests/test_derivepassphrase_ssh_agent.py 545)             monkeypatch2.setattr(client, 'request', request)
tests/test_derivepassphrase_ssh_agent.py 546)             KeyCommentPair = _types.KeyCommentPair  # noqa: N806
tests/test_derivepassphrase_ssh_agent.py 547)             loaded_keys = [
tests/test_derivepassphrase_ssh_agent.py 548)                 KeyCommentPair(v['public_key_data'], b'no comment')
tests/test_derivepassphrase_ssh_agent.py 549)                 for v in tests.SUPPORTED_KEYS.values()
tests/test_derivepassphrase_ssh_agent.py 550)             ]
tests/test_derivepassphrase_ssh_agent.py 551)             monkeypatch2.setattr(client, 'list_keys', lambda: loaded_keys)
tests/test_derivepassphrase_ssh_agent.py 552)             with pytest.raises(exc_type, match=exc_pattern):
tests/test_derivepassphrase_ssh_agent.py 553)                 client.sign(key, b'abc', check_if_key_loaded=check)

tests/test_derivepassphrase_ssh_agent.py 554) 
tests/test_derivepassphrase_ssh_agent.py 555)     @pytest.mark.parametrize(
tests/test_derivepassphrase_ssh_agent.py 556)         ['request_code', 'response_code', 'exc_type', 'exc_pattern'],
tests/test_derivepassphrase_ssh_agent.py 557)         [
tests/test_derivepassphrase_ssh_agent.py 558)             (
tests/test_derivepassphrase_ssh_agent.py 559)                 _types.SSH_AGENTC.REQUEST_IDENTITIES,
tests/test_derivepassphrase_ssh_agent.py 560)                 _types.SSH_AGENT.SUCCESS,
tests/test_derivepassphrase_ssh_agent.py 561)                 ssh_agent.SSHAgentFailedError,
tests/test_derivepassphrase_ssh_agent.py 562)                 f'[Code {_types.SSH_AGENT.IDENTITIES_ANSWER.value}]',
tests/test_derivepassphrase_ssh_agent.py 563)             ),
tests/test_derivepassphrase_ssh_agent.py 564)         ],
tests/test_derivepassphrase_ssh_agent.py 565)     )
tests/test_derivepassphrase_ssh_agent.py 566)     def test_340_request_error_responses(
tests/test_derivepassphrase_ssh_agent.py 567)         self,

tests/test_derivepassphrase_ssh_agent.py 568)         running_ssh_agent: str,

tests/test_derivepassphrase_ssh_agent.py 569)         request_code: _types.SSH_AGENTC,
tests/test_derivepassphrase_ssh_agent.py 570)         response_code: _types.SSH_AGENT,
tests/test_derivepassphrase_ssh_agent.py 571)         exc_type: type[Exception],
tests/test_derivepassphrase_ssh_agent.py 572)         exc_pattern: str,
tests/test_derivepassphrase_ssh_agent.py 573)     ) -> None:

tests/test_derivepassphrase_ssh_agent.py 574)         del running_ssh_agent
tests/test_derivepassphrase_ssh_agent.py 575) 

tests/test_derivepassphrase_ssh_agent.py 576)         with (
tests/test_derivepassphrase_ssh_agent.py 577)             pytest.raises(exc_type, match=exc_pattern),
tests/test_derivepassphrase_ssh_agent.py 578)             ssh_agent.SSHAgentClient() as client,
tests/test_derivepassphrase_ssh_agent.py 579)         ):
tests/test_derivepassphrase_ssh_agent.py 580)             client.request(request_code, b'', response_code=response_code)