def test_should_clear_state_on_response(self):
request = MagicMock(id='10')
middleware = VclRefreshMiddleware()
middleware.process_request(request)
assert_equals("test", middleware.process_response(request, "test"))
assert_equals({}, VclRefreshState.refresh)
def test_should_clear_state_on_response(self):
request = MagicMock(id='10')
middleware = VclRefreshMiddleware()
middleware.process_request(request)
assert_equals("test", middleware.process_response(request, "test"))
assert_equals({}, VclRefreshState.refresh)
def test_should_return_error_message_in_tastypie_if_exception_while_loading_vcl(self):
with patch('vaas.cluster.cluster.load_vcl_task.delay', side_effect=Exception('load vcl failed')):
request = MagicMock(id='10', session={})
middleware = VclRefreshMiddleware()
VclRefreshState.set_refresh('10', [MagicMock(id='1')])
response = middleware.process_response(request, HttpCreated())
assert_true(isinstance(response, HttpApplicationError))
assert_equals(b'Exception: load vcl failed', response.content)
def test_should_return_error_message_if_exception_while_loading_vcl(self):
with patch('vaas.cluster.cluster.load_vcl_task.delay', side_effect=Exception('load vcl failed')):
request = MagicMock(id='10', session={})
middleware = VclRefreshMiddleware()
VclRefreshState.set_refresh('10', [MagicMock(id='1')])
middleware.process_response(request, None)
assert_true('error_message' in request.session)
assert_equals('Exception: load vcl failed', request.session['error_message'])
def test_should_return_error_message_in_tastypie_if_exception_while_loading_vcl(self):
with patch.object(VarnishCluster, 'load_vcl', side_effect=Exception('load vcl failed')):
request = MagicMock(id='10', session={})
middleware = VclRefreshMiddleware()
VclRefreshState.set_refresh('10', ['cluster-1'])
response = middleware.process_response(request, HttpCreated())
assert_true(isinstance(response, HttpApplicationError))
assert_equals('Exception: load vcl failed', response.content)
def test_should_return_error_message_if_exception_while_loading_vcl(self):
with patch.object(VarnishCluster, 'load_vcl', side_effect=Exception('load vcl failed')):
request = MagicMock(id='10', session={})
middleware = VclRefreshMiddleware()
VclRefreshState.set_refresh('10', ['cluster-1'])
middleware.process_response(request, None)
assert_true('error_message' in request.session)
assert_equals('Exception: load vcl failed', request.session['error_message'])
def test_should_get_list_vcl_refresh_state_for_request_id(self):
request = MagicMock(id='10')
middleware = VclRefreshMiddleware()
assert_equals(None, middleware.process_request(request))
assert_equals({'10': []}, VclRefreshState.refresh)
middleware.process_response(request, None)
def test_should_get_list_vcl_refresh_state_for_request_id(self):
request = MagicMock(id='10')
middleware = VclRefreshMiddleware()
assert_equals(None, middleware.process_request(request))
assert_equals({'10': []}, VclRefreshState.refresh)
middleware.process_response(request, None)
def test_should_not_refresh_vcl_on_response_if_empty_cluster_list(self):
request = MagicMock(id='10')
middleware = VclRefreshMiddleware()
middleware.process_request(request)
clusters = []
VclRefreshState.set_refresh(request.id, clusters)
with patch.object(VarnishCluster, '__init__', return_value=None):
with patch.object(
VarnishCluster,
'load_vcl',
return_value=None
) as load_vcl_mock:
middleware.process_response(request, None)
assert_equals(0, len(load_vcl_mock.call_args_list))
def test_should_not_refresh_vcl_on_response_if_empty_cluster_list(self):
request = MagicMock(id='10')
middleware = VclRefreshMiddleware()
middleware.process_request(request)
clusters = []
VclRefreshState.set_refresh(request.id, clusters)
with patch.object(VarnishCluster, '__init__', return_value=None):
with patch.object(VarnishCluster, 'load_vcl',
return_value=None) as load_vcl_mock:
middleware.process_response(request, None)
assert_equals(0, len(load_vcl_mock.call_args_list))
