class ImportMysqlToVerticaTaskTest(unittest.TestCase): """Test for ImportMysqlToVerticaTask.""" def setUp(self): self.task = ImportMysqlToVerticaTask(exclude=('auth_user$', 'courseware_studentmodule*', 'oauth*')) @data( ('auth_user', True), ('auth_userprofile', False), ('courseware_studentmodule', True), ('courseware_studentmodulehistory', True), ('oauth2_accesstoken', True), ('oauth_provider_token', True), ) @unpack def test_should_exclude_table(self, table, expected): actual = self.task.should_exclude_table(table) self.assertEqual(actual, expected)
def setUp(self): self.task = ImportMysqlToVerticaTask(exclude=('auth_user$', 'courseware_studentmodule*', 'oauth*'))