def test_import_data(self): '''test importing data files''' dbs_cleanup() LOGGER.info('Test import_data') result_count, result_errors = import_data('csv_files', 'product_file.csv', 'customer_file.csv', 'rentals_file.csv') LOGGER.info('test_import_data, result_count = %s, result_errors = %s', str(result_count), str(result_errors)) self.assertEqual(result_count, (906, 906, 906)) self.assertEqual(result_errors, (0, 0, 0)) # err check filenotfound result_count, result_errors = import_data('csv_files', 'product_file.csvXXX', 'customer_file.csvXXX', 'rentals_file.csvXXX') LOGGER.info('test_import_data, result_count = %s, result_errors = %s', str(result_count), str(result_errors)) self.assertEqual(result_count, (0, 0, 0)) self.assertEqual(result_errors, (1, 1, 1)) LOGGER.info('test_import_data completed')
def test_import_customers(self): '''test import customers''' LOGGER.info('Test import_customers_started') stats_customers_imported = import_customers('csv_files', 'customer_file.csv') self.assertEqual(stats_customers_imported[0], (905)) stats_customers_imported = import_customers( 'csv_files', 'customer_file_XXX.csv', ) self.assertRaises(FileNotFoundError) dbs_cleanup() LOGGER.info('test_import_customers completed')
def setUp(self): ''' test setup ''' LOGGER.info('setup start') dropped_status = dbs_cleanup() LOGGER.info('test setup, dbs cleanup status = %s', dropped_status) ''' import data files ''' LOGGER.info('setup import data file to database') import_data('csv_files', 'product_file.csv', 'customer_file.csv', 'rentals_file.csv') LOGGER.info('test setup completed')
def test_main(self): ''' test main ''' LOGGER.info('Test main started') dbs_cleanup() results = main() #check products import self.assertEqual(results[0][0], 905) self.assertEqual(results[0][1], 0) self.assertEqual(results[0][2], 905) self.assertEqual(results[0][4], 0) #check customers import self.assertEqual(results[1][0], 905) self.assertEqual(results[1][1], 0) self.assertEqual(results[1][2], 905) self.assertEqual(results[1][4], 0) LOGGER.info('main() tests completed')
def test_dbs_cleanup(self): '''test drop dbs''' pass result_status = dbs_cleanup() expected_status = 'databases dropped' self.assertEqual(result_status, expected_status)