Add tests for Locative
Philip Lundrigan
parent
7d41ce4e46
commit
55d1ad94ef
|
|
@ -41,7 +41,6 @@ omit =
|
|||
homeassistant/components/device_tracker/asuswrt.py
|
||||
homeassistant/components/device_tracker/ddwrt.py
|
||||
homeassistant/components/device_tracker/fritz.py
|
||||
homeassistant/components/device_tracker/locative.py
|
||||
homeassistant/components/device_tracker/icloud.py
|
||||
homeassistant/components/device_tracker/luci.py
|
||||
homeassistant/components/device_tracker/netgear.py
|
||||
|
|
|
|||
|
|
@ -0,0 +1,246 @@
|
|||
"""
|
||||
tests.components.device_tracker.locative
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
Tests the locative device tracker component.
|
||||
"""
|
||||
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
|
||||
import requests
|
||||
|
||||
from homeassistant import bootstrap, const
|
||||
import homeassistant.core as ha
|
||||
import homeassistant.components.device_tracker as device_tracker
|
||||
import homeassistant.components.http as http
|
||||
import homeassistant.components.zone as zone
|
||||
|
||||
SERVER_PORT = 8126
|
||||
HTTP_BASE_URL = "http://127.0.0.1:{}".format(SERVER_PORT)
|
||||
|
||||
hass = None
|
||||
|
||||
|
||||
def _url(data={}):
|
||||
""" Helper method to generate urls. """
|
||||
data = "&".join(["{}={}".format(name, value) for name, value in data.items()])
|
||||
return "{}{}locative?{}".format(HTTP_BASE_URL, const.URL_API, data)
|
||||
|
||||
|
||||
@patch('homeassistant.components.http.util.get_local_ip',
|
||||
return_value='127.0.0.1')
|
||||
def setUpModule(mock_get_local_ip): # pylint: disable=invalid-name
|
||||
""" Initalizes a Home Assistant server. """
|
||||
global hass
|
||||
|
||||
hass = ha.HomeAssistant()
|
||||
|
||||
# Set up a couple of zones
|
||||
bootstrap.setup_component(hass, zone.DOMAIN, {
|
||||
zone.DOMAIN: [
|
||||
{
|
||||
'name': 'Home',
|
||||
'latitude': 41.7855,
|
||||
'longitude': -110.7367,
|
||||
'radius': 200
|
||||
},
|
||||
{
|
||||
'name': 'Work',
|
||||
'latitude': 41.5855,
|
||||
'longitude': -110.9367,
|
||||
'radius': 100
|
||||
}
|
||||
]
|
||||
})
|
||||
|
||||
# Set up server
|
||||
bootstrap.setup_component(hass, http.DOMAIN, {
|
||||
http.DOMAIN: {
|
||||
http.CONF_SERVER_PORT: SERVER_PORT
|
||||
}
|
||||
})
|
||||
|
||||
# Set up API
|
||||
bootstrap.setup_component(hass, 'api')
|
||||
|
||||
# Set up device tracker
|
||||
bootstrap.setup_component(hass, device_tracker.DOMAIN, {
|
||||
device_tracker.DOMAIN: {
|
||||
'platform': 'locative'
|
||||
}
|
||||
})
|
||||
|
||||
hass.start()
|
||||
|
||||
|
||||
def tearDownModule(): # pylint: disable=invalid-name
|
||||
""" Stops the Home Assistant server. """
|
||||
hass.stop()
|
||||
|
||||
|
||||
class TestLocative(unittest.TestCase):
|
||||
""" Test Locative """
|
||||
|
||||
def test_missing_data(self):
|
||||
data = {
|
||||
'latitude': 1.0,
|
||||
'longitude': 1.1,
|
||||
'device': '123',
|
||||
'id': 'Home',
|
||||
'trigger': 'enter'
|
||||
}
|
||||
|
||||
# No data
|
||||
req = requests.get(_url({}))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
# No latitude
|
||||
copy = data.copy()
|
||||
del copy['latitude']
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
# No device
|
||||
copy = data.copy()
|
||||
del copy['device']
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
# No location
|
||||
copy = data.copy()
|
||||
del copy['id']
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
# No trigger
|
||||
copy = data.copy()
|
||||
del copy['trigger']
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
# Bad longitude
|
||||
copy = data.copy()
|
||||
copy['longitude'] = 'hello world'
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
# Test message
|
||||
copy = data.copy()
|
||||
copy['trigger'] = 'test'
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(200, req.status_code)
|
||||
|
||||
# Unknown trigger
|
||||
copy = data.copy()
|
||||
copy['trigger'] = 'foobar'
|
||||
req = requests.get(_url(copy))
|
||||
self.assertEqual(422, req.status_code)
|
||||
|
||||
|
||||
def test_known_zone(self):
|
||||
""" Test when there is a known zone """
|
||||
data = {
|
||||
'latitude': 40.7855,
|
||||
'longitude': -111.7367,
|
||||
'device': '123',
|
||||
'id': 'Home',
|
||||
'trigger': 'enter'
|
||||
}
|
||||
|
||||
# Enter the Home
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
state_name = hass.states.get('{}.{}'.format('device_tracker', data['device'])).state
|
||||
self.assertEqual(state_name, 'home')
|
||||
|
||||
data['id'] = 'HOME'
|
||||
data['trigger'] = 'exit'
|
||||
|
||||
# Exit Home
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
state_name = hass.states.get('{}.{}'.format('device_tracker', data['device'])).state
|
||||
self.assertEqual(state_name, 'not_home')
|
||||
|
||||
data['id'] = 'hOmE'
|
||||
data['trigger'] = 'enter'
|
||||
|
||||
# Enter Home again
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
state_name = hass.states.get('{}.{}'.format('device_tracker', data['device'])).state
|
||||
self.assertEqual(state_name, 'home')
|
||||
|
||||
|
||||
def test_unknown_zone(self):
|
||||
""" Test when there is no known zone """
|
||||
data = {
|
||||
'latitude': 40.7855,
|
||||
'longitude': -111.7367,
|
||||
'device': '123',
|
||||
'id': 'Foobar',
|
||||
'trigger': 'enter'
|
||||
}
|
||||
|
||||
# Enter Foobar
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
|
||||
state = hass.states.get('{}.{}'.format('device_tracker', data['device']))
|
||||
self.assertEqual(state.state, 'not_home')
|
||||
self.assertEqual(state.attributes['latitude'], data['latitude'])
|
||||
self.assertEqual(state.attributes['longitude'], data['longitude'])
|
||||
|
||||
data['trigger'] = 'exit'
|
||||
|
||||
# Exit Foobar
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
|
||||
state = hass.states.get('{}.{}'.format('device_tracker', data['device']))
|
||||
self.assertEqual(state.state, 'not_home')
|
||||
self.assertEqual(state.attributes['latitude'], data['latitude'])
|
||||
self.assertEqual(state.attributes['longitude'], data['longitude'])
|
||||
|
||||
|
||||
def test_exit_after_enter(self):
|
||||
""" Test when an exit message comes after an enter message """
|
||||
|
||||
data = {
|
||||
'latitude': 40.7855,
|
||||
'longitude': -111.7367,
|
||||
'device': '123',
|
||||
'id': 'Home',
|
||||
'trigger': 'enter'
|
||||
}
|
||||
|
||||
# Enter Home
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
|
||||
state = hass.states.get('{}.{}'.format('device_tracker', data['device']))
|
||||
self.assertEqual(state.state, 'home')
|
||||
|
||||
data['id'] = 'Work'
|
||||
|
||||
# Enter Work
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
|
||||
state = hass.states.get('{}.{}'.format('device_tracker', data['device']))
|
||||
self.assertEqual(state.state, 'work')
|
||||
|
||||
data['id'] = 'Home'
|
||||
data['trigger'] = 'exit'
|
||||
|
||||
# Exit Home
|
||||
req = requests.get(_url(data))
|
||||
self.assertEqual(200, req.status_code)
|
||||
|
||||
state = hass.states.get('{}.{}'.format('device_tracker', data['device']))
|
||||
self.assertEqual(state.state, 'work')
|
||||
|
||||
print(req.text)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue