darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-03 18:08:24 -04:00
Code Issues Packages Projects Releases Wiki Activity

Refactor generate csv test to expect a stream (generator)

Browse Source
This commit is contained in:
Imdad Ahad
2017-01-06 17:43:46 +00:00
parent e28d009fa7
commit 1036f365fa
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/app/test_utils.py
View File

@@ -56,8 +56,7 @@ def test_generate_csv_from_notifications(
set_status=status set_status=status
)['notifications'] )['notifications']
) )
for row in DictReader(StringIO(next(csv_content))):
for row in DictReader(StringIO(csv_content)):
assert row['Time'] == 'Friday 01 January 2016 at 15:09' assert row['Time'] == 'Friday 01 January 2016 at 15:09'
assert row['Status'] == expected_status assert row['Status'] == expected_status