jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rewrite proximity tests to be pytest functions (#41711)

Browse Source
This commit is contained in:
Ariana Hlavaty 2020-10-16 12:13:34 +01:00 committed by GitHub
parent aabf26bbc8
commit c9ba5bfebd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 539 additions and 507 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

352
tests/components/proximity/test_init.py
View File

@ -1,33 +1,10 @@
"""The tests for the Proximity component.""" """The tests for the Proximity component."""
import unittest
from homeassistant.components import proximity
from homeassistant.components.proximity import DOMAIN from homeassistant.components.proximity import DOMAIN
from homeassistant.setup import setup_component from homeassistant.setup import async_setup_component
from tests.common import get_test_home_assistant
class TestProximity(unittest.TestCase): async def test_proximities(hass):
"""Test the Proximity component."""
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
self.hass.config.components.add("zone")
self.hass.states.set(
"zone.home",
"zoning",
{"name": "home", "latitude": 2.1, "longitude": 1.1, "radius": 10},
)
self.hass.states.set(
"zone.work",
"zoning",
{"name": "work", "latitude": 2.3, "longitude": 1.3, "radius": 10},
)
self.addCleanup(self.hass.stop)
def test_proximities(self):
"""Test a list of proximities.""" """Test a list of proximities."""
config = { config = {
"proximity": { "proximity": {
@ -40,22 +17,23 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
proximities = ["home", "work"] proximities = ["home", "work"]
for prox in proximities: for prox in proximities:
state = self.hass.states.get(f"proximity.{prox}") state = hass.states.get(f"proximity.{prox}")
assert state.state == "not set" assert state.state == "not set"
assert state.attributes.get("nearest") == "not set" assert state.attributes.get("nearest") == "not set"
assert state.attributes.get("dir_of_travel") == "not set" assert state.attributes.get("dir_of_travel") == "not set"
self.hass.states.set(f"proximity.{prox}", "0") hass.states.async_set(f"proximity.{prox}", "0")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get(f"proximity.{prox}") state = hass.states.get(f"proximity.{prox}")
assert state.state == "0" assert state.state == "0"
def test_proximities_setup(self):
async def test_proximities_setup(hass):
"""Test a list of proximities with missing devices.""" """Test a list of proximities with missing devices."""
config = { config = {
"proximity": { "proximity": {
@ -68,9 +46,10 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
def test_proximity(self):
async def test_proximity(hass):
"""Test the proximity.""" """Test the proximity."""
config = { config = {
"proximity": { "proximity": {
@ -82,19 +61,20 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.state == "not set" assert state.state == "not set"
assert state.attributes.get("nearest") == "not set" assert state.attributes.get("nearest") == "not set"
assert state.attributes.get("dir_of_travel") == "not set" assert state.attributes.get("dir_of_travel") == "not set"
self.hass.states.set("proximity.home", "0") hass.states.async_set("proximity.home", "0")
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.state == "0" assert state.state == "0"
def test_device_tracker_test1_in_zone(self):
async def test_device_tracker_test1_in_zone(hass):
"""Test for tracker in zone.""" """Test for tracker in zone."""
config = { config = {
"proximity": { "proximity": {
@ -106,20 +86,21 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"home", "home",
{"friendly_name": "test1", "latitude": 2.1, "longitude": 1.1}, {"friendly_name": "test1", "latitude": 2.1, "longitude": 1.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.state == "0" assert state.state == "0"
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "arrived" assert state.attributes.get("dir_of_travel") == "arrived"
def test_device_trackers_in_zone(self):
async def test_device_trackers_in_zone(hass):
"""Test for trackers in zone.""" """Test for trackers in zone."""
config = { config = {
"proximity": { "proximity": {
@ -131,28 +112,29 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"home", "home",
{"friendly_name": "test1", "latitude": 2.1, "longitude": 1.1}, {"friendly_name": "test1", "latitude": 2.1, "longitude": 1.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "device_tracker.test2",
"home", "home",
{"friendly_name": "test2", "latitude": 2.1, "longitude": 1.1}, {"friendly_name": "test2", "latitude": 2.1, "longitude": 1.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.state == "0" assert state.state == "0"
assert (state.attributes.get("nearest") == "test1, test2") or ( assert (state.attributes.get("nearest") == "test1, test2") or (
state.attributes.get("nearest") == "test2, test1" state.attributes.get("nearest") == "test2, test1"
) )
assert state.attributes.get("dir_of_travel") == "arrived" assert state.attributes.get("dir_of_travel") == "arrived"
def test_device_tracker_test1_away(self):
async def test_device_tracker_test1_away(hass):
"""Test for tracker state away.""" """Test for tracker state away."""
config = { config = {
"proximity": { "proximity": {
@ -164,21 +146,26 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
def test_device_tracker_test1_awayfurther(self):
async def test_device_tracker_test1_awayfurther(hass):
"""Test for tracker state away further.""" """Test for tracker state away further."""
config_zones(hass)
await hass.async_block_till_done()
config = { config = {
"proximity": { "proximity": {
"home": { "home": {
@ -189,30 +176,34 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 40.1, "longitude": 20.1}, {"friendly_name": "test1", "latitude": 40.1, "longitude": 20.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "away_from" assert state.attributes.get("dir_of_travel") == "away_from"
def test_device_tracker_test1_awaycloser(self):
async def test_device_tracker_test1_awaycloser(hass):
"""Test for tracker state away closer.""" """Test for tracker state away closer."""
config_zones(hass)
await hass.async_block_till_done()
config = { config = {
"proximity": { "proximity": {
"home": { "home": {
@ -223,29 +214,30 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 40.1, "longitude": 20.1}, {"friendly_name": "test1", "latitude": 40.1, "longitude": 20.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "towards" assert state.attributes.get("dir_of_travel") == "towards"
def test_all_device_trackers_in_ignored_zone(self):
async def test_all_device_trackers_in_ignored_zone(hass):
"""Test for tracker in ignored zone.""" """Test for tracker in ignored zone."""
config = { config = {
"proximity": { "proximity": {
@ -257,16 +249,17 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set("device_tracker.test1", "work", {"friendly_name": "test1"}) hass.states.async_set("device_tracker.test1", "work", {"friendly_name": "test1"})
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.state == "not set" assert state.state == "not set"
assert state.attributes.get("nearest") == "not set" assert state.attributes.get("nearest") == "not set"
assert state.attributes.get("dir_of_travel") == "not set" assert state.attributes.get("dir_of_travel") == "not set"
def test_device_tracker_test1_no_coordinates(self):
async def test_device_tracker_test1_no_coordinates(hass):
"""Test for tracker with no coordinates.""" """Test for tracker with no coordinates."""
config = { config = {
"proximity": { "proximity": {
@ -278,29 +271,34 @@ class TestProximity(unittest.TestCase):
} }
} }
assert setup_component(self.hass, DOMAIN, config) assert await async_setup_component(hass, DOMAIN, config)
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "not_home", {"friendly_name": "test1"} "device_tracker.test1", "not_home", {"friendly_name": "test1"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "not set" assert state.attributes.get("nearest") == "not set"
assert state.attributes.get("dir_of_travel") == "not set" assert state.attributes.get("dir_of_travel") == "not set"
def test_device_tracker_test1_awayfurther_than_test2_first_test1(self):
async def test_device_tracker_test1_awayfurther_than_test2_first_test1(hass):
"""Test for tracker ordering.""" """Test for tracker ordering."""
self.hass.states.set( config_zones(hass)
await hass.async_block_till_done()
hass.states.async_set(
"device_tracker.test1", "not_home", {"friendly_name": "test1"} "device_tracker.test1", "not_home", {"friendly_name": "test1"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "not_home", {"friendly_name": "test2"} "device_tracker.test2", "not_home", {"friendly_name": "test2"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
assert proximity.setup( assert await async_setup_component(
self.hass, hass,
DOMAIN,
{ {
"proximity": { "proximity": {
"home": { "home": {
@ -313,38 +311,43 @@ class TestProximity(unittest.TestCase):
}, },
) )
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "device_tracker.test2",
"not_home", "not_home",
{"friendly_name": "test2", "latitude": 40.1, "longitude": 20.1}, {"friendly_name": "test2", "latitude": 40.1, "longitude": 20.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
def test_device_tracker_test1_awayfurther_than_test2_first_test2(self):
async def test_device_tracker_test1_awayfurther_than_test2_first_test2(hass):
"""Test for tracker ordering.""" """Test for tracker ordering."""
self.hass.states.set( config_zones(hass)
await hass.async_block_till_done()
hass.states.async_set(
"device_tracker.test1", "not_home", {"friendly_name": "test1"} "device_tracker.test1", "not_home", {"friendly_name": "test1"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "not_home", {"friendly_name": "test2"} "device_tracker.test2", "not_home", {"friendly_name": "test2"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
assert proximity.setup( assert await async_setup_component(
self.hass, hass,
DOMAIN,
{ {
"proximity": { "proximity": {
"home": { "home": {
@ -356,36 +359,38 @@ class TestProximity(unittest.TestCase):
}, },
) )
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "device_tracker.test2",
"not_home", "not_home",
{"friendly_name": "test2", "latitude": 40.1, "longitude": 20.1}, {"friendly_name": "test2", "latitude": 40.1, "longitude": 20.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test2" assert state.attributes.get("nearest") == "test2"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
def test_device_tracker_test1_awayfurther_test2_in_ignored_zone(self):
async def test_device_tracker_test1_awayfurther_test2_in_ignored_zone(hass):
"""Test for tracker states.""" """Test for tracker states."""
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "not_home", {"friendly_name": "test1"} "device_tracker.test1", "not_home", {"friendly_name": "test1"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("device_tracker.test2", "work", {"friendly_name": "test2"}) hass.states.async_set("device_tracker.test2", "work", {"friendly_name": "test2"})
self.hass.block_till_done() await hass.async_block_till_done()
assert proximity.setup( assert await async_setup_component(
self.hass, hass,
DOMAIN,
{ {
"proximity": { "proximity": {
"home": { "home": {
@ -397,29 +402,34 @@ class TestProximity(unittest.TestCase):
}, },
) )
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
def test_device_tracker_test1_awayfurther_test2_first(self):
async def test_device_tracker_test1_awayfurther_test2_first(hass):
"""Test for tracker state.""" """Test for tracker state."""
self.hass.states.set( config_zones(hass)
await hass.async_block_till_done()
hass.states.async_set(
"device_tracker.test1", "not_home", {"friendly_name": "test1"} "device_tracker.test1", "not_home", {"friendly_name": "test1"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "not_home", {"friendly_name": "test2"} "device_tracker.test2", "not_home", {"friendly_name": "test2"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
assert proximity.setup( assert await async_setup_component(
self.hass, hass,
DOMAIN,
{ {
"proximity": { "proximity": {
"home": { "home": {
@ -431,45 +441,47 @@ class TestProximity(unittest.TestCase):
}, },
) )
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 10.1, "longitude": 5.1}, {"friendly_name": "test1", "latitude": 10.1, "longitude": 5.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "device_tracker.test2",
"not_home", "not_home",
{"friendly_name": "test2", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test2", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 40.1, "longitude": 20.1}, {"friendly_name": "test1", "latitude": 40.1, "longitude": 20.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 35.1, "longitude": 15.1}, {"friendly_name": "test1", "latitude": 35.1, "longitude": 15.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set("device_tracker.test1", "work", {"friendly_name": "test1"}) hass.states.async_set("device_tracker.test1", "work", {"friendly_name": "test1"})
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test2" assert state.attributes.get("nearest") == "test2"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
def test_device_tracker_test1_awayfurther_a_bit(self):
async def test_device_tracker_test1_awayfurther_a_bit(hass):
"""Test for tracker states.""" """Test for tracker states."""
assert proximity.setup( assert await async_setup_component(
self.hass, hass,
DOMAIN,
{ {
"proximity": { "proximity": {
"home": { "home": {
@ -482,39 +494,44 @@ class TestProximity(unittest.TestCase):
}, },
) )
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1000001, "longitude": 10.1000001}, {"friendly_name": "test1", "latitude": 20.1000001, "longitude": 10.1000001},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1000002, "longitude": 10.1000002}, {"friendly_name": "test1", "latitude": 20.1000002, "longitude": 10.1000002},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "stationary" assert state.attributes.get("dir_of_travel") == "stationary"
def test_device_tracker_test1_nearest_after_test2_in_ignored_zone(self):
async def test_device_tracker_test1_nearest_after_test2_in_ignored_zone(hass):
"""Test for tracker states.""" """Test for tracker states."""
self.hass.states.set( config_zones(hass)
await hass.async_block_till_done()
hass.states.async_set(
"device_tracker.test1", "not_home", {"friendly_name": "test1"} "device_tracker.test1", "not_home", {"friendly_name": "test1"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "not_home", {"friendly_name": "test2"} "device_tracker.test2", "not_home", {"friendly_name": "test2"}
) )
self.hass.block_till_done() await hass.async_block_till_done()
assert proximity.setup( assert await async_setup_component(
self.hass, hass,
DOMAIN,
{ {
"proximity": { "proximity": {
"home": { "home": {
@ -526,32 +543,47 @@ class TestProximity(unittest.TestCase):
}, },
) )
self.hass.states.set( hass.states.async_set(
"device_tracker.test1", "device_tracker.test1",
"not_home", "not_home",
{"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1}, {"friendly_name": "test1", "latitude": 20.1, "longitude": 10.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "device_tracker.test2",
"not_home", "not_home",
{"friendly_name": "test2", "latitude": 10.1, "longitude": 5.1}, {"friendly_name": "test2", "latitude": 10.1, "longitude": 5.1},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test2" assert state.attributes.get("nearest") == "test2"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
self.hass.states.set( hass.states.async_set(
"device_tracker.test2", "device_tracker.test2",
"work", "work",
{"friendly_name": "test2", "latitude": 12.6, "longitude": 7.6}, {"friendly_name": "test2", "latitude": 12.6, "longitude": 7.6},
) )
self.hass.block_till_done() await hass.async_block_till_done()
state = self.hass.states.get("proximity.home") state = hass.states.get("proximity.home")
assert state.attributes.get("nearest") == "test1" assert state.attributes.get("nearest") == "test1"
assert state.attributes.get("dir_of_travel") == "unknown" assert state.attributes.get("dir_of_travel") == "unknown"
def config_zones(hass):
"""Set up zones for test."""
hass.config.components.add("zone")
hass.states.async_set(
"zone.home",
"zoning",
{"name": "home", "latitude": 2.1, "longitude": 1.1, "radius": 10},
)
hass.states.async_set(
"zone.work",
"zoning",
{"name": "work", "latitude": 2.3, "longitude": 1.3, "radius": 10},
)