def test_purge_location_successfully(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 200
response.text = 'purged'
side_effect = mock.Mock()
side_effect.status_code = 404
side_effect.text = "Not Found"
requests.request.side_effect = [
response, side_effect, response, side_effect
]
purged = nginx.purge_location('myhost', '/foo/bar')
self.assertTrue(purged)
self.assertEqual(requests.request.call_count, 4)
expec_responses = []
for scheme in ['http', 'https']:
for header in self.cache_headers:
expec_responses.append(
mock.call(
'get',
'http://myhost:8089/purge/{}/foo/bar'.format(scheme),
headers=header,
timeout=2))
requests.request.assert_has_calls(expec_responses)
def test_add_session_ticket_success(self, requests, os_path):
nginx = Nginx({'CA_CERT': 'cert data'})
os_path.exists.return_value = True
response = mock.Mock()
response.status_code = 200
response.text = '\n\nticket was succsessfully added'
requests.request.return_value = response
nginx.add_session_ticket('host-1', 'random data', timeout=2)
requests.request.assert_called_once_with('post', 'https://host-1:8090/session_ticket', timeout=2,
data='random data', verify='/tmp/rpaas_ca.pem')
def test_wait_healthcheck_timeout(self, requests):
nginx = Nginx()
def side_effect(method, url, timeout, **params):
raise Exception('some error')
requests.request.side_effect = side_effect
with self.assertRaises(Exception):
nginx.wait_healthcheck('myhost.com', timeout=2)
self.assertGreaterEqual(requests.request.call_count, 2)
requests.request.assert_called_with('get', 'http://myhost.com:8089/healthcheck', timeout=2)
def test_wait_healthcheck_timeout(self, requests):
nginx = Nginx()
def side_effect(url, timeout):
raise Exception('some error')
requests.get.side_effect = side_effect
with self.assertRaises(Exception):
nginx.wait_healthcheck('myhost.com', timeout=2)
self.assertGreaterEqual(requests.get.call_count, 2)
requests.get.assert_called_with('http://myhost.com:8089/healthcheck', timeout=2)
def test_wait_healthcheck_timeout(self, requests):
nginx = Nginx()
def side_effect(url, timeout):
raise Exception("some error")
requests.get.side_effect = side_effect
with self.assertRaises(Exception):
nginx.wait_healthcheck("myhost.com", timeout=2)
self.assertGreaterEqual(requests.get.call_count, 2)
requests.get.assert_has_call("http://myhost.com:8089/healthcheck", timeout=2)
def test_purge_location_not_found(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 404
response.text = 'Not Found'
requests.get.side_effect = [response, response]
purged = nginx.purge_location('myhost.com', '/foo/bar')
self.assertFalse(purged)
self.assertEqual(requests.get.call_count, 2)
requests.get.assert_has_calls([mock.call('http://myhost.com:8089/purge/http/foo/bar', timeout=2),
mock.call('http://myhost.com:8089/purge/https/foo/bar', timeout=2)])
def test_add_session_ticket_success(self, requests, os_path):
nginx = Nginx({'CA_CERT': 'cert data'})
os_path.exists.return_value = True
response = mock.Mock()
response.status_code = 200
response.text = '\n\nticket was succsessfully added'
requests.request.return_value = response
nginx.add_session_ticket('host-1', 'random data', timeout=2)
requests.request.assert_called_once_with(
'post',
'https://host-1:8090/session_ticket',
timeout=2,
data='random data',
verify='/tmp/rpaas_ca.pem')
def test_purge_location_not_found(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 404
response.text = 'Not Found'
requests.get.side_effect = [response, response]
purged = nginx.purge_location('myhost.com', '/foo/bar')
self.assertFalse(purged)
self.assertEqual(requests.get.call_count, 2)
requests.get.assert_has_calls([
mock.call('http://myhost.com:8089/purge/http/foo/bar', timeout=2),
mock.call('http://myhost.com:8089/purge/https/foo/bar', timeout=2)
])
def test_init_config_location_url(self, requests):
def mocked_requests_get(*args, **kwargs):
class MockResponse:
def __init__(self, text, status_code):
self.text = text
self.status_code = status_code
if args[0] == 'http://my.com/default':
return MockResponse("my result default", 200)
elif args[0] == 'http://my.com/router':
return MockResponse("my result router", 200)
with mock.patch('rpaas.nginx.requests.get',
side_effect=mocked_requests_get) as requests_get:
nginx = Nginx({
'NGINX_LOCATION_TEMPLATE_DEFAULT_URL':
'http://my.com/default',
'NGINX_LOCATION_TEMPLATE_ROUTER_URL':
'http://my.com/router',
})
self.assertEqual(nginx.config_manager.location_template_default,
'my result default')
self.assertEqual(nginx.config_manager.location_template_router,
'my result router')
expected_calls = [
mock.call('http://my.com/default'),
mock.call('http://my.com/router')
]
requests_get.assert_has_calls(expected_calls)
def test_purge_location_preserve_path_successfully(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 200
response.text = 'purged'
requests.get.side_effect = [response]
purged = nginx.purge_location('myhost', 'http://example.com/foo/bar', True)
self.assertTrue(purged)
self.assertEqual(requests.get.call_count, 2)
expected_responses = []
for header in self.cache_headers:
expected_responses.append(mock.call('http://myhost:8089/purge/http://example.com/foo/bar',
headers=header, timeout=2))
requests.get.assert_has_calls(expected_responses)
def test_purge_location_preserve_path_successfully(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 200
response.text = 'purged'
requests.request.side_effect = [response]
purged = nginx.purge_location('myhost', 'http://example.com/foo/bar', True)
self.assertTrue(purged)
self.assertEqual(requests.request.call_count, 2)
expected_responses = []
for header in self.cache_headers:
expected_responses.append(mock.call('get', 'http://myhost:8089/purge/http://example.com/foo/bar',
headers=header, timeout=2))
requests.request.assert_has_calls(expected_responses)
def test_purge_location_not_found(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 404
response.text = 'Not Found'
requests.get.side_effect = [response, response, response, response]
purged = nginx.purge_location('myhost', '/foo/bar')
self.assertFalse(purged)
self.assertEqual(requests.get.call_count, 4)
expected_responses = []
for scheme in ['http', 'https']:
for header in self.cache_headers:
expected_responses.append(mock.call('http://myhost:8089/purge/{}/foo/bar'.format(scheme),
headers=header, timeout=2))
requests.get.assert_has_calls(expected_responses)
def test_wait_app_healthcheck(self, requests):
nginx = Nginx()
count = [0]
response = mock.Mock()
response.status_code = 200
response.text = '\n\nWORKING'
def side_effect(url, timeout):
count[0] += 1
if count[0] < 2:
raise Exception('some error')
return response
requests.get.side_effect = side_effect
nginx.wait_healthcheck('myhost.com', timeout=5, manage_healthcheck=False)
self.assertEqual(requests.get.call_count, 2)
requests.get.assert_called_with('http://myhost.com:8080/_nginx_healthcheck/', timeout=2)
def test_purge_location_successfully(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 200
response.text = 'purged'
side_effect = mock.Mock()
side_effect.status_code = 404
side_effect.text = "Not Found"
requests.get.side_effect = [response, side_effect]
purged = nginx.purge_location('myhost.com', '/foo/bar')
self.assertTrue(purged)
self.assertEqual(requests.get.call_count, 2)
requests.get.assert_has_calls([mock.call('http://myhost.com:8089/purge/http/foo/bar', timeout=2),
mock.call('http://myhost.com:8089/purge/https/foo/bar', timeout=2)])
def test_wait_app_healthcheck(self, requests):
nginx = Nginx()
count = [0]
response = mock.Mock()
response.status_code = 200
response.text = '\n\nWORKING'
def side_effect(method, url, timeout, **params):
count[0] += 1
if count[0] < 2:
raise Exception('some error')
return response
requests.request.side_effect = side_effect
nginx.wait_healthcheck('myhost.com', timeout=5, manage_healthcheck=False)
self.assertEqual(requests.request.call_count, 2)
requests.request.assert_called_with('get', 'http://myhost.com:8080/_nginx_healthcheck/', timeout=2)
def test_purge_location_not_found(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 404
response.text = 'Not Found'
requests.request.side_effect = [response, response, response, response]
purged = nginx.purge_location('myhost', '/foo/bar')
self.assertFalse(purged)
self.assertEqual(requests.request.call_count, 4)
expec_responses = []
for scheme in ['http', 'https']:
for header in self.cache_headers:
expec_responses.append(mock.call('get', 'http://myhost:8089/purge/{}/foo/bar'.format(scheme),
headers=header, timeout=2))
requests.request.assert_has_calls(expec_responses)
def test_wait_healthcheck(self, requests):
nginx = Nginx()
count = [0]
response = mock.Mock()
response.status_code = 200
response.text = 'WORKING'
def side_effect(url, timeout):
count[0] += 1
if count[0] < 2:
raise Exception('some error')
return response
requests.get.side_effect = side_effect
nginx.wait_healthcheck('myhost.com', timeout=5)
self.assertEqual(requests.get.call_count, 2)
requests.get.assert_has_call('http://myhost.com:8089/healthcheck', timeout=2)
def test_purge_location_preserve_path_successfully(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 200
response.text = 'purged'
requests.get.side_effect = [response]
purged = nginx.purge_location('myhost', 'http://example.com/foo/bar',
True)
self.assertTrue(purged)
self.assertEqual(requests.get.call_count, 1)
headers = {'Accept-Encoding': ''}
requests.get.assert_has_calls([
mock.call('http://myhost:8089/purge/http://example.com/foo/bar',
headers=headers,
timeout=2)
])
def test_init_config_location_url(self, requests):
rsp_get = requests.get.return_value
rsp_get.status_code = 200
rsp_get.text = 'my result'
nginx = Nginx({
'NGINX_LOCATION_TEMPLATE_URL': 'http://my.com/x',
})
self.assertEqual(nginx.config_manager.location_template, 'my result')
requests.get.assert_called_once_with('http://my.com/x')
def test_wait_healthcheck(self, requests):
nginx = Nginx()
count = [0]
response = mock.Mock()
response.status_code = 200
response.text = 'WORKING'
def side_effect(method, url, timeout, **params):
count[0] += 1
if count[0] < 2:
raise Exception('some error')
return response
requests.request.side_effect = side_effect
nginx.wait_healthcheck('myhost.com', timeout=5)
self.assertEqual(requests.request.call_count, 2)
requests.request.assert_called_with(
'get', 'http://myhost.com:8089/healthcheck', timeout=2)
def test_purge_location_successfully(self, requests):
nginx = Nginx()
response = mock.Mock()
response.status_code = 200
response.text = 'purged'
side_effect = mock.Mock()
side_effect.status_code = 404
side_effect.text = "Not Found"
requests.get.side_effect = [response, side_effect]
purged = nginx.purge_location('myhost.com', '/foo/bar')
self.assertTrue(purged)
self.assertEqual(requests.get.call_count, 2)
requests.get.assert_has_calls([
mock.call('http://myhost.com:8089/purge/http/foo/bar', timeout=2),
mock.call('http://myhost.com:8089/purge/https/foo/bar', timeout=2)
])
def test_init_config(self):
nginx = Nginx({
'NGINX_PURGE_PATH': '/2',
'NGINX_MANAGE_PORT': '4',
'NGINX_LOCATION_TEMPLATE_TXT': '5',
'NGINX_HEALTHCHECK_PATH': '6',
})
self.assertEqual(nginx.nginx_purge_path, '/2')
self.assertEqual(nginx.nginx_manage_port, '4')
self.assertEqual(nginx.config_manager.location_template, '5')
self.assertEqual(nginx.nginx_healthcheck_path, '6')
def test_wait_app_healthcheck_invalid_response(self, requests):
nginx = Nginx()
count = [0]
response = mock.Mock()
response.status_code = 200
response.text = '\nFAIL\n'
def side_effect(method, url, timeout, **params):
count[0] += 1
if count[0] < 2:
raise Exception('some error')
return response
requests.request.side_effect = side_effect
with self.assertRaises(NginxError):
nginx.wait_healthcheck('myhost.com',
timeout=5,
manage_healthcheck=False)
self.assertEqual(requests.request.call_count, 6)
requests.request.assert_called_with(
'get', 'http://myhost.com:8080/_nginx_healthcheck/', timeout=2)
def test_init_default(self):
nginx = Nginx()
self.assertEqual(nginx.nginx_manage_port, '8089')
self.assertEqual(nginx.nginx_purge_path, '/purge')
self.assertEqual(nginx.nginx_healthcheck_path, '/healthcheck')
self.assertEqual(nginx.config_manager.location_template, """
location {path} {{
proxy_set_header Host {host};
proxy_set_header X-Real-IP $remote_addr;
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
proxy_set_header X-Forwarded-Proto $scheme;
proxy_set_header X-Forwarded-Host $host;
proxy_pass http://{host}:80/;
proxy_redirect ~^http://{host}(:\d+)?/(.*)$ {path}$2;
}}
""")
def test_missing_ca_cert(self, requests):
nginx = Nginx()
with self.assertRaises(NginxError):
nginx.add_session_ticket('host-1', 'random data', timeout=2)
def test_init_default(self):
nginx = Nginx()
self.assertEqual(nginx.nginx_manage_port, '8089')
self.assertEqual(nginx.nginx_purge_path, '/purge')
self.assertEqual(nginx.nginx_healthcheck_path, '/healthcheck')
def test_missing_ca_cert(self, requests):
nginx = Nginx()
with self.assertRaises(NginxError):
nginx.add_session_ticket('host-1', 'random data', timeout=2)