def test_import_header(self):
line = 'h1,h2,h3,h4'
importer = Importer(None)
importer.import_line(line)
self.assertEqual('h1', importer.headers[0])
self.assertEqual('h2', importer.headers[1])
self.assertEqual('h3', importer.headers[2])
self.assertEqual('h4', importer.headers[3])
def test_import_valid_line(self):
header = 'h1,h2,h3,h4'
line = '0.2,3,76,-48'
importer = Importer(self.db)
importer.import_line(header)
importer.import_line(line)
result = self.db.report.find_one({'h1': 0.2})
self.assertIsNotNone(result)
self.assertEqual(result['h1'], 0.2)
self.assertEqual(result['h2'], 3)
self.assertEqual(result['h3'], 76)
self.assertEqual(result['h4'], -48)
def test_import_invalid_line(self):
header = 'h1,h2,h3,h4'
line1 = '0.2,3,76'
line2 = '34,2342,-98,111,223'
importer = Importer(self.db)
importer.import_line(header)
importer.import_line(line1)
result = self.db.report.find_one({'h1': 0.2})
self.assertIsNone(result)
importer.import_line(line2)
result = self.db.report.find_one({'h1': 34})
self.assertIsNone(result)
def test_empty_header_values_renamed_to_unknown(self):
header = 'h1,h2,,,\n'
importer = Importer(self.db)
importer.import_line(header)
self.assertListEqual(importer.headers, ['h1', 'h2', 'Unknown0','Unknown1', 'Unknown2'])
