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_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))