mirror of
https://github.com/sartography/uva-covid19-testing-communicator.git
synced 2025-02-25 05:15:14 +00:00
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from communicator.errors import CommError
|
||
|
|
from communicator.services.ivy_service import IvyService
|
||
|
|
|
||
|
|
|
||
|
|
class IvyServiceTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_read_file_and_build_records(self):
|
||
|
|
data = IvyService.import_ivy_file('../data/results.csv')
|
||
|
|
self.assertEquals(3, len(data))
|
||
|
|
# Quick spot check on values
|
||
|
|
self.assertEquals("987654321", data[0]["Student ID"])
|
||
|
|
self.assertEquals("testpositive@virginia.edu", data[1]["Student Email"])
|
||
|
|
self.assertEquals("1142270225", data[2]["Test Result Code"])
|
||
|
|
records = []
|
||
|
|
for d in data:
|
||
|
|
records.append(IvyService.to_test_event_record(d))
|
||
|
|
self.assertEquals("987654321", records[0].student_id)
|
||
|
|
self.assertEquals("testpositive@virginia.edu", records[1].email)
|
||
|
|
self.assertEquals("1142270225", records[2].result_code)
|
||
|
|
|
||
|
|
def test_invalid_file(self):
|
||
|
|
with self.assertRaises(CommError):
|
||
|
|
data = IvyService.import_ivy_file('../data/incorrect.csv')
|
||
|
|
records = []
|
||
|
|
for d in data:
|
||
|
|
records.append(IvyService.to_test_event_record(d))
|
||
|
|
|