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()
(options, args) = parser.parse_args() # basic checking if not options.input_file or not os.path.isfile(options.input_file): parser.error('--input-file invalid. try -h or --help') if not os.path.isfile(options.all_columns_file): parser.error('--column-headers-file invalid. try -h or --help') if not os.path.isfile(options.column_types_file): parser.error('--column-types-file invalid. try -h or --help') if not os.path.isfile(options.wanted_columns_file): parser.error('--wanted-columns-file invalid. try -h or --help') # the meat mysql_config = { 'user' : options.user, 'passwd' : options.password, 'host' : options.host, 'db' : options.db, } loader = MySQLLoader(mysql_config, options.column_types_file, options.wanted_columns_file) loader.connect() loader.initialize_table() with NamedCSVReader(options.input_file, options.all_columns_file) as reader: for row in reader: loader.insert(row) loader.commit()
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))