Skip to content

Commit

Permalink
Fix test errors
Browse files Browse the repository at this point in the history
  • Loading branch information
shenxn committed Jan 22, 2021
1 parent 35da89b commit e599d4e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/test_device.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,16 @@
from libpurecool.dyson_360_eye import Dyson360Eye
from paho.mqtt.client import Client
import pytest
import socket
from libpurecool.zeroconf import ServiceInfo, Zeroconf
from unittest import mock
from unittest.mock import MagicMock
import paho.mqtt.client as mqtt

from libpurecool.dyson_pure_cool_link import DysonPureCoolLink

IP_ADDRESS = "192.168.1.2"
SERIAL = "XXX-XX-XXXXXXXX"


def _mocked_zeroconf():
def _get_service_info(*args):
service_info = MagicMock(spec=ServiceInfo)
Expand All @@ -23,9 +22,10 @@ def _get_service_info(*args):
zeroconf.get_service_info = MagicMock(side_effect=_get_service_info)
return zeroconf


def _get_mocked_service_browser(serial):
def _mocked_service_browser(zeroconf, type, listener):
listener.add_service(zeroconf, type, f"{serial}.{type}")
listener.add_service(zeroconf, type, "{}.{}".format(serial, type))
return _mocked_service_browser


Expand Down Expand Up @@ -65,4 +65,4 @@ def test_auto_connect(device_class, serial_prefix):
assert connected is True
assert device.state is None
if hasattr(device, "disconnect"):
device.disconnect()
device.disconnect()

0 comments on commit e599d4e

Please sign in to comment.