File: util_test.py

package info (click to toggle)
python-certbot 4.0.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,688 kB
  • sloc: python: 21,764; makefile: 182; sh: 108
file content (40 lines) | stat: -rw-r--r-- 1,511 bytes parent folder | download
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
"""Tests for certbot.plugins.util."""
import sys
from unittest import mock

import pytest

from certbot.compat import os


def test_get_prefix():
    from certbot.plugins.util import get_prefixes
    assert get_prefixes('/a/b/c') == \
        [os.path.normpath(path) for path in ['/a/b/c', '/a/b', '/a', '/']]
    assert get_prefixes('/') == [os.path.normpath('/')]
    assert get_prefixes('a') == ['a']


@mock.patch("certbot.plugins.util.logger.debug")
def test_path_surgery(mock_debug):
    from certbot.plugins.util import path_surgery
    all_path = {"PATH": "/usr/local/bin:/bin/:/usr/sbin/:/usr/local/sbin/"}
    with mock.patch.dict('os.environ', all_path):
        with mock.patch('certbot.util.exe_exists') as mock_exists:
            mock_exists.return_value = True
            assert path_surgery("eg") is True
            assert mock_debug.call_count == 0
            assert os.environ["PATH"] == all_path["PATH"]
    if os.name != 'nt':
        # This part is specific to Linux since on Windows no PATH surgery is ever done.
        no_path = {"PATH": "/tmp/"}
        with mock.patch.dict('os.environ', no_path):
            path_surgery("thingy")
            assert mock_debug.call_count == (2 if os.name != 'nt' else 1)
            assert "Failed to find" in mock_debug.call_args[0][0]
            assert "/usr/local/bin" in os.environ["PATH"]
            assert "/tmp" in os.environ["PATH"]


if __name__ == "__main__":
    sys.exit(pytest.main(sys.argv[1:] + [__file__]))  # pragma: no cover