def setUp(self):
super(MySqlExporterTest, self).setUp()
self.constants_mock = constants_mock.configure()
modules = {"mysql_exporter_constants": self.constants_mock}
self.patches = mock.patch.dict("sys.modules", modules)
self.patches.start()
import mysql_exporter
self.mysql_exporter = mysql_exporter
def setUp(self):
super(CheckpointManangerTest, self).setUp()
self.constants_mock = constants_mock.configure()
modules = {"cloudsql_importer_constants": self.constants_mock}
self.patches = mock.patch.dict("sys.modules", modules)
self.patches.start()
import checkpoint_manager
self.checkpoint_manager = checkpoint_manager
self.expected_checkpoint_for_function = {
"force_drop_database":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"fail_if_database_exists":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"create_database":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"load_schema":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"create_temp_folder":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"generate_restore_secondary_indexes_commands":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"drop_secondary_indexes":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"chunk_data":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"load_data":
checkpoint_manager._build_checkpoint_stage(
"LOAD_DATA_CHECKPOINT", self.constants_mock.DATABASE_NAME,
self.constants_mock.TEMP_FOLDER),
"restore_secondary_indexes":
checkpoint_manager._build_checkpoint_stage(
"RESTORE_INDEX_CHECKPOINT", self.constants_mock.DATABASE_NAME,
self.constants_mock.TEMP_FOLDER),
"verify_checksum":
checkpoint_manager._build_checkpoint_stage(
"VERIFY_CHECKSUM_CHECKPOINT", self.constants_mock.DATABASE_NAME,
self.constants_mock.TEMP_FOLDER),
"drop_database":
checkpoint_manager.NO_CHECKPOINT_STAGE,
"cleanup":
checkpoint_manager.NO_CHECKPOINT_STAGE
}
def setUp(self):
super(SqlWrapperTest, self).setUp()
self.constants_mock = constants_mock.configure()
def setUp(self):
super(ChunkerTest, self).setUp()
self.constants_mock = constants_mock.configure()