Change test to parametrized test using pytest (#18047)

This commit is contained in:
Tsvi Mostovicz 2018-11-01 10:40:31 +02:00 committed by Paulus Schoutsen
parent f516550f9f
commit 329d128e03

View file

@ -1,9 +1,10 @@
"""The tests for the Jewish calendar sensor platform."""
import unittest
from datetime import time
from datetime import datetime as dt
from unittest.mock import patch
import pytest
from homeassistant.util.async_ import run_coroutine_threadsafe
from homeassistant.util.dt import get_time_zone, set_default_time_zone
from homeassistant.setup import setup_component
@ -11,17 +12,14 @@ from homeassistant.components.sensor.jewish_calendar import JewishCalSensor
from tests.common import get_test_home_assistant
class TestJewishCalenderSensor(unittest.TestCase):
class TestJewishCalenderSensor():
"""Test the Jewish Calendar sensor."""
TEST_LATITUDE = 31.778
TEST_LONGITUDE = 35.235
def setUp(self):
def setup_method(self, method):
"""Set up things to run when tests begin."""
self.hass = get_test_home_assistant()
def tearDown(self):
def teardown_method(self, method):
"""Stop everything that was started."""
self.hass.stop()
# Reset the default timezone, so we don't affect other tests
@ -62,182 +60,63 @@ class TestJewishCalenderSensor(unittest.TestCase):
assert setup_component(self.hass, 'sensor', config)
def test_jewish_calendar_sensor_date_output(self):
"""Test Jewish calendar sensor date output."""
test_time = dt(2018, 9, 3)
set_default_time_zone(get_time_zone('UTC'))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
test_params = [
(dt(2018, 9, 3), 'UTC', 31.778, 35.235, "english", "date",
False, "23 Elul 5778"),
(dt(2018, 9, 3), 'UTC', 31.778, 35.235, "hebrew", "date",
False, "כ\"ג באלול ה\' תשע\"ח"),
(dt(2018, 9, 10), 'UTC', 31.778, 35.235, "hebrew", "holiday_name",
False, "א\' ראש השנה"),
(dt(2018, 9, 10), 'UTC', 31.778, 35.235, "english", "holiday_name",
False, "Rosh Hashana I"),
(dt(2018, 9, 10), 'UTC', 31.778, 35.235, "english", "holyness",
False, 1),
(dt(2018, 9, 8), 'UTC', 31.778, 35.235, "hebrew", "weekly_portion",
False, "פרשת נצבים"),
(dt(2018, 9, 8), 'America/New_York', 40.7128, -74.0060, "hebrew",
"first_stars", True, time(19, 48)),
(dt(2018, 9, 8), "Asia/Jerusalem", 31.778, 35.235, "hebrew",
"first_stars", False, time(19, 21)),
(dt(2018, 10, 14), "Asia/Jerusalem", 31.778, 35.235, "hebrew",
"weekly_portion", False, "פרשת לך לך"),
(dt(2018, 10, 14, 17, 0, 0), "Asia/Jerusalem", 31.778, 35.235,
"hebrew", "date", False, "ה\' בחשון ה\' תשע\"ט"),
(dt(2018, 10, 14, 19, 0, 0), "Asia/Jerusalem", 31.778, 35.235,
"hebrew", "date", False, "ו\' בחשון ה\' תשע\"ט")
]
test_ids = [
"date_output",
"date_output_hebrew",
"holiday_name",
"holiday_name_english",
"holyness",
"torah_reading",
"first_stars_ny",
"first_stars_jerusalem",
"torah_reading_weekday",
"date_before_sunset",
"date_after_sunset"
]
@pytest.mark.parametrize(["time", "tzname", "latitude", "longitude",
"language", "sensor", "diaspora", "result"],
test_params, ids=test_ids)
def test_jewish_calendar_sensor(self, time, tzname, latitude, longitude,
language, sensor, diaspora, result):
"""Test Jewish calendar sensor output."""
tz = get_time_zone(tzname)
set_default_time_zone(tz)
test_time = tz.localize(time)
self.hass.config.latitude = latitude
self.hass.config.longitude = longitude
sensor = JewishCalSensor(
name='test', language='english', sensor_type='date',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("UTC"), diaspora=False)
name='test', language=language, sensor_type=sensor,
latitude=latitude, longitude=longitude,
timezone=tz, diaspora=diaspora)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(),
self.hass.loop).result()
assert sensor.state == '23 Elul 5778'
def test_jewish_calendar_sensor_date_output_hebrew(self):
"""Test Jewish calendar sensor date output in hebrew."""
test_time = dt(2018, 9, 3)
set_default_time_zone(get_time_zone('UTC'))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='date',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("UTC"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "כ\"ג באלול ה\' תשע\"ח"
def test_jewish_calendar_sensor_holiday_name(self):
"""Test Jewish calendar sensor holiday name output in hebrew."""
test_time = dt(2018, 9, 10)
set_default_time_zone(get_time_zone('UTC'))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='holiday_name',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("UTC"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "א\' ראש השנה"
def test_jewish_calendar_sensor_holiday_name_english(self):
"""Test Jewish calendar sensor holiday name output in english."""
test_time = dt(2018, 9, 10)
set_default_time_zone(get_time_zone('UTC'))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='english', sensor_type='holiday_name',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("UTC"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "Rosh Hashana I"
def test_jewish_calendar_sensor_holyness(self):
"""Test Jewish calendar sensor holyness value."""
test_time = dt(2018, 9, 10)
set_default_time_zone(get_time_zone('UTC'))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='holyness',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("UTC"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == 1
def test_jewish_calendar_sensor_torah_reading(self):
"""Test Jewish calendar sensor torah reading in hebrew."""
test_time = dt(2018, 9, 8)
set_default_time_zone(get_time_zone('UTC'))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='weekly_portion',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("UTC"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "פרשת נצבים"
def test_jewish_calendar_sensor_first_stars_ny(self):
"""Test Jewish calendar sensor first stars time in NY, US."""
test_time = dt(2018, 9, 8)
set_default_time_zone(get_time_zone('America/New_York'))
self.hass.config.latitude = 40.7128
self.hass.config.longitude = -74.0060
# self.hass.config.time_zone = get_time_zone("America/New_York")
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='first_stars',
latitude=40.7128, longitude=-74.0060,
timezone=get_time_zone("America/New_York"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == time(19, 48)
def test_jewish_calendar_sensor_first_stars_jerusalem(self):
"""Test Jewish calendar sensor first stars time in Jerusalem, IL."""
set_default_time_zone(get_time_zone('Asia/Jerusalem'))
test_time = dt(2018, 9, 8)
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='first_stars',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("Asia/Jerusalem"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == time(19, 21)
def test_jewish_calendar_sensor_torah_reading_weekday(self):
"""Test the sensor showing torah reading also on weekdays."""
set_default_time_zone(get_time_zone('Asia/Jerusalem'))
test_time = dt(2018, 10, 14)
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='weekly_portion',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("Asia/Jerusalem"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "פרשת לך לך"
def test_jewish_calendar_sensor_date_before_sunset(self):
"""Test the sensor showing the correct date before sunset."""
tz = get_time_zone('Asia/Jerusalem')
set_default_time_zone(tz)
test_time = tz.localize(dt(2018, 10, 14, 17, 0, 0))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='date',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("Asia/Jerusalem"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "ה\' בחשון ה\' תשע\"ט"
def test_jewish_calendar_sensor_date_after_sunset(self):
"""Test the sensor showing the correct date after sunset."""
tz = get_time_zone('Asia/Jerusalem')
set_default_time_zone(tz)
test_time = tz.localize(dt(2018, 10, 14, 19, 0, 0))
self.hass.config.latitude = self.TEST_LATITUDE
self.hass.config.longitude = self.TEST_LONGITUDE
sensor = JewishCalSensor(
name='test', language='hebrew', sensor_type='date',
latitude=self.TEST_LATITUDE, longitude=self.TEST_LONGITUDE,
timezone=get_time_zone("Asia/Jerusalem"), diaspora=False)
sensor.hass = self.hass
with patch('homeassistant.util.dt.now', return_value=test_time):
run_coroutine_threadsafe(
sensor.async_update(), self.hass.loop).result()
assert sensor.state == "ו\' בחשון ה\' תשע\"ט"
assert sensor.state == result