derivepassphrase.git

1) # SPDX-FileCopyrightText: 2024 Marco Ricci <m@the13thletter.info>
2) #
3) # SPDX-License-Identifier: MIT
4) 
5) from __future__ import annotations
6) 
7) import os
8) 
9) import click.testing
10) import pytest
11) 
12) import tests
13) from derivepassphrase import exporter

14) from derivepassphrase.exporter import cli

15) 
16) 
17) class Test001ExporterUtils:
18)     @pytest.mark.parametrize(
19)         ['expected', 'vault_key', 'logname', 'user', 'username'],
20)         [
21)             ('4username', None, None, None, '4username'),
22)             ('3user', None, None, '3user', None),
23)             ('3user', None, None, '3user', '4username'),
24)             ('2logname', None, '2logname', None, None),
25)             ('2logname', None, '2logname', None, '4username'),
26)             ('2logname', None, '2logname', '3user', None),
27)             ('2logname', None, '2logname', '3user', '4username'),
28)             ('1vault_key', '1vault_key', None, None, None),
29)             ('1vault_key', '1vault_key', None, None, '4username'),
30)             ('1vault_key', '1vault_key', None, '3user', None),
31)             ('1vault_key', '1vault_key', None, '3user', '4username'),
32)             ('1vault_key', '1vault_key', '2logname', None, None),
33)             ('1vault_key', '1vault_key', '2logname', None, '4username'),
34)             ('1vault_key', '1vault_key', '2logname', '3user', None),
35)             ('1vault_key', '1vault_key', '2logname', '3user', '4username'),
36)         ],
37)     )
38)     def test200_get_vault_key(
39)         self,
40)         monkeypatch: pytest.MonkeyPatch,
41)         expected: str,
42)         vault_key: str | None,
43)         logname: str | None,
44)         user: str | None,
45)         username: str | None,
46)     ) -> None:
47)         priority_list = [
48)             ('VAULT_KEY', vault_key),
49)             ('LOGNAME', logname),
50)             ('USER', user),
51)             ('USERNAME', username),
52)         ]
53)         runner = click.testing.CliRunner(mix_stderr=False)
54)         with tests.isolated_vault_exporter_config(
55)             monkeypatch=monkeypatch, runner=runner
56)         ):
57)             for key, value in priority_list:
58)                 if value is not None:
59)                     monkeypatch.setenv(key, value)
60)             assert os.fsdecode(exporter.get_vault_key()) == expected
61) 
62)     @pytest.mark.parametrize(
63)         ['expected', 'path'],
64)         [
65)             ('/tmp', '/tmp'),
66)             ('~', os.path.curdir),
67)             ('~/.vault', None),
68)         ],
69)     )
70)     def test_210_get_vault_path(
71)         self,
72)         monkeypatch: pytest.MonkeyPatch,
73)         expected: str,
74)         path: str | None,
75)     ) -> None:
76)         runner = click.testing.CliRunner(mix_stderr=False)
77)         with tests.isolated_vault_exporter_config(
78)             monkeypatch=monkeypatch, runner=runner
79)         ):
80)             if path:
81)                 monkeypatch.setenv('VAULT_PATH', path)
82)             assert os.fsdecode(
83)                 os.path.realpath(exporter.get_vault_path())
84)             ) == os.path.realpath(os.path.expanduser(expected))
85) 
86)     def test_300_get_vault_key_without_envs(
87)         self, monkeypatch: pytest.MonkeyPatch
88)     ) -> None:
89)         monkeypatch.delenv('VAULT_KEY', raising=False)
90)         monkeypatch.delenv('LOGNAME', raising=False)
91)         monkeypatch.delenv('USER', raising=False)
92)         monkeypatch.delenv('USERNAME', raising=False)
93)         with pytest.raises(KeyError, match='VAULT_KEY'):
94)             exporter.get_vault_key()
95) 
96)     def test_310_get_vault_path_without_home(
97)         self, monkeypatch: pytest.MonkeyPatch
98)     ) -> None:
99)         monkeypatch.setattr(os.path, 'expanduser', lambda x: x)
100)         with pytest.raises(
101)             RuntimeError, match='[Cc]annot determine home directory'
102)         ):
103)             exporter.get_vault_path()
104) 
105) 
106) class Test002CLI:
107)     def test_300_invalid_format(
108)         self,
109)         monkeypatch: pytest.MonkeyPatch,
110)     ) -> None:
111)         runner = click.testing.CliRunner(mix_stderr=False)
112)         with tests.isolated_vault_exporter_config(
113)             monkeypatch=monkeypatch,
114)             runner=runner,

115)             vault_config=tests.VAULT_V03_CONFIG,
116)             vault_key=tests.VAULT_MASTER_KEY,

117)         ):
118)             result = runner.invoke(

119)                 cli.derivepassphrase_export,

120)                 ['-f', 'INVALID', 'VAULT_PATH'],
121)                 catch_exceptions=False,
122)             )
123)         assert isinstance(result.exception, SystemExit)
124)         assert result.exit_code
125)         assert result.stderr_bytes
126)         assert b'Invalid value for' in result.stderr_bytes
127)         assert b'-f' in result.stderr_bytes
128)         assert b'--format' in result.stderr_bytes
129)         assert b'INVALID' in result.stderr_bytes