dcarrillo/digaws
1
0
Fork 0
mirror of https://github.com/dcarrillo/digaws.git synced 2026-01-02 08:07:07 +00:00
Code Issues Projects Releases Wiki Activity

chore: Use black formatter (#9)

Browse Source
This commit is contained in:
Daniel Carrillo
2024-06-08 13:43:41 +02:00
committed by Daniel Carrillo
parent e5685511ba
commit a90dbd0123
8 changed files with 240 additions and 235 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
tests/test_get_aws_ip_ranges_json.py
View File

@@ -1,12 +1,10 @@
import json
import os
import digaws.digaws as digaws
import pytest
import requests
import digaws.digaws as digaws
import tests
@@ -28,61 +26,61 @@ def create_cache_dir(fs):
digaws.CACHE_DIR.mkdir(parents=True)
@pytest.mark.parametrize('fs', [[None, [digaws]]], indirect=True)
@pytest.mark.parametrize("fs", [[None, [digaws]]], indirect=True)
def test_get_aws_ip_ranges_cached_valid_file(mocker, fs, create_cache_dir) -> None:
with open(digaws.CACHE_FILE, 'w') as out:
with open(digaws.CACHE_FILE, "w") as out:
out.write(tests.AWS_IP_RANGES)
response = requests.Response
response.status_code = 304
mocker.patch('requests.get', return_value=response)
mocker.patch("requests.get", return_value=response)
result = digaws.get_aws_ip_ranges()
assert result['syncToken'] == '1608245058'
assert result["syncToken"] == "1608245058"
@pytest.mark.parametrize('fs', [[None, [digaws]]], indirect=True)
@pytest.mark.parametrize("fs", [[None, [digaws]]], indirect=True)
def test_get_aws_ip_ranges_cached_invalid_file(mocker, fs, create_cache_dir) -> None:
with open(digaws.CACHE_FILE, 'w'):
with open(digaws.CACHE_FILE, "w"):
pass
response = requests.Response
response.status_code = 304
mocker.patch('requests.get', return_value=response)
mocker.patch("requests.get", return_value=response)
with pytest.raises(digaws.CachedFileException):
digaws.get_aws_ip_ranges()
@pytest.mark.parametrize('fs', [[None, [digaws]]], indirect=True)
@pytest.mark.parametrize("fs", [[None, [digaws]]], indirect=True)
def test_get_aws_ip_ranges_cached_deprecated_file(monkeypatch, fs, create_cache_dir) -> None:
with open(digaws.CACHE_FILE, 'w'):
with open(digaws.CACHE_FILE, "w"):
pass
digaws.CACHE_FILE.touch()
os.utime(digaws.CACHE_FILE, times=(0, 0))
monkeypatch.setattr(requests, 'get', mock_get)
monkeypatch.setattr(requests, "get", mock_get)
result = digaws.get_aws_ip_ranges()
assert result['syncToken'] == '1608245058'
assert result["syncToken"] == "1608245058"
@pytest.mark.parametrize('fs', [[None, [digaws]]], indirect=True)
@pytest.mark.parametrize("fs", [[None, [digaws]]], indirect=True)
def test_get_aws_ip_ranges_no_file(monkeypatch, fs, create_cache_dir) -> None:
monkeypatch.setattr(requests, 'get', mock_get)
monkeypatch.setattr(requests, "get", mock_get)
result = digaws.get_aws_ip_ranges()
assert result['syncToken'] == '1608245058'
assert result["syncToken"] == "1608245058"
@pytest.mark.parametrize('fs', [[None, [digaws]]], indirect=True)
@pytest.mark.parametrize("fs", [[None, [digaws]]], indirect=True)
def test_get_aws_ip_ranges_invalid_status(mocker, fs, create_cache_dir) -> None:
response = requests.Response
response.status_code = 301
mocker.patch('requests.get', return_value=response)
mocker.patch("requests.get", return_value=response)
with pytest.raises(digaws.UnexpectedRequestException) as e:
digaws.get_aws_ip_ranges()
assert e.match('^Unexpected response from')
assert e.match("^Unexpected response from")