jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 13:47:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests closing properly (#5203)

Browse Source
This commit is contained in:
Paulus Schoutsen 2017-01-07 01:47:25 +01:00 committed by GitHub
parent 2b991e2f32
commit aa1e4c564c
14 changed files with 108 additions and 83 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/components/binary_sensor/test_sleepiq.py
View File

@ -31,6 +31,10 @@ class TestSleepIQBinarySensorSetup(unittest.TestCase):
'password': self.password, 'password': self.password,
} }
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
@requests_mock.Mocker() @requests_mock.Mocker()
def test_setup(self, mock): def test_setup(self, mock):
"""Test for successfully setting up the SleepIQ platform.""" """Test for successfully setting up the SleepIQ platform."""

1
tests/components/device_tracker/test_asuswrt.py
View File

@ -47,6 +47,7 @@ class TestComponentsDeviceTrackerASUSWRT(unittest.TestCase):
def teardown_method(self, _): def teardown_method(self, _):
"""Stop everything that was started.""" """Stop everything that was started."""
self.hass.stop()
try: try:
os.remove(self.hass.config.path(device_tracker.YAML_DEVICES)) os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
except FileNotFoundError: except FileNotFoundError:

1
tests/components/device_tracker/test_automatic.py
View File

@ -210,6 +210,7 @@ class TestAutomatic(unittest.TestCase):
def tearDown(self): def tearDown(self):
"""Tear down test data.""" """Tear down test data."""
self.hass.stop()
@patch('requests.get', side_effect=mocked_requests) @patch('requests.get', side_effect=mocked_requests)
@patch('requests.post', side_effect=mocked_requests) @patch('requests.post', side_effect=mocked_requests)

1
tests/components/device_tracker/test_ddwrt.py
View File

@ -43,6 +43,7 @@ class TestDdwrt(unittest.TestCase):
def teardown_method(self, _): def teardown_method(self, _):
"""Stop everything that was started.""" """Stop everything that was started."""
self.hass.stop()
try: try:
os.remove(self.hass.config.path(device_tracker.YAML_DEVICES)) os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
except FileNotFoundError: except FileNotFoundError:

1
tests/components/device_tracker/test_mqtt.py
View File

@ -24,6 +24,7 @@ class TestComponentsDeviceTrackerMQTT(unittest.TestCase):
def tearDown(self): # pylint: disable=invalid-name def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started.""" """Stop everything that was started."""
self.hass.stop()
try: try:
os.remove(self.hass.config.path(device_tracker.YAML_DEVICES)) os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
except FileNotFoundError: except FileNotFoundError:

4
tests/components/device_tracker/test_owntracks.py
View File

@ -691,6 +691,10 @@ class TestDeviceTrackerOwnTrackConfigs(BaseMQTT):
self.hass = get_test_home_assistant() self.hass = get_test_home_assistant()
mock_mqtt_component(self.hass) mock_mqtt_component(self.hass)
def teardown_method(self, method):
"""Tear down resources."""
self.hass.stop()
def mock_cipher(): # pylint: disable=no-method-argument def mock_cipher(): # pylint: disable=no-method-argument
"""Return a dummy pickle-based cipher.""" """Return a dummy pickle-based cipher."""
def mock_decrypt(ciphertext, key): def mock_decrypt(ciphertext, key):

1
tests/components/device_tracker/test_tplink.py
View File

@ -21,6 +21,7 @@ class TestTplink4DeviceScanner(unittest.TestCase):
def tearDown(self): # pylint: disable=invalid-name def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started.""" """Stop everything that was started."""
self.hass.stop()
try: try:
os.remove(self.hass.config.path(device_tracker.YAML_DEVICES)) os.remove(self.hass.config.path(device_tracker.YAML_DEVICES))
except FileNotFoundError: except FileNotFoundError:

141
tests/components/notify/test_apns.py
View File

@ -14,6 +14,14 @@ from apns2.errors import Unregistered
class TestApns(unittest.TestCase): class TestApns(unittest.TestCase):
"""Test the APNS component.""" """Test the APNS component."""
def setUp(self): # pylint: disable=invalid-name
"""Setup things to be run when tests are started."""
self.hass = get_test_home_assistant()
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
def test_apns_setup_full(self): def test_apns_setup_full(self):
"""Test setup with all data.""" """Test setup with all data."""
config = { config = {
@ -25,9 +33,8 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
self.assertTrue(notify.setup(hass, config)) self.assertTrue(notify.setup(self.hass, config))
def test_apns_setup_missing_name(self): def test_apns_setup_missing_name(self):
"""Test setup with missing name.""" """Test setup with missing name."""
@ -39,8 +46,7 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant() self.assertFalse(notify.setup(self.hass, config))
self.assertFalse(notify.setup(hass, config))
def test_apns_setup_missing_certificate(self): def test_apns_setup_missing_certificate(self):
"""Test setup with missing name.""" """Test setup with missing name."""
@ -51,8 +57,7 @@ class TestApns(unittest.TestCase):
'name': 'test_app' 'name': 'test_app'
} }
} }
hass = get_test_home_assistant() self.assertFalse(notify.setup(self.hass, config))
self.assertFalse(notify.setup(hass, config))
def test_apns_setup_missing_topic(self): def test_apns_setup_missing_topic(self):
"""Test setup with missing topic.""" """Test setup with missing topic."""
@ -63,8 +68,7 @@ class TestApns(unittest.TestCase):
'name': 'test_app' 'name': 'test_app'
} }
} }
hass = get_test_home_assistant() self.assertFalse(notify.setup(self.hass, config))
self.assertFalse(notify.setup(hass, config))
def test_register_new_device(self): def test_register_new_device(self):
"""Test registering a new device with a name.""" """Test registering a new device with a name."""
@ -76,18 +80,17 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('5678: {name: test device 2}\n') out.write('5678: {name: test device 2}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('apns', self.assertTrue(self.hass.services.call('apns',
'test_app', 'test_app',
{'push_id': '1234', {'push_id': '1234',
'name': 'test device'}, 'name': 'test device'},
blocking=True)) blocking=True))
devices = {str(key): value for (key, value) in devices = {str(key): value for (key, value) in
load_yaml_config_file(devices_path).items()} load_yaml_config_file(devices_path).items()}
@ -112,16 +115,15 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('5678: {name: test device 2}\n') out.write('5678: {name: test device 2}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('apns', 'test_app', self.assertTrue(self.hass.services.call('apns', 'test_app',
{'push_id': '1234'}, {'push_id': '1234'},
blocking=True)) blocking=True))
devices = {str(key): value for (key, value) in devices = {str(key): value for (key, value) in
load_yaml_config_file(devices_path).items()} load_yaml_config_file(devices_path).items()}
@ -143,19 +145,18 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('1234: {name: test device 1}\n') out.write('1234: {name: test device 1}\n')
out.write('5678: {name: test device 2}\n') out.write('5678: {name: test device 2}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('apns', self.assertTrue(self.hass.services.call('apns',
'test_app', 'test_app',
{'push_id': '1234', {'push_id': '1234',
'name': 'updated device 1'}, 'name': 'updated device 1'},
blocking=True)) blocking=True))
devices = {str(key): value for (key, value) in devices = {str(key): value for (key, value) in
load_yaml_config_file(devices_path).items()} load_yaml_config_file(devices_path).items()}
@ -180,21 +181,20 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('1234: {name: test device 1, ' out.write('1234: {name: test device 1, '
'tracking_device_id: tracking123}\n') 'tracking_device_id: tracking123}\n')
out.write('5678: {name: test device 2, ' out.write('5678: {name: test device 2, '
'tracking_device_id: tracking456}\n') 'tracking_device_id: tracking456}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('apns', self.assertTrue(self.hass.services.call('apns',
'test_app', 'test_app',
{'push_id': '1234', {'push_id': '1234',
'name': 'updated device 1'}, 'name': 'updated device 1'},
blocking=True)) blocking=True))
devices = {str(key): value for (key, value) in devices = {str(key): value for (key, value) in
load_yaml_config_file(devices_path).items()} load_yaml_config_file(devices_path).items()}
@ -224,23 +224,22 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('1234: {name: test device 1}\n') out.write('1234: {name: test device 1}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('notify', 'test_app', self.assertTrue(self.hass.services.call('notify', 'test_app',
{'message': 'Hello', {'message': 'Hello',
'data': { 'data': {
'badge': 1, 'badge': 1,
'sound': 'test.mp3', 'sound': 'test.mp3',
'category': 'testing' 'category': 'testing'
} }
}, },
blocking=True)) blocking=True))
self.assertTrue(send.called) self.assertTrue(send.called)
self.assertEqual(1, len(send.mock_calls)) self.assertEqual(1, len(send.mock_calls))
@ -266,23 +265,22 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('1234: {name: test device 1, disabled: True}\n') out.write('1234: {name: test device 1, disabled: True}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('notify', 'test_app', self.assertTrue(self.hass.services.call('notify', 'test_app',
{'message': 'Hello', {'message': 'Hello',
'data': { 'data': {
'badge': 1, 'badge': 1,
'sound': 'test.mp3', 'sound': 'test.mp3',
'category': 'testing' 'category': 'testing'
} }
}, },
blocking=True)) blocking=True))
self.assertFalse(send.called) self.assertFalse(send.called)
@ -291,9 +289,7 @@ class TestApns(unittest.TestCase):
"""Test updating an existing device.""" """Test updating an existing device."""
send = mock_client.return_value.send_notification send = mock_client.return_value.send_notification
hass = get_test_home_assistant() devices_path = self.hass.config.path('test_app_apns.yaml')
devices_path = hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('1234: {name: test device 1, ' out.write('1234: {name: test device 1, '
'tracking_device_id: tracking123}\n') 'tracking_device_id: tracking123}\n')
@ -301,7 +297,7 @@ class TestApns(unittest.TestCase):
'tracking_device_id: tracking456}\n') 'tracking_device_id: tracking456}\n')
notify_service = ApnsNotificationService( notify_service = ApnsNotificationService(
hass, self.hass,
'test_app', 'test_app',
'testapp.appname', 'testapp.appname',
False, False,
@ -313,7 +309,7 @@ class TestApns(unittest.TestCase):
State('device_tracker.tracking456', None), State('device_tracker.tracking456', None),
State('device_tracker.tracking456', 'home')) State('device_tracker.tracking456', 'home'))
hass.block_till_done() self.hass.block_till_done()
notify_service.send_message(message='Hello', target='home') notify_service.send_message(message='Hello', target='home')
@ -340,17 +336,16 @@ class TestApns(unittest.TestCase):
'cert_file': 'test_app.pem' 'cert_file': 'test_app.pem'
} }
} }
hass = get_test_home_assistant()
devices_path = hass.config.path('test_app_apns.yaml') devices_path = self.hass.config.path('test_app_apns.yaml')
with open(devices_path, 'w+') as out: with open(devices_path, 'w+') as out:
out.write('1234: {name: test device 1}\n') out.write('1234: {name: test device 1}\n')
notify.setup(hass, config) notify.setup(self.hass, config)
self.assertTrue(hass.services.call('notify', 'test_app', self.assertTrue(self.hass.services.call('notify', 'test_app',
{'message': 'Hello'}, {'message': 'Hello'},
blocking=True)) blocking=True))
devices = {str(key): value for (key, value) in devices = {str(key): value for (key, value) in
load_yaml_config_file(devices_path).items()} load_yaml_config_file(devices_path).items()}

4
tests/components/sensor/test_darksky.py
View File

@ -41,6 +41,10 @@ class TestDarkSkySetup(unittest.TestCase):
self.hass.config.longitude = self.lon self.hass.config.longitude = self.lon
self.entities = [] self.entities = []
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
def test_setup_with_config(self): def test_setup_with_config(self):
"""Test the platform setup with configuration.""" """Test the platform setup with configuration."""
self.assertTrue( self.assertTrue(

4
tests/components/sensor/test_sleepiq.py
View File

@ -30,6 +30,10 @@ class TestSleepIQSensorSetup(unittest.TestCase):
'password': self.password, 'password': self.password,
} }
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
@requests_mock.Mocker() @requests_mock.Mocker()
def test_setup(self, mock): def test_setup(self, mock):
"""Test for successfully setting up the SleepIQ platform.""" """Test for successfully setting up the SleepIQ platform."""

4
tests/components/sensor/test_sonarr.py
View File

@ -572,6 +572,10 @@ class TestSonarrSetup(unittest.TestCase):
self.hass = get_test_home_assistant() self.hass = get_test_home_assistant()
self.hass.config.time_zone = 'America/Los_Angeles' self.hass.config.time_zone = 'America/Los_Angeles'
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get) @unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
def test_diskspace_no_paths(self, req_mock): def test_diskspace_no_paths(self, req_mock):
"""Test getting all disk space.""" """Test getting all disk space."""

4
tests/components/sensor/test_wunderground.py
View File

@ -129,6 +129,10 @@ class TestWundergroundSetup(unittest.TestCase):
self.hass.config.latitude = self.lat self.hass.config.latitude = self.lat
self.hass.config.longitude = self.lon self.hass.config.longitude = self.lon
def tearDown(self): # pylint: disable=invalid-name
"""Stop everything that was started."""
self.hass.stop()
@unittest.mock.patch('requests.get', side_effect=mocked_requests_get) @unittest.mock.patch('requests.get', side_effect=mocked_requests_get)
def test_setup(self, req_mock): def test_setup(self, req_mock):
"""Test that the component is loaded if passed in PWS Id.""" """Test that the component is loaded if passed in PWS Id."""

4
tests/components/switch/test_command_line.py
View File

@ -161,8 +161,6 @@ class TestCommandSwitch(unittest.TestCase):
def test_assumed_state_should_be_true_if_command_state_is_false(self): def test_assumed_state_should_be_true_if_command_state_is_false(self):
"""Test with state value.""" """Test with state value."""
self.hass = get_test_home_assistant()
# args: hass, device_name, friendly_name, command_on, command_off, # args: hass, device_name, friendly_name, command_on, command_off,
# command_state, value_template # command_state, value_template
init_args = [ init_args = [
@ -186,8 +184,6 @@ class TestCommandSwitch(unittest.TestCase):
def test_entity_id_set_correctly(self): def test_entity_id_set_correctly(self):
"""Test that entity_id is set correctly from object_id.""" """Test that entity_id is set correctly from object_id."""
self.hass = get_test_home_assistant()
init_args = [ init_args = [
self.hass, self.hass,
"test_device_name", "test_device_name",

17
tests/helpers/test_config_validation.py
View File

@ -191,15 +191,20 @@ def test_event_schema():
def test_platform_validator(): def test_platform_validator():
"""Test platform validation.""" """Test platform validation."""
# Prepares loading hass = None
get_test_home_assistant()
schema = vol.Schema(cv.platform_validator('light')) try:
hass = get_test_home_assistant()
with pytest.raises(vol.MultipleInvalid): schema = vol.Schema(cv.platform_validator('light'))
schema('platform_that_does_not_exist')
schema('hue') with pytest.raises(vol.MultipleInvalid):
schema('platform_that_does_not_exist')
schema('hue')
finally:
if hass is not None:
hass.stop()
def test_icon(): def test_icon():