def test_remove_listener_from_universe(): receiver, socket = get_receiver() test_universe_one = 1 test_universe_two = 2 packet_send = DataPacket(cid=tuple(range(0, 16)), sourceName='Test', universe=test_universe_one, dmxData=tuple(range(0, 16))) called = 0 def callback_packet(packet): assert packet_send.__dict__ == packet.__dict__ nonlocal called called += 1 # register listener multiple times receiver.register_listener('universe', callback_packet, universe=test_universe_one) receiver.register_listener('universe', callback_packet, universe=test_universe_two) packet_send.universe = test_universe_one socket.call_on_data(bytes(packet_send.getBytes()), 0) assert called == 1 packet_send.universe = test_universe_two socket.call_on_data(bytes(packet_send.getBytes()), 0) assert called == 2 # change DMX data to trigger a change packet_send.dmxData = tuple(range(16, 32)) packet_send.sequence_increase() test_universe_removed = test_universe_one receiver.remove_listener_from_universe(test_universe_removed) # removing from a universe that does not exist, should do nothing receiver.remove_listener_from_universe(12345) # call to the removed universe should not happen packet_send.universe = test_universe_removed socket.call_on_data(bytes(packet_send.getBytes()), 0) assert called == 2 # other universes should not be affected packet_send.universe = test_universe_two socket.call_on_data(bytes(packet_send.getBytes()), 0) assert called == 3
def test_property_adjustment_and_deconstruction(): # Converting DataPacket -> bytes -> DataPacket should produce the same result, # but with changed properties that are not the default built_packet = DataPacket(cid=(16, 1, 15, 2, 14, 3, 13, 4, 12, 5, 11, 6, 10, 7, 9, 8), sourceName='Test Name', universe=30) built_packet.cid = tuple(range(16)) built_packet.sourceName = '2nd Test Name' built_packet.universe = 31425 built_packet.dmxData = ((200, ) + tuple(range(255, 0, -1)) + tuple(range(255)) + (0, )) built_packet.priority = 12 built_packet.sequence = 45 built_packet.option_StreamTerminated = True built_packet.option_PreviewData = True built_packet.option_ForceSync = True built_packet.syncAddr = 34003 built_packet.dmxStartCode = 8 read_packet = DataPacket.make_data_packet(built_packet.getBytes()) assert read_packet.cid == tuple(range(16)) assert read_packet.sourceName == '2nd Test Name' assert read_packet.universe == 31425 assert read_packet.dmxData == ((200, ) + tuple(range(255, 0, -1)) + tuple(range(255)) + (0, )) assert read_packet.priority == 12 assert read_packet.sequence == 45 assert read_packet.option_StreamTerminated is True assert read_packet.option_PreviewData is True assert read_packet.option_ForceSync is True assert read_packet.syncAddr == 34003 assert read_packet.dmxStartCode == 8
def test_possible_universes(): handler, _, socket = get_handler() assert handler.get_possible_universes() == [] packet = DataPacket(cid=tuple(range(0, 16)), sourceName='Test', universe=1, dmxData=tuple(range(0, 16))) # add universe 1 socket.call_on_data(bytes(packet.getBytes()), 0) assert handler.get_possible_universes() == [1] # add universe 2 packet.universe = 2 socket.call_on_data(bytes(packet.getBytes()), 0) assert handler.get_possible_universes() == [1, 2] # remove universe 2 packet.option_StreamTerminated = True socket.call_on_data(bytes(packet.getBytes()), 0) assert handler.get_possible_universes() == [1]