added a second test to preserve the \r\n formatting for line endings

This commit is contained in:
Beverly Nguyen
2025-05-14 09:58:52 -07:00
parent ecbde5ab1f
commit bd884012d4
2 changed files with 16 additions and 1 deletions

View File

@@ -1,6 +1,6 @@
from unittest.mock import Mock
from app.s3_client.s3_csv_client import set_metadata_on_csv_upload
from app.s3_client.s3_csv_client import remove_blank_lines, set_metadata_on_csv_upload
def test_sets_metadata(client_request, mocker):
@@ -21,3 +21,9 @@ def test_sets_metadata(client_request, mocker):
MetadataDirective="REPLACE",
ServerSideEncryption="AES256",
)
def test_removes_blank_lines():
filedata = { "data": "variable,phone number\r\ntest,+15555555555\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n" }
file_data = remove_blank_lines(filedata)
assert file_data == {"data": "variable,phone number\r\ntest,+15555555555"}