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)