add tests

This commit is contained in:
Kenneth Kehl
2024-12-17 10:56:27 -08:00
parent 020af71574
commit c2d822b288
2 changed files with 16 additions and 1 deletions

View File

@@ -402,7 +402,12 @@ def extract_phones(job):
phone_index = 0
for item in first_row:
# Note: may contain a BOM and look like \ufeffphone number
if item.lower() in ["phone number", "\\ufeffphone number", "phone number\n"]:
if item.lower() in [
"phone number",
"\\ufeffphone number",
"\\ufeffphone number\n",
"phone number\n",
]:
break
phone_index = phone_index + 1

View File

@@ -219,6 +219,16 @@ def test_get_s3_file_makes_correct_call(notify_api, mocker):
2,
"5555555552",
),
(
# simulate file saved with utf8withbom
"\\ufeffPHONE NUMBER\n",
"5555555552",
),
(
# simulate file saved without utf8withbom
"\\PHONE NUMBER\n",
"5555555552",
),
],
)
def test_get_phone_number_from_s3(