def test_should_pause_write(self): api.set_shard_at_rest('dummy', 1, "dest1/test_sharding") api.set_shard_to_migration_status( 'dummy', 1, api.ShardStatus.POST_MIGRATION_PAUSED_AT_DESTINATION) self.assertTrue(operations._should_pause_write("dummy", {'x': 1}))