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*'))
