def test_get_tenant_from_cache_calls_add_correlation_info_to_message(self): add_correlation_info_to_message_func = MagicMock() with patch.object(correlator.cache_handler.TenantCache, 'get_tenant', self.get_tenant), \ patch('meniscus.correlation.correlator.' '_add_correlation_info_to_message', add_correlation_info_to_message_func): correlator._get_tenant_from_cache(self.tenant_id, self.message_token, self.src_msg) add_correlation_info_to_message_func.assert_called_once_with( self.tenant, self.src_msg)
def test_get_tenant_from_cache_calls_get_tenant_from_coordinator(self): get_tenant_from_coordinator_func = MagicMock() with patch.object(correlator.cache_handler.TenantCache, 'get_tenant', self.get_none), \ patch('meniscus.correlation.correlator.' '_get_tenant_from_coordinator', get_tenant_from_coordinator_func): correlator._get_tenant_from_cache(self.tenant_id, self.message_token, self.src_msg) get_tenant_from_coordinator_func.assert_called_once_with( self.tenant_id, self.message_token, self.src_msg)
def test_get_tenant_from_cache_calls_add_correlation_info_to_message(self): add_correlation_info_to_message_func = MagicMock() with patch.object(correlator.cache_handler.TenantCache, 'get_tenant', self.get_tenant), \ patch('meniscus.correlation.correlator.' '_add_correlation_info_to_message', add_correlation_info_to_message_func): correlator._get_tenant_from_cache(self.tenant_id, self.message_token, self.src_msg) add_correlation_info_to_message_func.assert_called_once_with( self.tenant, self.src_msg)
def test_get_tenant_from_cache_calls_get_tenant_from_coordinator(self): get_tenant_from_coordinator_func = MagicMock() with patch.object(correlator.cache_handler.TenantCache, 'get_tenant', self.get_none), \ patch('meniscus.correlation.correlator.' '_get_tenant_from_coordinator', get_tenant_from_coordinator_func): correlator._get_tenant_from_cache(self.tenant_id, self.message_token, self.src_msg) get_tenant_from_coordinator_func.assert_called_once_with( self.tenant_id, self.message_token, self.src_msg)