Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
 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')
Exemplo n.º 3
0
    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)
Exemplo n.º 4
0
    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)
Exemplo n.º 5
0
    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)
Exemplo n.º 6
0
    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)])
Exemplo n.º 7
0
 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')
Exemplo n.º 8
0
    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)
        ])
Exemplo n.º 9
0
    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)
Exemplo n.º 10
0
    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)
Exemplo n.º 11
0
    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)
Exemplo n.º 12
0
    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)
Exemplo n.º 13
0
    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)
Exemplo n.º 14
0
    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)])
Exemplo n.º 15
0
    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)
Exemplo n.º 16
0
    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)
Exemplo n.º 17
0
    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)
Exemplo n.º 18
0
    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)
        ])
Exemplo n.º 19
0
 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')
Exemplo n.º 20
0
    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)
Exemplo n.º 21
0
    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)
        ])
Exemplo n.º 22
0
 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')
Exemplo n.º 23
0
    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)
Exemplo n.º 24
0
    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;
}}
""")
Exemplo n.º 25
0
 def test_missing_ca_cert(self, requests):
     nginx = Nginx()
     with self.assertRaises(NginxError):
         nginx.add_session_ticket('host-1', 'random data', timeout=2)
Exemplo n.º 26
0
 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')
Exemplo n.º 27
0
 def test_missing_ca_cert(self, requests):
     nginx = Nginx()
     with self.assertRaises(NginxError):
         nginx.add_session_ticket('host-1', 'random data', timeout=2)