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)