pacifica/pacifica-policy

#!/usr/bin/python
# -*- coding: utf-8 -*-
"""Test the connections to external service."""
from unittest import TestCase
from json import dumps
import httpretty
from pacifica.policy.config import get_config
from pacifica.policy.root import Root


class TestMetaConnect(TestCase):
    """Test the metadata connection."""

    @httpretty.activate
    def test_meta_connect(self):
        """Test the try meta connect method."""
        httpretty.register_uri(
            httpretty.GET,
            get_config().get('metadata', 'status_url'),
            body=dumps([]),
            content_type='application/json'
        )
        Root.try_meta_connect(0)
        self.assertEqual(httpretty.last_request().method, 'GET')

    @httpretty.activate
    def test_meta_connect_failure(self):
        """Test the try meta connect method but fail."""
        httpretty.register_uri(
            httpretty.GET,
            get_config().get('metadata', 'status_url'),
            body='',
            status=500,
            content_type='application/json'
        )
        hit_exception = False
        try:
            Root.try_meta_connect(39)
        # pylint: disable=broad-except
        except Exception:
            hit_exception = True
        # pylint: enable=broad-except
        self.assertTrue(hit_exception)
        self.assertEqual(httpretty.last_request().method, 'GET')