counterweight/lolafect
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Test formatting

Browse source
This commit is contained in:
Pablo Martin 2023-01-23 18:21:23 +01:00
parent f75b832903
commit dd07cd7959
1 changed files with 21 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/test_integration/test_connections.py
View file

@ -36,13 +36,19 @@ def test_temporal_download_of_secret_file_works_properly_in_happy_path():
with _temp_secret_file_from_s3( with _temp_secret_file_from_s3(
TEST_LOLACONFIG.S3_BUCKET_NAME, TEST_LOLACONFIG.S3_BUCKET_NAME,
s3_file_key="env/env_prd.json", # Not a secret file, but then again, this is a test, s3_file_key="env/env_prd.json", # Not a secret file, but then again, this is a test,
local_temp_file_path=temp_file_name local_temp_file_path=temp_file_name,
) as temp: ) as temp:
temp_file_found_when_in_context_manager = pathlib.Path(temp).exists() temp_file_found_when_in_context_manager = pathlib.Path(temp).exists()
temp_file_missing_when_outside_context_manager = not pathlib.Path(temp_file_name).exists() temp_file_missing_when_outside_context_manager = not pathlib.Path(
temp_file_name
).exists()
assert (
temp_file_found_when_in_context_manager
and temp_file_missing_when_outside_context_manager
)
assert temp_file_found_when_in_context_manager and temp_file_missing_when_outside_context_manager
def test_temporal_download_of_secret_file_works_properly_even_with_exception(): def test_temporal_download_of_secret_file_works_properly_even_with_exception():
temp_file_name = "test_temp_file" temp_file_name = "test_temp_file"
@ -51,13 +57,18 @@ def test_temporal_download_of_secret_file_works_properly_even_with_exception():
with _temp_secret_file_from_s3( with _temp_secret_file_from_s3(
TEST_LOLACONFIG.S3_BUCKET_NAME, TEST_LOLACONFIG.S3_BUCKET_NAME,
s3_file_key="env/env_prd.json", # Not a secret file, but then again, this is a test, s3_file_key="env/env_prd.json", # Not a secret file, but then again, this is a test,
local_temp_file_path=temp_file_name local_temp_file_path=temp_file_name,
) as temp: ) as temp:
temp_file_found_when_in_context_manager = pathlib.Path(temp).exists() temp_file_found_when_in_context_manager = pathlib.Path(temp).exists()
raise Exception # Something nasty happens within the context manager raise Exception # Something nasty happens within the context manager
except: except:
pass # We go with the test, ignoring the forced exception pass # We go with the test, ignoring the forced exception
temp_file_missing_when_outside_context_manager = not pathlib.Path(temp_file_name).exists() temp_file_missing_when_outside_context_manager = not pathlib.Path(
temp_file_name
).exists()
assert temp_file_found_when_in_context_manager and temp_file_missing_when_outside_context_manager assert (
temp_file_found_when_in_context_manager
and temp_file_missing_when_outside_context_manager
)