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)