zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-19 10:59:21 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update test_reports.py

Browse source
This commit is contained in:
zandercymatics 2023-11-28 12:35:25 -07:00
parent 37456d57cc
commit e23ef4f5eb
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 6 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/registrar/tests/test_reports.py
View file

@ -9,9 +9,7 @@ from registrar.utility.csv_export import export_domains_to_writer
from django.core.management import call_command from django.core.management import call_command
from unittest.mock import call, mock_open, patch from unittest.mock import call, mock_open, patch
from api.views import get_current_federal, get_current_full from api.views import get_current_federal, get_current_full
import logging
logger = logging.getLogger(__name__)
class CsvReportsTest(TestCase): class CsvReportsTest(TestCase):
"""Tests to determine if we are uploading our reports correctly""" """Tests to determine if we are uploading our reports correctly"""
@ -146,10 +144,10 @@ class CsvReportsTest(TestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
# Check that the response contains what we expect # Check that the response contains what we expect
file_content = b"".join(response.streaming_content).decode("utf-8") file_content = b"".join(response.streaming_content).decode("utf-8")
logger.info(f"data to expect fed: {file_content}")
expected_file_content = ( expected_file_content = (
"Domain name,Domain type,Agency,Organization name,City,State,Security Contact Email \r\n" "Domain name,Domain type,Agency,Organization name,City,State,Security Contact Email\n"
"cdomain1.gov,Federal - Executive,World War I Centennial Commission,,,, \r\n" "cdomain1.gov,Federal - Executive,World War I Centennial Commission,,,, \n"
"ddomain3.gov,Federal,Armed Forces Retirement Home,,,," "ddomain3.gov,Federal,Armed Forces Retirement Home,,,,"
) )
@ -164,11 +162,10 @@ class CsvReportsTest(TestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
# Check that the response contains what we expect # Check that the response contains what we expect
file_content = b"".join(response.streaming_content).decode("utf-8") file_content = b"".join(response.streaming_content).decode("utf-8")
logger.info(f"data to expect full: {file_content}")
expected_file_content = ( expected_file_content = (
"Domain name,Domain type,Agency,Organization name,City,State,Security Contact Email\r\n" "Domain name,Domain type,Agency,Organization name,City,State,Security Contact Email\n"
"cdomain1.gov,Federal - Executive,World War I Centennial Commission,,,, \r\n" "cdomain1.gov,Federal - Executive,World War I Centennial Commission,,,,\n"
"ddomain3.gov,Federal,Armed Forces Retirement Home,,,, \r\n" "ddomain3.gov,Federal,Armed Forces Retirement Home,,,,\n"
"adomain2.gov,Interstate,,,,," "adomain2.gov,Interstate,,,,,"
) )