pytest-localstack/tests/unit/test_utils.py at 08093383d6d55a4f9feb97b65690afd8c82b740c · mintel/pytest-localstack · GitHub

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
"""Unit tests for pytest_localstack.utils."""

import os
from unittest import mock

import pytest
from hypothesis import assume, given, settings
from hypothesis import strategies as st

from pytest_localstack import utils


def _get_env_var(name):
    return os.environ.get(name) or os.environ.get(name.upper(), "")


def _set_env_var(name, value):
    if value is None:
        os.environ.pop(name, None)
    else:
        os.environ[name] = value


@given(
    http_proxy=st.sampled_from([None, "http://proxy:3128"]),
    https_proxy=st.sampled_from([None, "http://proxy:3128"]),
    HTTP_PROXY=st.sampled_from([None, "http://proxy:3128"]),
    HTTPS_PROXY=st.sampled_from([None, "http://proxy:3128"]),
    no_proxy=st.sampled_from([None, "localhost,127.0.0.1", "localhost", "foobar"]),
    NO_PROXY=st.sampled_from([None, "localhost,127.0.0.1", "localhost", "foobar"]),
)
@settings(deadline=1000)
def test_check_proxy_env_vars(
    http_proxy, https_proxy, HTTP_PROXY, HTTPS_PROXY, no_proxy, NO_PROXY
):
    """Test pytest_localstack.utils.check_proxy_env_vars."""
    with mock.patch.dict(os.environ):
        # mock.patch.dict can't delete keys.
        # Patch os.environ manually.
        _set_env_var("http_proxy", http_proxy)
        _set_env_var("https_proxy", https_proxy)
        _set_env_var("HTTP_PROXY", HTTP_PROXY)
        _set_env_var("HTTPS_PROXY", HTTPS_PROXY)
        _set_env_var("no_proxy", no_proxy)
        _set_env_var("NO_PROXY", NO_PROXY)

        settings_match = (
            ((http_proxy or HTTP_PROXY) == (HTTP_PROXY or http_proxy))
            and ((https_proxy or HTTPS_PROXY) == (HTTPS_PROXY or https_proxy))
            and ((no_proxy or NO_PROXY) == (NO_PROXY or no_proxy))
        )
        has_http_proxy = bool(_get_env_var("http_proxy"))
        has_https_proxy = bool(_get_env_var("https_proxy"))
        good_no_proxy = "127.0.0.1" in _get_env_var("no_proxy")

        if (has_http_proxy or has_https_proxy) and not (
            settings_match and good_no_proxy
        ):
            with pytest.raises(UserWarning):
                utils.check_proxy_env_vars()
        else:
            utils.check_proxy_env_vars()


@given(
    string=st.text(),
    newline=st.sampled_from(["\n", "\r\n", "\r", "\n\r"]),
    num_newlines=st.integers(min_value=0, max_value=100),
    n=st.integers(min_value=-100, max_value=100),
)
def test_remove_newline(string, newline, num_newlines, n):
    assume(not (string.endswith("\n") or string.endswith("\r")))
    string_with_newlines = string + (newline * num_newlines)
    result = utils.remove_newline(string_with_newlines, n)
    assert result == string + (newline * (num_newlines - max(n, 0)))


def test_get_version_tuple():
    assert utils.get_version_tuple("1.2.3") == (1, 2, 3)
    with pytest.raises(ValueError):
        utils.get_version_tuple("latest")