You've already forked homehub
Release 1.4.0
This commit is contained in:
@@ -357,22 +357,22 @@ class Test_TradfriService:
|
||||
service.worker.return_value = fake_worker
|
||||
|
||||
with mock.patch('pytradfri.Gateway') as mock_gateway:
|
||||
with mock.patch.object(pytradfri_api, 'APIFactory') as mock_api_factory:
|
||||
with mock.patch.object(pytradfri_api.APIFactory, 'init') as mock_api_factory_init:
|
||||
with mock.patch.object(service, 'set_state'):
|
||||
mock_gateway.return_value = fake_gateway
|
||||
mock_api_factory.return_value = fake_api_factory
|
||||
mock_api_factory_init.return_value = fake_api_factory
|
||||
|
||||
# When
|
||||
await service.start()
|
||||
|
||||
# Then
|
||||
assert service.api_factory == fake_api_factory
|
||||
pytradfri_api.APIFactory.assert_called_with(
|
||||
pytradfri_api.APIFactory.init.assert_called_with(
|
||||
host=characteristics['host'],
|
||||
psk_id=mock.ANY,
|
||||
psk=None,
|
||||
)
|
||||
psk_id = pytradfri_api.APIFactory.call_args.kwargs['psk_id']
|
||||
psk_id = pytradfri_api.APIFactory.init.call_args.kwargs['psk_id']
|
||||
assert isinstance(psk_id, str)
|
||||
|
||||
fake_api_factory.generate_psk.assert_called_with(
|
||||
@@ -419,17 +419,17 @@ class Test_TradfriService:
|
||||
service.worker.return_value = fake_worker
|
||||
|
||||
with mock.patch('pytradfri.Gateway') as mock_gateway:
|
||||
with mock.patch.object(pytradfri_api, 'APIFactory') as mock_api_factory:
|
||||
with mock.patch.object(pytradfri_api.APIFactory, 'init') as mock_api_factory_init:
|
||||
with mock.patch.object(service, 'set_state'):
|
||||
mock_gateway.return_value = fake_gateway
|
||||
mock_api_factory.return_value = fake_api_factory
|
||||
mock_api_factory_init.return_value = fake_api_factory
|
||||
|
||||
# When
|
||||
await service.start()
|
||||
|
||||
# Then
|
||||
assert service.api_factory == fake_api_factory
|
||||
mock_api_factory.assert_called_with(
|
||||
mock_api_factory_init.assert_called_with(
|
||||
host=characteristics['host'],
|
||||
psk_id='state_psk_id',
|
||||
psk='state_psk',
|
||||
@@ -464,14 +464,14 @@ class Test_TradfriService:
|
||||
service.api_factory = fake_api_factory
|
||||
|
||||
with mock.patch('pytradfri.Gateway') as mock_gateway:
|
||||
with mock.patch.object(pytradfri_api, 'APIFactory') as mock_api_factory:
|
||||
with mock.patch.object(pytradfri_api.APIFactory, 'init') as mock_api_factory_init:
|
||||
with mock.patch.object(service, 'set_state'):
|
||||
# When
|
||||
await service.start()
|
||||
|
||||
# Then
|
||||
assert service.api_factory == fake_api_factory
|
||||
assert not mock_api_factory.called
|
||||
assert not mock_api_factory_init.called
|
||||
|
||||
assert not fake_api_factory.generate_psk.called
|
||||
|
||||
@@ -503,16 +503,13 @@ class Test_TradfriService:
|
||||
fake_api_factory = mock.Mock(spec=pytradfri_api.APIFactory)
|
||||
service.api_factory = fake_api_factory
|
||||
|
||||
with mock.patch('pytradfri.Gateway'):
|
||||
with mock.patch.object(pytradfri_api, 'APIFactory'):
|
||||
with mock.patch.object(service, 'set_state'):
|
||||
# When
|
||||
await service.stop()
|
||||
# When
|
||||
await service.stop()
|
||||
|
||||
# Then
|
||||
assert service.api_factory is None
|
||||
assert service.gateway is None
|
||||
assert service.job.close.called
|
||||
# Then
|
||||
assert service.api_factory is None
|
||||
assert service.gateway is None
|
||||
assert service.job.close.called
|
||||
|
||||
async def test_stop_already_stopped(self, homehub_app):
|
||||
# Given
|
||||
@@ -531,11 +528,8 @@ class Test_TradfriService:
|
||||
fake_api_factory = mock.Mock(spec=pytradfri_api.APIFactory)
|
||||
service.api_factory = fake_api_factory
|
||||
|
||||
with mock.patch('pytradfri.Gateway'):
|
||||
with mock.patch.object(pytradfri_api, 'APIFactory'):
|
||||
with mock.patch.object(service, 'set_state'):
|
||||
# When
|
||||
await service.stop()
|
||||
# When
|
||||
await service.stop()
|
||||
|
||||
# Then
|
||||
assert not service.job.close.called
|
||||
# Then
|
||||
assert not service.job.close.called
|
||||
|
||||
Reference in New Issue
Block a user