fix(test-csv): fix test using tempfile (#4356)

This commit is contained in:
Pedro Martín
2024-07-02 15:16:12 +02:00
committed by GitHub
parent ca033745c9
commit da547b2bbe
2 changed files with 21 additions and 17 deletions
View File
+21 -17
View File
@@ -1,3 +1,4 @@
import tempfile
from datetime import datetime
from io import StringIO, TextIOWrapper
from typing import List
@@ -160,29 +161,32 @@ class TestCSV:
mock_output_class.batch_write_data_to_file = MagicMock()
findings = [MagicMock(spec=Finding)]
file_path = "test_file.txt"
# Instantiate the mock class
output_instance = mock_output_class(
findings, create_file_descriptor=True, file_path=file_path
)
# Create a temporary file
with tempfile.NamedTemporaryFile() as file:
file_path = file.name
# Check that transform was called once
output_instance.transform.assert_called_once_with(findings)
# Instantiate the mock class
output_instance = mock_output_class(
findings, create_file_descriptor=True, file_path=file_path
)
# Check that create_file_descriptor was called and the file descriptor was created
assert output_instance.file_descriptor is not None
# Check that transform was called once
output_instance.transform.assert_called_once_with(findings)
# Check the type
assert isinstance(output_instance.file_descriptor, TextIOWrapper)
# Check that create_file_descriptor was called and the file descriptor was created
assert output_instance.file_descriptor is not None
# Assuming we need to call batch_write_data_to_file for this test
output_instance.batch_write_data_to_file(output_instance.file_descriptor)
# Check the type
assert isinstance(output_instance.file_descriptor, TextIOWrapper)
# Check that batch_write_data_to_file was called once
output_instance.batch_write_data_to_file.assert_called_once_with(
output_instance.file_descriptor
)
# Assuming we need to call batch_write_data_to_file for this test
output_instance.batch_write_data_to_file(output_instance.file_descriptor)
# Check that batch_write_data_to_file was called once
output_instance.batch_write_data_to_file.assert_called_once_with(
output_instance.file_descriptor
)
def test_write_csv_with_dict(self):
headers = ["provider", "account", "check_id"]