def test_handle_session_persistence_with_app_cookie_with_no_sess_pers_type(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
self.pool.session_persistence = APP_COOKIE_SESS_PERS_WITH_NO_TYPE
mock_session_persist.execute(VTHUNDER, self.pool)
self.client_mock.slb.template.cookie_persistence.delete.assert_not_called()
self.client_mock.slb.template.cookie_persistence.create.assert_not_called()
def test_revert_handle_session_persistence_with_source_ip(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
self.pool.session_persistence = SOURCE_IP_SESS_PERS
mock_session_persist.execute(VTHUNDER, self.pool)
self.client_mock.slb.template.src_ip_persistence.delete.assert_called_with(
POOL.id)
def test_revert_handle_session_persistence_with_http_cookie(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
self.pool.session_persistence = HTTP_COOKIE_SESS_PERS
mock_session_persist.execute(VTHUNDER, self.pool)
self.client_mock.slb.template.cookie_persistence.delete.assert_called_with(
POOL.id)
def _get_sess_pers_subflow(self, pool_task):
get_pool_create_with_sess_pers = graph_flow.Flow(a10constants.HANDLE_SESS_PERS)
sess_pers = persist_tasks.HandleSessionPersistenceDelta(
requires=[a10constants.VTHUNDER, constants.POOL])
get_pool_create_with_sess_pers.add(pool_task, sess_pers)
get_pool_create_with_sess_pers.link(pool_task, sess_pers,
decider=self._is_sess_pers_decider)
return get_pool_create_with_sess_pers
def test_handle_session_persistence_with_source_ip_with_no_sess_pers(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
mock_session_persist.execute(VTHUNDER, self.pool)
self.client_mock.slb.template.src_ip_persistence.delete.assert_called_with(
POOL.id)
self.client_mock.slb.template.src_ip_persistence.create.assert_not_called(
)
def test_handle_session_persistence_with_app_cookie(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
self.pool.session_persistence = APP_COOKIE_SESS_PERS
mock_session_persist.execute(VTHUNDER, self.pool)
self.client_mock.slb.template.cookie_persistence.delete.assert_called_with(
POOL.id)
self.client_mock.slb.template.cookie_persistence.create.assert_called_with(
POOL.id, cookie_name=a10_test_constants.MOCK_COOKIE_NAME)
def test_unset_session_persistence(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
self.pool.session_persistence = None
mock_session_persist.execute(VTHUNDER, self.pool)
self.client_mock.slb.template.src_ip_persistence.delete.assert_called_with(
POOL.id)
self.client_mock.slb.template.src_ip_persistence.create.assert_not_called(
)
def test_handle_session_persistence_exception(self):
mock_session_persist = persist_tasks.HandleSessionPersistenceDelta()
mock_session_persist.axapi_client = self.client_mock
self.client_mock.side_effect = acos_errors.Exists
mock_session_persist.execute(VTHUNDER, self.pool)
