def test_set_power():
bulb = lazylights.Bulb(GATEWAY, '\x01\x02\x03\x04\x05\x06',
('192.168.100.100', 56700))
with patch('lazylights.socket') as socket_mock:
lazylights.set_power([bulb], True)
socket_mock.socket.assert_called_once_with(
socket_mock.AF_INET, socket_mock.SOCK_DGRAM, socket_mock.IPPROTO_UDP)
socket_mock.socket().sendto.assert_called_once_with(
build_packet(lazylights.REQ_SET_POWER_STATE,
bulb.gateway_mac, bulb.mac, '2s', '\xff\xff'),
bulb.addr)
def test_build_packet():
packet = build_packet(lazylights.REQ_SET_POWER_STATE,
GATEWAY, lazylights.ALL_BULBS,
'2s', '\x00\x00')
eq_(packet, OFF_PACKET)
