def test_run_importer_no_args(self):
with self.assertRaises(SystemExit):
run()
def test_run_importer_non_cfpb_part_args(self):
"""The Part number must be on our whitelist"""
with self.assertRaises(ValueError):
run('9999')
with self.assertRaises(ValueError):
run('DROP TABLE')
def test_run_all(self, mock_importer):
run('ALL')
self.assertEqual(mock_importer.call_count, 11)
def test_run_all_with_local_file(self, mock_importer):
run('ALL', '/mock/local/file.xml')
self.assertEqual(mock_importer.call_count, 11)
def test_run_with_one_arg_calls_importer(self, mock_importer):
run('1002')
self.assertEqual(mock_importer.call_count, 1)
def test_run_works_with_local_file(self, mock_importer):
run('1002', '/mock/local/file.xml')
self.assertEqual(mock_importer.call_count, 1)
def test_run_importer_non_cfpb_part_args(self):
"""The Part number must be on our whitelist"""
with self.assertRaises(ValueError):
ecfr_importer.run('9999')
with self.assertRaises(ValueError):
ecfr_importer.run('DROP TABLE')
def test_run_importer_no_args(self):
with self.assertRaises(SystemExit):
ecfr_importer.run()
def test_run_all_with_local_file(self, mock_importer):
ecfr_importer.run('ALL', '/mock/local/file.xml')
self.assertEqual(mock_importer.call_count, 11)
def test_run_all(self, mock_importer):
ecfr_importer.run('ALL')
self.assertEqual(mock_importer.call_count, 11)
def test_run_with_one_arg_calls_importer(self, mock_importer):
ecfr_importer.run('1002')
self.assertEqual(mock_importer.call_count, 1)
