Skip to content

Commit

Permalink
format tests using black
Browse files Browse the repository at this point in the history
  • Loading branch information
@escoand
escoand committed Aug 8, 2019
1 parent 0c7e4dc commit e126b4b
Showing 1 changed file with 89 additions and 93 deletions.
182 changes: 89 additions & 93 deletions tests/components/traccar/test_init.py
View file
Expand Up @@ -5,11 +5,14 @@

from homeassistant import data_entry_flow
from homeassistant.components import traccar, zone
from homeassistant.components.device_tracker import (
DOMAIN as DEVICE_TRACKER_DOMAIN)
from homeassistant.components.device_tracker import DOMAIN as DEVICE_TRACKER_DOMAIN
from homeassistant.components.traccar import DOMAIN, TRACKER_UPDATE
from homeassistant.const import (
HTTP_OK, HTTP_UNPROCESSABLE_ENTITY, STATE_HOME, STATE_NOT_HOME)
HTTP_OK,
HTTP_UNPROCESSABLE_ENTITY,
STATE_HOME,
STATE_NOT_HOME,
)
from homeassistant.helpers.dispatcher import DATA_DISPATCHER
from homeassistant.setup import async_setup_component

Expand All @@ -28,59 +31,54 @@ def mock_dev_track(mock_device_tracker_conf):
@pytest.fixture
async def traccar_client(loop, hass, aiohttp_client):
"""Mock client for Traccar (unauthenticated)."""
assert await async_setup_component(
hass, 'persistent_notification', {})
assert await async_setup_component(hass, "persistent_notification", {})

assert await async_setup_component(
hass, DOMAIN, {
DOMAIN: {}
})
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {}})

await hass.async_block_till_done()

with patch('homeassistant.components.device_tracker.legacy.update_config'):
with patch("homeassistant.components.device_tracker.legacy.update_config"):
return await aiohttp_client(hass.http.app)


@pytest.fixture(autouse=True)
async def setup_zones(loop, hass):
"""Set up Zone config in HA."""
assert await async_setup_component(
hass, zone.DOMAIN, {
'zone': {
'name': 'Home',
'latitude': HOME_LATITUDE,
'longitude': HOME_LONGITUDE,
'radius': 100,
}})
hass,
zone.DOMAIN,
{
"zone": {
"name": "Home",
"latitude": HOME_LATITUDE,
"longitude": HOME_LONGITUDE,
"radius": 100,
}
},
)
await hass.async_block_till_done()


@pytest.fixture
async def webhook_id(hass, traccar_client):
"""Initialize the Traccar component and get the webhook_id."""
hass.config.api = Mock(base_url='http://example.com')
result = await hass.config_entries.flow.async_init(DOMAIN, context={
'source': 'user'
})
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM, result
hass.config.api = Mock(base_url="http://example.com")
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": "user"}
)
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM, result

result = await hass.config_entries.flow.async_configure(
result['flow_id'], {})
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY

await hass.async_block_till_done()
return result['result'].data['webhook_id']
return result["result"].data["webhook_id"]


async def test_missing_data(hass, traccar_client, webhook_id):
"""Test missing data."""
url = '/api/webhook/{}'.format(webhook_id)
data = {
'lat': 1.0,
'lon': 1.1,
'id': '123',
}
url = "/api/webhook/{}".format(webhook_id)
data = {"lat": 1.0, "lon": 1.1, "id": "123"}

# No data
req = await traccar_client.post(url)
Expand All @@ -89,59 +87,62 @@ async def test_missing_data(hass, traccar_client, webhook_id):

# No latitude
copy = data.copy()
del copy['lat']
del copy["lat"]
req = await traccar_client.post(url, params=copy)
await hass.async_block_till_done()
assert req.status == HTTP_UNPROCESSABLE_ENTITY

# No device
copy = data.copy()
del copy['id']
del copy["id"]
req = await traccar_client.post(url, params=copy)
await hass.async_block_till_done()
assert req.status == HTTP_UNPROCESSABLE_ENTITY


async def test_enter_and_exit(hass, traccar_client, webhook_id):
"""Test when there is a known zone."""
url = '/api/webhook/{}'.format(webhook_id)
url = "/api/webhook/{}".format(webhook_id)
data = {
'timestamp': 123456789,
'lat': HOME_LATITUDE,
'lon': HOME_LONGITUDE,
'id': '123',
'accuracy': 10.5,
'batt': 10,
'speed': 100,
'bearing': 105.32,
'altitude': 102
"timestamp": 123456789,
"lat": HOME_LATITUDE,
"lon": HOME_LONGITUDE,
"id": "123",
"accuracy": 10.5,
"batt": 10,
"speed": 100,
"bearing": 105.32,
"altitude": 102,
}

# Enter the Home
req = await traccar_client.post(url, params=data)
await hass.async_block_till_done()
assert req.status == HTTP_OK
state_name = hass.states.get('{}.{}'.format(DEVICE_TRACKER_DOMAIN,
data['id'])).state
state_name = hass.states.get(
"{}.{}".format(DEVICE_TRACKER_DOMAIN, data["id"])
).state
assert STATE_HOME == state_name

# Enter Home again
req = await traccar_client.post(url, params=data)
await hass.async_block_till_done()
assert req.status == HTTP_OK
state_name = hass.states.get('{}.{}'.format(DEVICE_TRACKER_DOMAIN,
data['id'])).state
state_name = hass.states.get(
"{}.{}".format(DEVICE_TRACKER_DOMAIN, data["id"])
).state
assert STATE_HOME == state_name

data['lon'] = 0
data['lat'] = 0
data["lon"] = 0
data["lat"] = 0

# Enter Somewhere else
req = await traccar_client.post(url, params=data)
await hass.async_block_till_done()
assert req.status == HTTP_OK
state_name = hass.states.get('{}.{}'.format(DEVICE_TRACKER_DOMAIN,
data['device'])).state
state_name = hass.states.get(
"{}.{}".format(DEVICE_TRACKER_DOMAIN, data["device"])
).state
assert STATE_NOT_HOME == state_name

dev_reg = await hass.helpers.device_registry.async_get_registry()
Expand All @@ -153,73 +154,68 @@ async def test_enter_and_exit(hass, traccar_client, webhook_id):

async def test_enter_with_attrs(hass, traccar_client, webhook_id):
"""Test when additional attributes are present."""
url = '/api/webhook/{}'.format(webhook_id)
url = "/api/webhook/{}".format(webhook_id)
data = {
'timestamp': 123456789,
'lat': 1.0,
'lon': 1.1,
'id': '123',
'accuracy': 10.5,
'batt': 10,
'speed': 100,
'bearing': 105.32,
'altitude': 102
"timestamp": 123456789,
"lat": 1.0,
"lon": 1.1,
"id": "123",
"accuracy": 10.5,
"batt": 10,
"speed": 100,
"bearing": 105.32,
"altitude": 102,
}

req = await traccar_client.post(url, params=data)
await hass.async_block_till_done()
assert req.status == HTTP_OK
state = hass.states.get('{}.{}'.format(DEVICE_TRACKER_DOMAIN,
data['id']))
state = hass.states.get("{}.{}".format(DEVICE_TRACKER_DOMAIN, data["id"]))
assert state.state == STATE_NOT_HOME
assert state.attributes['gps_accuracy'] == 10.5
assert state.attributes['battery_level'] == 10.0
assert state.attributes['speed'] == 100.0
assert state.attributes['bearing'] == 105.32
assert state.attributes['altitude'] == 102.0
assert state.attributes["gps_accuracy"] == 10.5
assert state.attributes["battery_level"] == 10.0
assert state.attributes["speed"] == 100.0
assert state.attributes["bearing"] == 105.32
assert state.attributes["altitude"] == 102.0

data = {
'lat': HOME_LATITUDE,
'lon': HOME_LONGITUDE,
'id': '123',
'accuracy': 123,
'batt': 23,
'speed': 23,
'bearing': 123,
'altitude': 123
"lat": HOME_LATITUDE,
"lon": HOME_LONGITUDE,
"id": "123",
"accuracy": 123,
"batt": 23,
"speed": 23,
"bearing": 123,
"altitude": 123,
}

req = await traccar_client.post(url, params=data)
await hass.async_block_till_done()
assert req.status == HTTP_OK
state = hass.states.get('{}.{}'.format(DEVICE_TRACKER_DOMAIN,
data['id']))
state = hass.states.get("{}.{}".format(DEVICE_TRACKER_DOMAIN, data["id"]))
assert state.state == STATE_HOME
assert state.attributes['gps_accuracy'] == 123
assert state.attributes['battery_level'] == 23
assert state.attributes['speed'] == 23
assert state.attributes['bearing'] == 123
assert state.attributes['altitude'] == 123
assert state.attributes["gps_accuracy"] == 123
assert state.attributes["battery_level"] == 23
assert state.attributes["speed"] == 23
assert state.attributes["bearing"] == 123
assert state.attributes["altitude"] == 123


@pytest.mark.xfail(
reason='The device_tracker component does not support unloading yet.'
reason="The device_tracker component does not support unloading yet."
)
async def test_load_unload_entry(hass, traccar_client, webhook_id):
"""Test that the appropriate dispatch signals are added and removed."""
url = '/api/webhook/{}'.format(webhook_id)
data = {
'lat': HOME_LATITUDE,
'lon': HOME_LONGITUDE,
'id': '123',
}
url = "/api/webhook/{}".format(webhook_id)
data = {"lat": HOME_LATITUDE, "lon": HOME_LONGITUDE, "id": "123"}

# Enter the Home
req = await traccar_client.post(url, params=data)
await hass.async_block_till_done()
assert req.status == HTTP_OK
state_name = hass.states.get('{}.{}'.format(DEVICE_TRACKER_DOMAIN,
data['id'])).state
state_name = hass.states.get(
"{}.{}".format(DEVICE_TRACKER_DOMAIN, data["id"])
).state
assert STATE_HOME == state_name
assert len(hass.data[DATA_DISPATCHER][TRACKER_UPDATE]) == 1

Expand Down

0 comments on commit e126b4b

Please sign in to comment.