68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
"""Tests for init module."""
|
|
from homeassistant.components import nws
|
|
from homeassistant.components.nws.const import CONF_STATION, DOMAIN
|
|
from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import assert_setup_component
|
|
from tests.components.nws.const import MINIMAL_CONFIG
|
|
|
|
LATLON_CONFIG = {
|
|
DOMAIN: [{CONF_API_KEY: "test", CONF_LATITUDE: 45.0, CONF_LONGITUDE: -75.0}]
|
|
}
|
|
FULL_CONFIG = {
|
|
DOMAIN: [
|
|
{
|
|
CONF_API_KEY: "test",
|
|
CONF_LATITUDE: 45.0,
|
|
CONF_LONGITUDE: -75.0,
|
|
CONF_STATION: "XYZ",
|
|
}
|
|
]
|
|
}
|
|
DUPLICATE_CONFIG = {
|
|
DOMAIN: [
|
|
{CONF_API_KEY: "test", CONF_LATITUDE: 45.0, CONF_LONGITUDE: -75.0},
|
|
{CONF_API_KEY: "test", CONF_LATITUDE: 45.0, CONF_LONGITUDE: -75.0},
|
|
]
|
|
}
|
|
|
|
|
|
async def test_no_config(hass, mock_simple_nws):
|
|
"""Test that nws does not setup with no config."""
|
|
with assert_setup_component(0):
|
|
assert await async_setup_component(hass, DOMAIN, {}) is True
|
|
assert DOMAIN not in hass.data
|
|
|
|
|
|
async def test_successful_minimal_config(hass, mock_simple_nws):
|
|
"""Test that nws setup with minimal config."""
|
|
hass.config.latitude = 40.0
|
|
hass.config.longitude = -75.0
|
|
with assert_setup_component(1):
|
|
assert await async_setup_component(hass, DOMAIN, MINIMAL_CONFIG) is True
|
|
assert DOMAIN in hass.data
|
|
assert nws.base_unique_id(40.0, -75.0) in hass.data[DOMAIN]
|
|
|
|
|
|
async def test_successful_latlon_config(hass, mock_simple_nws):
|
|
"""Test that nws setup with latlon config."""
|
|
with assert_setup_component(1):
|
|
assert await async_setup_component(hass, DOMAIN, LATLON_CONFIG) is True
|
|
assert DOMAIN in hass.data
|
|
assert nws.base_unique_id(45.0, -75.0) in hass.data[DOMAIN]
|
|
|
|
|
|
async def test_successful_full_config(hass, mock_simple_nws):
|
|
"""Test that nws setup with full config."""
|
|
with assert_setup_component(1):
|
|
assert await async_setup_component(hass, DOMAIN, FULL_CONFIG) is True
|
|
assert DOMAIN in hass.data
|
|
assert nws.base_unique_id(45.0, -75.0) in hass.data[DOMAIN]
|
|
|
|
|
|
async def test_unsuccessful_duplicate_config(hass, mock_simple_nws):
|
|
"""Test that nws setup with duplicate config."""
|
|
assert await async_setup_component(hass, DOMAIN, DUPLICATE_CONFIG) is True
|
|
assert len(hass.data[DOMAIN]) == 1
|