def test_migrate_partition_skips_missing_destinations(self): del config.DESTINATIONS['crate_m'] partition_val = 13 source_shard = 'shn' destination_shard = 'crate' configs = [ worker.ChunkConfig('mid', data.TableConfig('TableA', 'filter_col_a', 'chunk_col_a', None), None, partition_val, '', source_shard, destination_shard, None, None, None, 'mysql', 'crate'), worker.ChunkConfig('mid', data.TableConfig('TableB', 'filter_col_b', 'chunk_col_b', None), None, partition_val, '', source_shard, destination_shard, None, None, None, 'mysql', 'crate'), worker.ChunkConfig('mid', data.TableConfig('WTableA', 'w_filter_col_a', 'w_chunk_col_a', None), None, partition_val, '', source_shard + '_w', destination_shard + '_w', None, None, None, 'mysql_w', 'crate_w'), worker.ChunkConfig('mid', data.TableConfig('WTableB', 'w_filter_col_b', 'w_chunk_col_b', None), None, partition_val, '', source_shard + '_w', destination_shard + '_w', None, None, None, 'mysql_w', 'crate_w'), ] self.expected_configs(configs) namespace = '' self.mox.ReplayAll() queuer.migrate_partition(partition_val, namespace, source_shard, destination_shard)
def test_migrate_partition_namespace(self): partition_val = 13 source_shard = 'shn' destination_shard = 'crate' namespace = 'ns' configs = [ worker.ChunkConfig('mid', data.TableConfig('TableA', 'filter_col_a', 'chunk_col_a', None), None, partition_val, 'ns', source_shard, destination_shard, None, None, None, 'ns', 'ns'), worker.ChunkConfig('mid', data.TableConfig('TableB', 'filter_col_b', 'chunk_col_b', None), None, partition_val, 'ns', source_shard, destination_shard, None, None, None, 'ns', 'ns'), worker.ChunkConfig('mid', data.TableConfig('WTableA', 'w_filter_col_a', 'w_chunk_col_a', None), None, partition_val, 'ns_w', source_shard + '_w', destination_shard + '_w', None, None, None, 'ns_w', 'ns_w'), worker.ChunkConfig('mid', data.TableConfig('WTableB', 'w_filter_col_b', 'w_chunk_col_b', None), None, partition_val, 'ns_w', source_shard + '_w', destination_shard + '_w', None, None, None, 'ns_w', 'ns_w'), worker.ChunkConfig('mid', data.TableConfig('MTableA', 'm_filter_col_a', 'm_chunk_col_a', None), None, partition_val, 'ns_m', source_shard + '_m', destination_shard + '_m', None, None, None, 'ns_m', 'ns_m'), worker.ChunkConfig('mid', data.TableConfig('MTableB', 'm_filter_col_b', 'm_chunk_col_b', None), None, partition_val, 'ns_m', source_shard + '_m', destination_shard + '_m', None, None, None, 'ns_m', 'ns_m'), ] self.expected_configs(configs) self.mox.ReplayAll() queuer.migrate_partition(partition_val, namespace, source_shard, destination_shard)