100 lines
3.1 KiB
Python
100 lines
3.1 KiB
Python
"""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)
|