def test_handle_when_req_result_is_not_in_cache(self):
# Given
next_response = TestResponse(success=True)
async def next_handle(req):
return next_response
mock_test_pipeline = MagicMock()
mock_test_pipeline.handle = next_handle
mock_cache_provider = MagicMock()
mock_cache_provider.get.return_value = None
cache_pipeline = CachePipeline(mock_cache_provider)
cache_pipeline.set_next(mock_test_pipeline)
test_request = TestRequestWithCacheable("cache_key", 1, CacheType.DISTRIBUTED)
# When
response = self.async_loop(cache_pipeline.handle(test_request))
# Then
assert response is not None
assert isinstance(response, TestResponse)
assert response == next_response
mock_cache_provider.get.assert_called_once_with(test_request.get_cache_key())
mock_cache_provider.add.assert_called_once_with(test_request.get_cache_key(),
SerializerFactory.get_serializer().dumps(next_response),
test_request.get_cache_duration())
def test_handle_when_next_handle_is_not_callable(self):
# Given
mock_test_pipeline = MagicMock()
mock_test_pipeline.handle = 1
cache_pipeline = CachePipeline(None)
cache_pipeline.set_next(mock_test_pipeline)
# When
with self.assertRaises(Exception) as context:
self.async_loop(cache_pipeline.handle(TestRequest()))
# Then
assert context.exception.args[0] == 'handle_function_of_next_pipeline_is_not_valid_for_cache_pipeline'
def test_handle_when_req_result_is_in_cache(self):
# Given
next_response = TestResponse(success=True)
mock_cache_provider = MagicMock()
mock_cache_provider.get.return_value = SerializerFactory.get_serializer().dumps(next_response)
cache_pipeline = CachePipeline(mock_cache_provider)
cache_pipeline.set_next(TestPipeline(True))
# When
response = self.async_loop(
cache_pipeline.handle(TestRequestWithCacheable("cache_key", 1, CacheType.DISTRIBUTED)))
# Then
assert response is not None
assert mock_cache_provider.get.call_count == 1
def test_handle_continue_when_cache_provider_is_none(self):
# Given
next_response = TestResponse(success=True)
async def next_handle(req):
return next_response
mock_test_pipeline = MagicMock()
mock_test_pipeline.handle = next_handle
cache_pipeline = CachePipeline(None)
cache_pipeline.set_next(mock_test_pipeline)
# When
response = self.async_loop(cache_pipeline.handle(TestRequest()))
# Then
assert response is not None
assert response == next_response
def test_handle_when_req_cache_key_is_empty(self):
# Given
next_response = TestResponse(success=True)
async def next_handle(req):
return next_response
mock_test_pipeline = MagicMock()
mock_test_pipeline.handle = next_handle
cache_pipeline = CachePipeline(FakeCacheProvider())
cache_pipeline.set_next(mock_test_pipeline)
# When
response = self.async_loop(cache_pipeline.handle(TestRequestWithCacheable("", 1, CacheType.DISTRIBUTED)))
# Then
assert response is not None
assert response == next_response
def test_handle_when_req_cache_type_is_none(self):
# Given
next_response = TestResponse(success=True)
async def next_handle(req):
return next_response
cache_pipeline = CachePipeline(FakeCacheProvider())
mock_test_pipeline = MagicMock()
mock_test_pipeline.handle = next_handle
cache_pipeline.set_next(mock_test_pipeline)
test_request = TestRequestWithCacheable("cache_key", 1, CacheType.NONE)
# When
response = self.async_loop(cache_pipeline.handle(test_request))
# Then
assert response is not None
assert response == next_response