core/tests/components/test_frontend.py

"""The tests for Home Assistant frontend."""
# pylint: disable=protected-access
import re
import unittest

import requests

import homeassistant.bootstrap as bootstrap
from homeassistant.components import frontend, http
from homeassistant.const import HTTP_HEADER_HA_AUTH

from tests.common import get_test_instance_port, get_test_home_assistant

API_PASSWORD = "test1234"
SERVER_PORT = get_test_instance_port()
HTTP_BASE_URL = "http://127.0.0.1:{}".format(SERVER_PORT)
HA_HEADERS = {HTTP_HEADER_HA_AUTH: API_PASSWORD}

hass = None


def _url(path=""):
    """Helper method to generate URLs."""
    return HTTP_BASE_URL + path


# pylint: disable=invalid-name
def setUpModule():
    """Initialize a Home Assistant server."""
    global hass

    hass = get_test_home_assistant()

    assert bootstrap.setup_component(
        hass, http.DOMAIN,
        {http.DOMAIN: {http.CONF_API_PASSWORD: API_PASSWORD,
                       http.CONF_SERVER_PORT: SERVER_PORT}})

    assert bootstrap.setup_component(hass, 'frontend')

    hass.start()


# pylint: disable=invalid-name
def tearDownModule():
    """Stop everything that was started."""
    hass.stop()
    frontend.PANELS = {}


class TestFrontend(unittest.TestCase):
    """Test the frontend."""

    def tearDown(self):
        """Stop everything that was started."""
        hass.block_till_done()

    def test_frontend_and_static(self):
        """Test if we can get the frontend."""
        req = requests.get(_url(""))
        self.assertEqual(200, req.status_code)

        # Test we can retrieve frontend.js
        frontendjs = re.search(
            r'(?P<app>\/static\/frontend-[A-Za-z0-9]{32}.html)',
            req.text)

        self.assertIsNotNone(frontendjs)
        req = requests.get(_url(frontendjs.groups(0)[0]))
        self.assertEqual(200, req.status_code)

    def test_404(self):
        """Test for HTTP 404 error."""
        self.assertEqual(404, requests.get(_url("/not-existing")).status_code)

    def test_we_cannot_POST_to_root(self):
        """Test that POST is not allow to root."""
        self.assertEqual(405, requests.post(_url("")).status_code)

    def test_states_routes(self):
        """All served by index."""
        req = requests.get(_url("/states"))
        self.assertEqual(200, req.status_code)

        req = requests.get(_url("/states/group.non_existing"))
        self.assertEqual(404, req.status_code)

        hass.states.set('group.existing', 'on', {'view': True})
        req = requests.get(_url("/states/group.existing"))
        self.assertEqual(200, req.status_code)