Torbjörn Axelsson
123 lines
4.4 KiB
Python
123 lines
4.4 KiB
Python
from xml.etree import ElementTree
|
|
|
|
from nose.tools import *
|
|
|
|
from sucks import *
|
|
|
|
|
|
def test_custom_command():
|
|
# Ensure a custom-built command generates the expected XML payload
|
|
c = VacBotCommand('CustomCommand', {'type': 'customtype'})
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="CustomCommand" type="customtype" />')
|
|
|
|
|
|
def test_custom_command_inner_tag():
|
|
# Ensure a custom-built command generates the expected XML payload
|
|
c = VacBotCommand('CustomCommand', {'customtag': {'customvar': 'customvalue'}})
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="CustomCommand"><customtag customvar="customvalue" /></ctl>')
|
|
|
|
|
|
def test_custom_command_noargs():
|
|
# Ensure a custom-built command with no args generates XML without an args element
|
|
c = VacBotCommand('CustomCommand')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="CustomCommand" />')
|
|
|
|
|
|
def test_clean_command():
|
|
c = Clean()
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Clean"><clean speed="standard" type="auto" /></ctl>') # protocol has attribs in other order
|
|
c = Clean('edge', 'high', 10, terminal=True)
|
|
assert_equals(c.wait, 10)
|
|
assert_equals(c.terminal, True)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Clean"><clean speed="strong" type="border" /></ctl>') # protocol has attribs in other order
|
|
|
|
|
|
def test_edge_command():
|
|
c = Edge(wait=10)
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(c.wait, 10)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Clean"><clean speed="strong" type="border" /></ctl>') # protocol has attribs in other order
|
|
|
|
|
|
def test_spot_command():
|
|
c = Spot(wait=5)
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(c.wait, 5)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Clean"><clean speed="strong" type="spot" /></ctl>') # protocol has attribs in other order
|
|
|
|
|
|
def test_charge_command():
|
|
c = Charge(terminal=True)
|
|
assert_equals(c.terminal, True)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Charge"><charge type="go" /></ctl>')
|
|
|
|
|
|
def test_stop_command():
|
|
c = Stop(terminal=True)
|
|
assert_equals(c.terminal, True)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Clean"><clean speed="standard" type="stop" /></ctl>')
|
|
|
|
|
|
def test_get_clean_state_command():
|
|
c = GetCleanState()
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="GetCleanState" />')
|
|
|
|
|
|
def test_get_charge_state_command():
|
|
c = GetChargeState()
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="GetChargeState" />')
|
|
|
|
|
|
def test_get_battery_state_command():
|
|
c = GetBatteryState()
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="GetBatteryInfo" />')
|
|
|
|
|
|
|
|
def test_move_command():
|
|
c = Move(action='left')
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Move"><move action="SpinLeft" /></ctl>')
|
|
c = Move(action='right')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Move"><move action="SpinRight" /></ctl>')
|
|
c = Move(action='turn_around')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Move"><move action="TurnAround" /></ctl>')
|
|
c = Move(action='forward')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Move"><move action="forward" /></ctl>')
|
|
c = Move(action='stop')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="Move"><move action="stop" /></ctl>')
|
|
|
|
|
|
def test_get_lifepsan_command():
|
|
c = GetLifeSpan('main_brush')
|
|
assert_equals(c.terminal, False)
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="GetLifeSpan" type="Brush" />')
|
|
c = GetLifeSpan('side_brush')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="GetLifeSpan" type="SideBrush" />')
|
|
c = GetLifeSpan('filter')
|
|
assert_equals(ElementTree.tostring(c.to_xml()),
|
|
b'<ctl td="GetLifeSpan" type="DustCaseHeap" />')
|