Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Black

Browse Source
This commit is contained in:
Paulus Schoutsen
2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download Patch File Download Diff File Expand all files Collapse all files

181
tests/components/smartthings/conftest.py
View File

@@ -3,21 +3,41 @@ from uuid import uuid4
from asynctest import Mock, patch
from pysmartthings import (
CLASSIFICATION_AUTOMATION, AppEntity, AppOAuthClient, AppSettings,
DeviceEntity, DeviceStatus, InstalledApp, InstalledAppStatus,
InstalledAppType, Location, SceneEntity, SmartThings, Subscription)
CLASSIFICATION_AUTOMATION,
AppEntity,
AppOAuthClient,
AppSettings,
DeviceEntity,
DeviceStatus,
InstalledApp,
InstalledAppStatus,
InstalledAppType,
Location,
SceneEntity,
SmartThings,
Subscription,
)
from pysmartthings.api import Api
import pytest
from homeassistant.components import webhook
from homeassistant.components.smartthings import DeviceBroker
from homeassistant.components.smartthings.const import (
APP_NAME_PREFIX, CONF_APP_ID, CONF_INSTALLED_APP_ID, CONF_INSTANCE_ID,
CONF_LOCATION_ID, CONF_OAUTH_CLIENT_ID, CONF_OAUTH_CLIENT_SECRET,
CONF_REFRESH_TOKEN, DATA_BROKERS, DOMAIN, SETTINGS_INSTANCE_ID,
STORAGE_KEY, STORAGE_VERSION)
from homeassistant.config_entries import (
CONN_CLASS_CLOUD_PUSH, SOURCE_USER, ConfigEntry)
APP_NAME_PREFIX,
CONF_APP_ID,
CONF_INSTALLED_APP_ID,
CONF_INSTANCE_ID,
CONF_LOCATION_ID,
CONF_OAUTH_CLIENT_ID,
CONF_OAUTH_CLIENT_SECRET,
CONF_REFRESH_TOKEN,
DATA_BROKERS,
DOMAIN,
SETTINGS_INSTANCE_ID,
STORAGE_KEY,
STORAGE_VERSION,
)
from homeassistant.config_entries import CONN_CLASS_CLOUD_PUSH, SOURCE_USER, ConfigEntry
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_WEBHOOK_ID
from homeassistant.setup import async_setup_component
@@ -26,23 +46,23 @@ from tests.common import MockConfigEntry
COMPONENT_PREFIX = "homeassistant.components.smartthings."
async def setup_platform(hass, platform: str, *,
devices=None, scenes=None):
async def setup_platform(hass, platform: str, *, devices=None, scenes=None):
"""Set up the SmartThings platform and prerequisites."""
hass.config.components.add(DOMAIN)
config_entry = ConfigEntry(2, DOMAIN, "Test",
{CONF_INSTALLED_APP_ID: str(uuid4())},
SOURCE_USER, CONN_CLASS_CLOUD_PUSH)
broker = DeviceBroker(hass, config_entry, Mock(), Mock(),
devices or [], scenes or [])
config_entry = ConfigEntry(
2,
DOMAIN,
"Test",
{CONF_INSTALLED_APP_ID: str(uuid4())},
SOURCE_USER,
CONN_CLASS_CLOUD_PUSH,
)
broker = DeviceBroker(
hass, config_entry, Mock(), Mock(), devices or [], scenes or []
)
hass.data[DOMAIN] = {
DATA_BROKERS: {
config_entry.entry_id: broker
}
}
await hass.config_entries.async_forward_entry_setup(
config_entry, platform)
hass.data[DOMAIN] = {DATA_BROKERS: {config_entry.entry_id: broker}}
await hass.config_entries.async_forward_entry_setup(config_entry, platform)
await hass.async_block_till_done()
return config_entry
@@ -50,26 +70,25 @@ async def setup_platform(hass, platform: str, *,
@pytest.fixture(autouse=True)
async def setup_component(hass, config_file, hass_storage):
"""Load the SmartThing component."""
hass_storage[STORAGE_KEY] = {'data': config_file,
"version": STORAGE_VERSION}
await async_setup_component(hass, 'smartthings', {})
hass.config.api.base_url = 'https://test.local'
hass_storage[STORAGE_KEY] = {"data": config_file, "version": STORAGE_VERSION}
await async_setup_component(hass, "smartthings", {})
hass.config.api.base_url = "https://test.local"
def _create_location():
loc = Mock(Location)
loc.name = 'Test Location'
loc.name = "Test Location"
loc.location_id = str(uuid4())
return loc
@pytest.fixture(name='location')
@pytest.fixture(name="location")
def location_fixture():
"""Fixture for a single location."""
return _create_location()
@pytest.fixture(name='locations')
@pytest.fixture(name="locations")
def locations_fixture(location):
"""Fixture for 2 locations."""
return [location, _create_location()]
@@ -81,14 +100,14 @@ def app_fixture(hass, config_file):
app = Mock(AppEntity)
app.app_name = APP_NAME_PREFIX + str(uuid4())
app.app_id = str(uuid4())
app.app_type = 'WEBHOOK_SMART_APP'
app.app_type = "WEBHOOK_SMART_APP"
app.classifications = [CLASSIFICATION_AUTOMATION]
app.display_name = 'Home Assistant'
app.description = hass.config.location_name + " at " + \
hass.config.api.base_url
app.display_name = "Home Assistant"
app.description = hass.config.location_name + " at " + hass.config.api.base_url
app.single_instance = True
app.webhook_target_url = webhook.async_generate_url(
hass, hass.data[DOMAIN][CONF_WEBHOOK_ID])
hass, hass.data[DOMAIN][CONF_WEBHOOK_ID]
)
settings = Mock(AppSettings)
settings.app_id = app.app_id
@@ -106,7 +125,7 @@ def app_oauth_client_fixture():
return client
@pytest.fixture(name='app_settings')
@pytest.fixture(name="app_settings")
def app_settings_fixture(app, config_file):
"""Fixture for an app settings."""
settings = Mock(AppSettings)
@@ -125,45 +144,43 @@ def _create_installed_app(location_id, app_id):
return item
@pytest.fixture(name='installed_app')
@pytest.fixture(name="installed_app")
def installed_app_fixture(location, app):
"""Fixture for a single installed app."""
return _create_installed_app(location.location_id, app.app_id)
@pytest.fixture(name='installed_apps')
@pytest.fixture(name="installed_apps")
def installed_apps_fixture(installed_app, locations, app):
"""Fixture for 2 installed apps."""
return [installed_app,
_create_installed_app(locations[1].location_id, app.app_id)]
return [installed_app, _create_installed_app(locations[1].location_id, app.app_id)]
@pytest.fixture(name='config_file')
@pytest.fixture(name="config_file")
def config_file_fixture():
"""Fixture representing the local config file contents."""
return {
CONF_INSTANCE_ID: str(uuid4()),
CONF_WEBHOOK_ID: webhook.generate_secret()
}
return {CONF_INSTANCE_ID: str(uuid4()), CONF_WEBHOOK_ID: webhook.generate_secret()}
@pytest.fixture(name='smartthings_mock')
@pytest.fixture(name="smartthings_mock")
def smartthings_mock_fixture(locations):
"""Fixture to mock smartthings API calls."""
async def _location(location_id):
return next(location for location in locations
if location.location_id == location_id)
return next(
location for location in locations if location.location_id == location_id
)
smartthings_mock = Mock(SmartThings)
smartthings_mock.location.side_effect = _location
mock = Mock(return_value=smartthings_mock)
with patch(COMPONENT_PREFIX + "SmartThings", new=mock), \
patch(COMPONENT_PREFIX + "config_flow.SmartThings", new=mock), \
patch(COMPONENT_PREFIX + "smartapp.SmartThings", new=mock):
with patch(COMPONENT_PREFIX + "SmartThings", new=mock), patch(
COMPONENT_PREFIX + "config_flow.SmartThings", new=mock
), patch(COMPONENT_PREFIX + "smartapp.SmartThings", new=mock):
yield smartthings_mock
@pytest.fixture(name='device')
@pytest.fixture(name="device")
def device_fixture(location):
"""Fixture representing devices loaded."""
item = Mock(DeviceEntity)
@@ -172,15 +189,21 @@ def device_fixture(location):
item.label = "Front Porch Lights"
item.location_id = location.location_id
item.capabilities = [
"switch", "switchLevel", "refresh", "indicator", "sensor", "actuator",
"healthCheck", "light"
"switch",
"switchLevel",
"refresh",
"indicator",
"sensor",
"actuator",
"healthCheck",
"light",
]
item.components = {"main": item.capabilities}
item.status = Mock(DeviceStatus)
return item
@pytest.fixture(name='config_entry')
@pytest.fixture(name="config_entry")
def config_entry_fixture(hass, installed_app, location):
"""Fixture representing a config entry."""
data = {
@@ -190,20 +213,27 @@ def config_entry_fixture(hass, installed_app, location):
CONF_LOCATION_ID: location.location_id,
CONF_REFRESH_TOKEN: str(uuid4()),
CONF_OAUTH_CLIENT_ID: str(uuid4()),
CONF_OAUTH_CLIENT_SECRET: str(uuid4())
CONF_OAUTH_CLIENT_SECRET: str(uuid4()),
}
return MockConfigEntry(domain=DOMAIN, data=data, title=location.name,
version=2, source=SOURCE_USER,
connection_class=CONN_CLASS_CLOUD_PUSH)
return MockConfigEntry(
domain=DOMAIN,
data=data,
title=location.name,
version=2,
source=SOURCE_USER,
connection_class=CONN_CLASS_CLOUD_PUSH,
)
@pytest.fixture(name="subscription_factory")
def subscription_factory_fixture():
"""Fixture for creating mock subscriptions."""
def _factory(capability):
sub = Subscription()
sub.capability = capability
return sub
return _factory
@@ -224,66 +254,76 @@ def device_factory_fixture():
{
"id": "main",
"capabilities": [
{"id": capability, "version": 1}
for capability in capabilities
]
{"id": capability, "version": 1} for capability in capabilities
],
}
],
"dth": {
"deviceTypeId": "b678b29d-2726-4e4f-9c3f-7aa05bd08964",
"deviceTypeName": "Switch",
"deviceNetworkType": "ZWAVE"
"deviceNetworkType": "ZWAVE",
},
"type": "DTH"
"type": "DTH",
}
device = DeviceEntity(api, data=device_data)
if status:
for attribute, value in status.items():
device.status.apply_attribute_update(
'main', '', attribute, value)
device.status.apply_attribute_update("main", "", attribute, value)
return device
return _factory
@pytest.fixture(name="scene_factory")
def scene_factory_fixture(location):
"""Fixture for creating mock devices."""
def _factory(name):
scene = Mock(SceneEntity)
scene.scene_id = str(uuid4())
scene.name = name
scene.location_id = location.location_id
return scene
return _factory
@pytest.fixture(name="scene")
def scene_fixture(scene_factory):
"""Fixture for an individual scene."""
return scene_factory('Test Scene')
return scene_factory("Test Scene")
@pytest.fixture(name="event_factory")
def event_factory_fixture():
"""Fixture for creating mock devices."""
def _factory(device_id, event_type="DEVICE_EVENT", capability='',
attribute='Updated', value='Value', data=None):
def _factory(
device_id,
event_type="DEVICE_EVENT",
capability="",
attribute="Updated",
value="Value",
data=None,
):
event = Mock()
event.event_type = event_type
event.device_id = device_id
event.component_id = 'main'
event.component_id = "main"
event.capability = capability
event.attribute = attribute
event.value = value
event.data = data
event.location_id = str(uuid4())
return event
return _factory
@pytest.fixture(name="event_request_factory")
def event_request_factory_fixture(event_factory):
"""Fixture for creating mock smartapp event requests."""
def _factory(device_ids=None, events=None):
request = Mock()
request.installed_app_id = uuid4()
@@ -295,4 +335,5 @@ def event_request_factory_fixture(event_factory):
events.append(event_factory(device_ids[0], event_type="OTHER"))
request.events = events
return request
return _factory