jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 05:07:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rewrite aurora tests to pytest style (#41144)

Browse Source
This commit is contained in:
Claudio Catterina 2020-10-04 13:58:16 +02:00 committed by GitHub
parent 9ff8f03322
commit 6765d395f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
tests/components/aurora/test_binary_sensor.py
View File

@ -1,38 +1,15 @@
"""The tests for the Aurora sensor platform."""
import re
import unittest
import requests_mock
from homeassistant.components.aurora import binary_sensor as aurora
from tests.common import get_test_home_assistant, load_fixture
from tests.common import load_fixture
class TestAuroraSensorSetUp(unittest.TestCase):
"""Test the aurora platform."""
def setUp(self):
"""Initialize values for this testcase class."""
self.hass = get_test_home_assistant()
self.lat = 37.8267
self.lon = -122.423
self.hass.config.latitude = self.lat
self.hass.config.longitude = self.lon
self.entities = []
self.addCleanup(self.tear_down_cleanup)
def tear_down_cleanup(self):
"""Stop everything that was started."""
self.hass.stop()
@requests_mock.Mocker()
def test_setup_and_initial_state(self, mock_req):
def test_setup_and_initial_state(hass, requests_mock):
"""Test that the component is created and initialized as expected."""
uri = re.compile(
r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt"
)
mock_req.get(uri, text=load_fixture("aurora.txt"))
uri = re.compile(r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt")
requests_mock.get(uri, text=load_fixture("aurora.txt"))
entities = []
@ -46,7 +23,7 @@ class TestAuroraSensorSetUp(unittest.TestCase):
entities.append(entity)
config = {"name": "Test", "forecast_threshold": 75}
aurora.setup_platform(self.hass, config, mock_add_entities)
aurora.setup_platform(hass, config, mock_add_entities)
aurora_component = entities[0]
assert len(entities) == 1
@ -55,13 +32,11 @@ class TestAuroraSensorSetUp(unittest.TestCase):
assert aurora_component.device_state_attributes["message"] == "nothing's out"
assert not aurora_component.is_on
@requests_mock.Mocker()
def test_custom_threshold_works(self, mock_req):
def test_custom_threshold_works(hass, requests_mock):
"""Test that the config can take a custom forecast threshold."""
uri = re.compile(
r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt"
)
mock_req.get(uri, text=load_fixture("aurora.txt"))
uri = re.compile(r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt")
requests_mock.get(uri, text=load_fixture("aurora.txt"))
entities = []
@ -75,10 +50,10 @@ class TestAuroraSensorSetUp(unittest.TestCase):
entities.append(entity)
config = {"name": "Test", "forecast_threshold": 1}
self.hass.config.longitude = 18.987
self.hass.config.latitude = 69.648
hass.config.longitude = 18.987
hass.config.latitude = 69.648
aurora.setup_platform(self.hass, config, mock_add_entities)
aurora.setup_platform(hass, config, mock_add_entities)
aurora_component = entities[0]
assert aurora_component.aurora_data.visibility_level == "16"