/*
* Check that we update when ports have new data
*/
void UniverseTest::testReceiveDmx() {
ola::PortBroker broker;
ola::PortManager port_manager(m_store, &broker);
TimeStamp time_stamp;
MockSelectServer ss(&time_stamp);
ola::PluginAdaptor plugin_adaptor(NULL, &ss, NULL, NULL, NULL, NULL);
MockDevice device(NULL, "foo");
TestMockInputPort port(&device, 1, &plugin_adaptor); // input port
port_manager.PatchPort(&port, TEST_UNIVERSE);
Universe *universe = m_store->GetUniverseOrCreate(TEST_UNIVERSE);
OLA_ASSERT(universe);
OLA_ASSERT_EQ((unsigned int) 1, universe->InputPortCount());
OLA_ASSERT_EQ((unsigned int) 0, universe->OutputPortCount());
OLA_ASSERT(universe->IsActive());
// Setup the port with some data, and check that signalling the universe
// works.
m_clock.CurrentTime(&time_stamp);
port.WriteDMX(m_buffer);
port.DmxChanged();
OLA_ASSERT_EQ(ola::dmx::SOURCE_PRIORITY_DEFAULT, universe->ActivePriority());
OLA_ASSERT_EQ(m_buffer.Size(), universe->GetDMX().Size());
OLA_ASSERT(m_buffer == universe->GetDMX());
// Remove the port from the universe
universe->RemovePort(&port);
OLA_ASSERT_FALSE(universe->IsActive());
OLA_ASSERT_EQ((unsigned int) 0, universe->InputPortCount());
OLA_ASSERT_EQ((unsigned int) 0, universe->OutputPortCount());
}
/*
* Check that SendDmx updates all ports
*/
void UniverseTest::testSendDmx() {
Universe *universe = m_store->GetUniverseOrCreate(TEST_UNIVERSE);
OLA_ASSERT(universe);
TestMockOutputPort port(NULL, 1); // output port
universe->AddPort(&port);
OLA_ASSERT_EQ((unsigned int) 0, universe->InputPortCount());
OLA_ASSERT_EQ((unsigned int) 1, universe->OutputPortCount());
OLA_ASSERT(universe->IsActive());
// send some data to the universe and check the port gets it
OLA_ASSERT(universe->SetDMX(m_buffer));
OLA_ASSERT(m_buffer == port.ReadDMX());
// remove the port from the universe
universe->RemovePort(&port);
OLA_ASSERT_EQ((unsigned int) 0, universe->InputPortCount());
OLA_ASSERT_EQ((unsigned int) 0, universe->OutputPortCount());
OLA_ASSERT_FALSE(universe->IsActive());
}
/*
* Returns information on the active universes.
*/
void OlaServerServiceImpl::GetUniverseInfo(
RpcController* controller,
const OptionalUniverseRequest* request,
UniverseInfoReply* response,
ola::rpc::RpcService::CompletionCallback* done) {
ClosureRunner runner(done);
UniverseInfo *universe_info;
if (request->has_universe()) {
// return info for a single universe
Universe *universe = m_universe_store->GetUniverse(request->universe());
if (!universe)
return MissingUniverseError(controller);
universe_info = response->add_universe();
universe_info->set_universe(universe->UniverseId());
universe_info->set_name(universe->Name());
universe_info->set_merge_mode(universe->MergeMode() == Universe::MERGE_HTP
? ola::proto::HTP: ola::proto::LTP);
universe_info->set_input_port_count(universe->InputPortCount());
universe_info->set_output_port_count(universe->OutputPortCount());
universe_info->set_rdm_devices(universe->UIDCount());
} else {
// return all
vector<Universe*> uni_list;
m_universe_store->GetList(&uni_list);
vector<Universe*>::const_iterator iter;
for (iter = uni_list.begin(); iter != uni_list.end(); ++iter) {
universe_info = response->add_universe();
universe_info->set_universe((*iter)->UniverseId());
universe_info->set_name((*iter)->Name());
universe_info->set_merge_mode((*iter)->MergeMode() == Universe::MERGE_HTP
? ola::proto::HTP: ola::proto::LTP);
universe_info->set_input_port_count((*iter)->InputPortCount());
universe_info->set_output_port_count((*iter)->OutputPortCount());
universe_info->set_rdm_devices((*iter)->UIDCount());
}
}
}
/**
* Test RDM discovery for a universe/
*/
void UniverseTest::testRDMDiscovery() {
Universe *universe = m_store->GetUniverseOrCreate(TEST_UNIVERSE);
OLA_ASSERT(universe);
// check the uid set is initially empty
UIDSet universe_uids;
universe->GetUIDs(&universe_uids);
OLA_ASSERT_EQ(0u, universe_uids.Size());
UID uid1(0x7a70, 1);
UID uid2(0x7a70, 2);
UID uid3(0x7a70, 3);
UIDSet port1_uids, port2_uids;
port1_uids.AddUID(uid1);
port2_uids.AddUID(uid2);
TestMockRDMOutputPort port1(NULL, 1, &port1_uids);
// this port is configured to update the uids on patch
TestMockRDMOutputPort port2(NULL, 2, &port2_uids, true);
universe->AddPort(&port1);
port1.SetUniverse(universe);
universe->AddPort(&port2);
port2.SetUniverse(universe);
OLA_ASSERT_EQ((unsigned int) 0, universe->InputPortCount());
OLA_ASSERT_EQ((unsigned int) 2, universe->OutputPortCount());
universe->GetUIDs(&universe_uids);
OLA_ASSERT_EQ(1u, universe_uids.Size());
OLA_ASSERT(universe_uids.Contains(uid2));
OLA_ASSERT(universe->IsActive());
// now trigger discovery
UIDSet expected_uids;
expected_uids.AddUID(uid1);
expected_uids.AddUID(uid2);
universe->RunRDMDiscovery(
NewSingleCallback(this, &UniverseTest::ConfirmUIDs, &expected_uids),
true);
// now add a uid to one port, and remove a uid from another
port1_uids.AddUID(uid3);
port2_uids.RemoveUID(uid2);
expected_uids.AddUID(uid3);
expected_uids.RemoveUID(uid2);
universe->RunRDMDiscovery(
NewSingleCallback(this, &UniverseTest::ConfirmUIDs, &expected_uids),
true);
// remove the first port from the universe and confirm there are no more UIDs
universe->RemovePort(&port1);
expected_uids.Clear();
universe->RunRDMDiscovery(
NewSingleCallback(this, &UniverseTest::ConfirmUIDs, &expected_uids),
true);
universe_uids.Clear();
universe->GetUIDs(&universe_uids);
OLA_ASSERT_EQ(0u, universe_uids.Size());
universe->RemovePort(&port2);
OLA_ASSERT_EQ((unsigned int) 0, universe->InputPortCount());
OLA_ASSERT_EQ((unsigned int) 0, universe->OutputPortCount());
OLA_ASSERT_FALSE(universe->IsActive());
}
/*
* Check that HTP merging works correctly
*/
void UniverseTest::testHtpMerging() {
DmxBuffer buffer1, buffer2, htp_buffer;
buffer1.SetFromString("1,0,0,10");
buffer2.SetFromString("0,255,0,5,6,7");
htp_buffer.SetFromString("1,255,0,10,6,7");
ola::PortBroker broker;
ola::PortManager port_manager(m_store, &broker);
TimeStamp time_stamp;
MockSelectServer ss(&time_stamp);
ola::PluginAdaptor plugin_adaptor(NULL, &ss, NULL, NULL, NULL, NULL);
MockDevice device(NULL, "foo");
MockDevice device2(NULL, "bar");
TestMockInputPort port(&device, 1, &plugin_adaptor); // input port
TestMockInputPort port2(&device2, 1, &plugin_adaptor); // input port
port_manager.PatchPort(&port, TEST_UNIVERSE);
port_manager.PatchPort(&port2, TEST_UNIVERSE);
Universe *universe = m_store->GetUniverseOrCreate(TEST_UNIVERSE);
OLA_ASSERT(universe);
universe->SetMergeMode(Universe::MERGE_HTP);
OLA_ASSERT_EQ(universe->OutputPortCount(), (unsigned int) 0);
OLA_ASSERT_EQ(universe->OutputPortCount(), (unsigned int) 0);
OLA_ASSERT(universe->IsActive());
OLA_ASSERT_EQ((unsigned int) 0, universe->GetDMX().Size());
// Setup the ports with some data, and check that signalling the universe
// works.
m_clock.CurrentTime(&time_stamp);
port.WriteDMX(buffer1);
port.DmxChanged();
OLA_ASSERT_EQ(ola::dmx::SOURCE_PRIORITY_DEFAULT, universe->ActivePriority());
OLA_ASSERT_EQ(buffer1.Size(), universe->GetDMX().Size());
OLA_ASSERT(buffer1 == universe->GetDMX());
// Now the second port gets data
m_clock.CurrentTime(&time_stamp);
port2.WriteDMX(buffer2);
port2.DmxChanged();
OLA_ASSERT_EQ(ola::dmx::SOURCE_PRIORITY_DEFAULT, universe->ActivePriority());
OLA_ASSERT_EQ(htp_buffer.Size(), universe->GetDMX().Size());
OLA_ASSERT(htp_buffer == universe->GetDMX());
// now raise the priority of the second port
uint8_t new_priority = 120;
port2.SetPriority(new_priority);
m_clock.CurrentTime(&time_stamp);
port2.DmxChanged();
OLA_ASSERT_EQ(new_priority, universe->ActivePriority());
OLA_ASSERT_EQ(buffer2.Size(), universe->GetDMX().Size());
OLA_ASSERT(buffer2 == universe->GetDMX());
// raise the priority of the first port
port.SetPriority(new_priority);
m_clock.CurrentTime(&time_stamp);
port.DmxChanged();
OLA_ASSERT_EQ(new_priority, universe->ActivePriority());
OLA_ASSERT_EQ(htp_buffer.Size(), universe->GetDMX().Size());
OLA_ASSERT(htp_buffer == universe->GetDMX());
// now check a client
DmxBuffer client_buffer;
client_buffer.SetFromString("255,0,0,255,10");
m_clock.CurrentTime(&time_stamp);
ola::DmxSource source(client_buffer, time_stamp, new_priority);
MockClient input_client;
input_client.DMXReceived(TEST_UNIVERSE, source);
universe->SourceClientDataChanged(&input_client);
DmxBuffer client_htp_merge_result;
client_htp_merge_result.SetFromString("255,255,0,255,10,7");
OLA_ASSERT_EQ(new_priority, universe->ActivePriority());
OLA_ASSERT_EQ(client_htp_merge_result.Size(),
universe->GetDMX().Size());
OLA_ASSERT(client_htp_merge_result == universe->GetDMX());
// clean up
universe->RemoveSourceClient(&input_client);
universe->RemovePort(&port);
universe->RemovePort(&port2);
OLA_ASSERT_FALSE(universe->IsActive());
}