core/tests/components/weather/test_darksky.py

52 lines
1.7 KiB
Python

"""The tests for the Dark Sky weather component."""
import re
import unittest
from unittest.mock import patch
import forecastio
import requests_mock
from homeassistant.components import weather
from homeassistant.util.unit_system import METRIC_SYSTEM
from homeassistant.setup import setup_component
from tests.common import load_fixture, get_test_home_assistant
class TestDarkSky(unittest.TestCase):
"""Test the Dark Sky weather component."""
def setUp(self):
"""Setup things to be run when tests are started."""
self.hass = get_test_home_assistant()
self.hass.config.units = METRIC_SYSTEM
self.lat = self.hass.config.latitude = 37.8267
self.lon = self.hass.config.longitude = -122.423
def tearDown(self):
"""Stop down everything that was started."""
self.hass.stop()
@requests_mock.Mocker()
@patch('forecastio.api.get_forecast', wraps=forecastio.api.get_forecast)
def test_setup(self, mock_req, mock_get_forecast):
"""Test for successfully setting up the forecast.io platform."""
uri = (r'https://api.(darksky.net|forecast.io)\/forecast\/(\w+)\/'
r'(-?\d+\.?\d*),(-?\d+\.?\d*)')
mock_req.get(re.compile(uri),
text=load_fixture('darksky.json'))
self.assertTrue(setup_component(self.hass, weather.DOMAIN, {
'weather': {
'name': 'test',
'platform': 'darksky',
'api_key': 'foo',
}
}))
self.assertTrue(mock_get_forecast.called)
self.assertEqual(mock_get_forecast.call_count, 1)
state = self.hass.states.get('weather.test')
self.assertEqual(state.state, 'Clear')