def test_should_not_raise_any_exception_when_switching_to_another_cluster_version(self):
senza_mock = SenzaWrapper(CONFIG)
senza_switch_mock = senza_mock.switch_traffic = MagicMock()
controller = ClusterBootstrapController(base_url=BASE_URL, stack_name=STACK_NAME,
sharding_level=SHARDING_LEVEL,
replication_factor=REPLICATION_FACTOR,
image_version=IMAGE_VERSION, oauth_token=OAUTH_TOKEN,
senza_wrapper=senza_mock)
controller.switch_on_traffic()
senza_switch_mock.assert_called_once_with(STACK_NAME, INITIAL_BOOTSTRAP_VERSION, 100)
def test_should_not_raise_any_exception_when_switching_to_another_cluster_version(
self):
senza_mock = SenzaWrapper(CONFIG)
senza_switch_mock = senza_mock.switch_traffic = MagicMock()
controller = ClusterBootstrapController(
base_url=BASE_URL,
stack_name=STACK_NAME,
sharding_level=SHARDING_LEVEL,
replication_factor=REPLICATION_FACTOR,
image_version=IMAGE_VERSION,
oauth_token=OAUTH_TOKEN,
senza_wrapper=senza_mock)
controller.switch_on_traffic()
senza_switch_mock.assert_called_once_with(STACK_NAME,
INITIAL_BOOTSTRAP_VERSION,
100)
