File: test_main.py

package info (click to toggle)
pass-audit 1.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 396 kB
  • sloc: python: 798; makefile: 41; sh: 17
file content (104 lines) | stat: -rw-r--r-- 3,586 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
# -*- encoding: utf-8 -*-
# pass-audit - test suite
# Copyright (C) 2018-2022 Alexandre PUJOL <alexandre@pujol.io>.
#

import os
import shutil
from unittest import mock

from pass_audit.passwordstore import PasswordStore
import tests


class TestMain(tests.Test):
    """Test common pass-audit main."""

    @classmethod
    def setUpClass(cls):
        cls.prefix = tests.prefix
        cls.store = PasswordStore(tests.prefix)
        os.environ['PASSWORD_STORE_DIR'] = tests.prefix
        os.environ['_PASSWORD_STORE_EXTENSION'] = 'audit'  # nosec

    def test_main_help(self):
        """Testing: pass audit --help."""
        cmd = ['--help']
        print(self.store.prefix)
        self.main(cmd, 0)

    def test_main_version(self):
        """Testing: pass audit --version."""
        cmd = ['--version']
        self.main(cmd, 0)

    def test_main_not_an_option(self):
        """Testing: pass audit --not-an-option."""
        cmd = ['--not-an-option', '-q']
        self.main(cmd, 2)

    def test_main_store_do_not_exist(self):
        """Testing: store not initialized."""
        cmd = ['Password/', '-v']
        os.rename(os.path.join(self.store.prefix, '.gpg-id'),
                  os.path.join(self.store.prefix, 'backup.gpg-id'))
        self.main(cmd, 1, 'no password store to audit.')
        os.rename(os.path.join(self.store.prefix, 'backup.gpg-id'),
                  os.path.join(self.store.prefix, '.gpg-id'))

    def test_main_invalid_keys(self):
        """Testing: invalid user ID."""
        os.rename(os.path.join(self.store.prefix, '.gpg-id'),
                  os.path.join(self.store.prefix, 'backup.gpg-id'))
        self.gpgids = ['']
        self._init_pass()
        self.main(
            [''], 1,
            'invalid user ID, password access aborted.')

        os.rename(os.path.join(self.store.prefix, 'backup.gpg-id'),
                  os.path.join(self.store.prefix, '.gpg-id'))

    def test_main_not_inside_pass(self):
        """Testing: python3 -m audit."""
        os.environ.pop('_PASSWORD_STORE_EXTENSION')
        self.main([''], 1, 'not running inside password-store.')
        os.environ['_PASSWORD_STORE_EXTENSION'] = 'audit'  # nosec

    def test_main_not_a_file(self):
        """Testing: pass audit not_a_file."""
        cmd = ['not_a_file']
        self.main(cmd, 1, 'not_a_file is not in the password store.')

    @mock.patch('requests.get', tests.mock_request)
    def test_main_passwords_notpwned(self):
        """Testing: pass audit Password/notpwned."""
        cmd = ['Password/notpwned']
        self.main(cmd)

    @mock.patch('requests.get', tests.mock_request)
    def test_main_passwords_pwned(self):
        """Testing: pass audit Password/pwned."""
        cmd = ['Password/pwned']
        self.main(cmd)

    @mock.patch('requests.get', tests.mock_request)
    def test_main_passwords_duplicate(self):
        """Testing: pass audit for duplicates."""
        shutil.copy(os.path.join(self.store.prefix, 'Password/good/1.gpg'),
                    os.path.join(self.store.prefix, 'Password/good/10.gpg'))
        cmd = ['Password/good']
        self.main(cmd)
        os.remove(os.path.join(self.store.prefix, 'Password/good/10.gpg'))

    @mock.patch('requests.get', tests.mock_request)
    def test_main_passwords_good(self):
        """Testing: pass audit Password/good."""
        cmd = ['Password/good']
        self.main(cmd)

    @mock.patch('requests.get', tests.mock_request)
    def test_main_passwords_all(self):
        """Testing: pass audit ."""
        cmd = ['']
        self.main(cmd)