Brian Martin
87 lines
4.5 KiB
Python
87 lines
4.5 KiB
Python
from re import compile
|
|
|
|
import requests_mock
|
|
from nose.tools import *
|
|
|
|
from sucks import *
|
|
from test_ecovacs_api import make_api
|
|
|
|
|
|
# There are few tests for the IOT stuff here because it's relatively complicated to test given
|
|
# the library's design and its multithreaded nature and lack of explicit testing support.
|
|
|
|
def test_wrap_command():
|
|
x = make_ecovacs_iot()
|
|
|
|
c = x._wrap_command(Charge(), 'E0000000001234567890')
|
|
assert_equal(c['cmdName'], Charge().name)
|
|
assert_equal(c['toId'], 'E0000000001234567890')
|
|
assert_equal(c['payload'], '<ctl><charge type="go" /></ctl>')
|
|
|
|
def test_is_iot():
|
|
x = make_ecovacs_iot()
|
|
|
|
# TODO - Error response from command
|
|
#'cmdName': 'Charge', 'payload': '<ctl><charge type="go" /></ctl>', 'payloadType': 'x', 'td': 'q', 'toId': '0e084f6c-0846-4342-a947-fe14c293301f', 'toRes': 'wC3g', 'toType': 'ls1ok3'}
|
|
# - Already charging on dock
|
|
#{'ret': 'ok', 'resp': "<ctl ret='fail' errno='8'/>", 'id': 'NLQy'}
|
|
|
|
#Timeout
|
|
# {'ret': 'fail', 'errno': 500, 'debug': 'wait for response timed out'}
|
|
|
|
# def test_subscribe_to_ctls():
|
|
# response = None
|
|
|
|
# def save_response(value):
|
|
# nonlocal response
|
|
# response = value
|
|
|
|
# x = make_ecovacs_iot()
|
|
|
|
# query = x.make_iq_query()
|
|
# query.set_payload(
|
|
# ET.fromstring('<query xmlns="com:ctl"><ctl td="CleanReport"> <clean type="auto" /> </ctl></query>'))
|
|
|
|
# x.subscribe_to_ctls(save_response)
|
|
# x._handle_ctl(query)
|
|
# assert_dict_equal(response, {'event': 'clean_report', 'type': 'auto'})
|
|
|
|
|
|
# def test_xml_to_dict():
|
|
# x = make_ecovacs_iot()
|
|
|
|
# assert_dict_equal(
|
|
# x._ctl_to_dict(make_ctl('<ctl td="CleanReport"> <clean type="auto" /> </ctl>')),
|
|
# {'event': 'clean_report', 'type': 'auto'})
|
|
# assert_dict_equal(
|
|
# x._ctl_to_dict(make_ctl('<ctl td="CleanReport"> <clean type="auto" speed="strong" /> </ctl>')),
|
|
# {'event': 'clean_report', 'type': 'auto', 'speed': 'strong'})
|
|
|
|
# assert_dict_equal(
|
|
# x._ctl_to_dict(make_ctl('<ctl td="BatteryInfo"><battery power="095"/></ctl>')),
|
|
# {'event': 'battery_info', 'power': '095'})
|
|
|
|
# assert_dict_equal(
|
|
# x._ctl_to_dict(make_ctl('# <ctl td="LifeSpan" type="Brush" val="099" total="365"/>')),
|
|
# {'event': 'life_span', 'type': 'brush', 'val': '099', 'total': '365'})
|
|
|
|
|
|
def make_ecovacs_iot():
|
|
eapi = make_api()
|
|
|
|
with requests_mock.mock() as m:
|
|
device_resource = 'test_resource'
|
|
device_class = 'ls1ok3' #this is for a D900 series
|
|
r = m.post(compile('user.do'),
|
|
text='{"todo": "result", "devices": [{"did": "E0000000001234567890", "class": "%s", "nick": "bob"}], "result": "ok"}' % (device_class))
|
|
r = m.post(compile('pim/product/getProductIotMap'),
|
|
text='{"code":0,"data":[{"classid":"dl8fht","product":{"_id":"5acb0fa87c295c0001876ecf","name":"DEEBOT 600 Series","icon":"5acc32067c295c0001876eea","UILogicId":"dl8fht","ota":false,"iconUrl":"https://portal-ww.ecouser.net/api/pim/file/get/5acc32067c295c0001876eea"}},{"classid":"02uwxm","product":{"_id":"5ae1481e7ccd1a0001e1f69e","name":"DEEBOT OZMO Slim10 Series","icon":"5b1dddc48bc45700014035a1","UILogicId":"02uwxm","ota":false,"iconUrl":"https://portal-ww.ecouser.net/api/pim/file/get/5b1dddc48bc45700014035a1"}},{"classid":"y79a7u","product":{"_id":"5b04c0227ccd1a0001e1f6a8","name":"DEEBOT OZMO 900","icon":"5b04c0217ccd1a0001e1f6a7","UILogicId":"y79a7u","ota":true,"iconUrl":"https://portal-ww.ecouser.net/api/pim/file/get/5b04c0217ccd1a0001e1f6a7"}},{"classid":"jr3pqa","product":{"_id":"5b43077b8bc457000140363e","name":"DEEBOT 711","icon":"5b5ac4cc8d5a56000111e769","UILogicId":"jr3pqa","ota":true,"iconUrl":"https://portal-ww.ecouser.net/api/pim/file/get/5b5ac4cc8d5a56000111e769"}},{"classid":"uv242z","product":{"_id":"5b5149b4ac0b87000148c128","name":"DEEBOT 710","icon":"5b5ac4e45f21100001882bb9","UILogicId":"uv242z","ota":true,"iconUrl":"https://portal-ww.ecouser.net/api/pim/file/get/5b5ac4e45f21100001882bb9"}},{"classid":"ls1ok3","product":{"_id":"5b6561060506b100015c8868","name":"DEEBOT 900 Series","icon":"5ba4a2cb6c2f120001c32839","UILogicId":"ls1ok3","ota":true,"iconUrl":"https://portal-ww.ecouser.net/api/pim/file/get/5ba4a2cb6c2f120001c32839"}}]}')
|
|
d = eapi.devices()
|
|
|
|
eiotvacuum = d[0]
|
|
eiotvacuum['resource'] = device_resource
|
|
return EcoVacsIOT('20170101abcdefabcdefa', 'ecouser.net', 'abcdef12', 'base64base64base64base64base64ba', 'na', eiotvacuum)
|
|
|
|
def make_ctl(string):
|
|
return ET.fromstring('<query xmlns="com:ctl">' + string + '</query>')[0]
|