TEST_F(AgentConsolidatorTest, add) {
AgentConsolidatorHandle *handle;
SubscriptionRequest request;
SubscriptionRequest system_accepted_request;
telemetry::Path *path;
// Build a request of valid paths
path = request.add_path_list();
path->set_path("firewall");
path = request.add_path_list();
path->set_path("port");
// Add it to the consolidator
handle = cons->addRequest(61, &request, &system_accepted_request);
EXPECT_TRUE(handle != NULL);
EXPECT_EQ(2, cons->getSystemRequestCount());
EXPECT_EQ(1, cons->getAddCount());
EXPECT_EQ(0, cons->getErrors());
for (int i = 0; i < request.path_list_size(); i++) {
EXPECT_STREQ(request.path_list(i).path().c_str(),
system_accepted_request.path_list(i).path().c_str());
}
// Simple check get call is good
SubscriptionRequest *test_ptr;
test_ptr = cons->getRequest(handle, false);
EXPECT_TRUE(test_ptr != NULL);
// Remove it now
cons->removeRequest(handle);
EXPECT_EQ(0, cons->getSystemRequestCount());
EXPECT_EQ(1, cons->getRemCount());
EXPECT_EQ(0, cons->getErrors());
}
TEST_F(AgentConsolidatorTest, add_multiple) {
AgentConsolidatorHandle *handle1, *handle2;
SubscriptionRequest request;
SubscriptionRequest system_accepted_request;
telemetry::Path *path;
// Build a request
path = request.add_path_list();
path->set_path("firewall");
path = request.add_path_list();
path->set_path("port");
// Add it to the consolidator
handle1 = cons->addRequest(61, &request, &system_accepted_request);
EXPECT_TRUE(handle1 != NULL);
for (int i = 0; i < request.path_list_size(); i++) {
EXPECT_STREQ(request.path_list(i).path().c_str(),
system_accepted_request.path_list(i).path().c_str());
}
// Add it to the consolidator
handle2 = cons->addRequest(62, &request, &system_accepted_request);
EXPECT_TRUE(handle2 != NULL);
// Only two system requests should have been created
EXPECT_EQ(2, cons->getSystemRequestCount());
EXPECT_EQ(2, cons->getAddCount());
EXPECT_EQ(0, cons->getErrors());
// Remove it now
cons->removeRequest(handle1);
EXPECT_EQ(2, cons->getSystemRequestCount());
EXPECT_EQ(1, cons->getRemCount());
EXPECT_EQ(0, cons->getErrors());
cons->removeRequest(handle2);
EXPECT_EQ(0, cons->getSystemRequestCount());
EXPECT_EQ(2, cons->getRemCount());
EXPECT_EQ(0, cons->getErrors());
}
