class TestClass():
def setup(self):
self.loader = MySQLLoader(mysql_config, column_types_file, wanted_columns_file)
self.loader.connect()
self.loader.initialize_table()
with NamedCSVReader(hit_data_file, all_columns_file) as reader:
for row in reader:
self.loader.insert(row)
self.loader.commit()
def teardown(self):
self.loader.disconnect()
def test_Mysql_count(self):
eq_(self.loader.get_count(), hit_data_count)
def test_Mysql_count(self):
rows = self.loader.fetch_test()
assert_equal(rows[0] , (None , None , None , "pSZ6Ts6aAO5q7xQTc6" , 3))
assert_equal(rows[1] , (None , None , None , "KeTOFYW pOahZY5mgJe" , 184))
assert_equal(rows[2] , ("LOUDkf" , None , "wyS6oQwSLMX" , "f3mQ70GmvxLfkgf" , None))
assert_equal(rows[3] , ("923vp0svpt" , 2443444867 , None , " 30PYcy" , 123))
assert_equal(rows[4] , (None , None , "q" , None , None))
assert_equal(rows[5] , (None , None , None , None , 35))
assert_equal(rows[6] , ("Sh19bvV7QY0V" , None , None , None , None))
assert_equal(rows[7] , ("RN8PkRc" , None , None , "5zg3iYkKnKj" , None))
assert_equal(rows[8] , (None , 1633796482 , "UZshP7vdVll" , None , None))
assert_equal(rows[9] , (None , 3167102661 , None , "RKu" , 99))
assert_equal(rows[10] , ("Zcdwp7ac5p3PL08" , 1444331307 , None , None , None))
assert_equal(rows[11] , ("r" , 279371533 , "5iHmsNlZbYKzY" , "WUy9tj8HlAYlamQu0o" , None))
