core/tests/components/sensor/test_bom.py

"""The tests for the BOM Weather sensor platform."""
import json
import re
import unittest
from unittest.mock import patch
from urllib.parse import urlparse

import requests
from tests.common import (
    assert_setup_component, get_test_home_assistant, load_fixture)

from homeassistant.components import sensor
from homeassistant.setup import setup_component

VALID_CONFIG = {
    'platform': 'bom',
    'station': 'IDN60901.94767',
    'name': 'Fake',
    'monitored_conditions': [
        'apparent_t',
        'press',
        'weather'
    ]
}


def mocked_requests(*args, **kwargs):
    """Mock requests.get invocations."""
    class MockResponse:
        """Class to represent a mocked response."""

        def __init__(self, json_data, status_code):
            """Initialize the mock response class."""
            self.json_data = json_data
            self.status_code = status_code

        def json(self):
            """Return the json of the response."""
            return self.json_data

        @property
        def content(self):
            """Return the content of the response."""
            return self.json()

        def raise_for_status(self):
            """Raise an HTTPError if status is not 200."""
            if self.status_code != 200:
                raise requests.HTTPError(self.status_code)

    url = urlparse(args[0])
    if re.match(r'^/fwo/[\w]+/[\w.]+\.json', url.path):
        return MockResponse(json.loads(load_fixture('bom_weather.json')), 200)

    raise NotImplementedError('Unknown route {}'.format(url.path))


class TestBOMWeatherSensor(unittest.TestCase):
    """Test the BOM Weather sensor."""

    def setUp(self):
        """Set up things to be run when tests are started."""
        self.hass = get_test_home_assistant()
        self.config = VALID_CONFIG

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

    @patch('requests.get', side_effect=mocked_requests)
    def test_setup(self, mock_get):
        """Test the setup with custom settings."""
        with assert_setup_component(1, sensor.DOMAIN):
            self.assertTrue(setup_component(self.hass, sensor.DOMAIN, {
                'sensor': VALID_CONFIG}))

        fake_entities = [
            'bom_fake_feels_like_c',
            'bom_fake_pressure_mb',
            'bom_fake_weather']

        for entity_id in fake_entities:
            state = self.hass.states.get('sensor.{}'.format(entity_id))
            self.assertIsNotNone(state)

    @patch('requests.get', side_effect=mocked_requests)
    def test_sensor_values(self, mock_get):
        """Test retrieval of sensor values."""
        self.assertTrue(setup_component(
            self.hass, sensor.DOMAIN, {'sensor': VALID_CONFIG}))

        weather = self.hass.states.get('sensor.bom_fake_weather').state
        self.assertEqual('Fine', weather)

        pressure = self.hass.states.get('sensor.bom_fake_pressure_mb').state
        self.assertEqual('1021.7', pressure)

        feels_like = self.hass.states.get('sensor.bom_fake_feels_like_c').state
        self.assertEqual('25.0', feels_like)