From 0ed095353ba61b7e709be446ff4d6457b49d73fc Mon Sep 17 00:00:00 2001 From: Sirma Date: Thu, 28 Apr 2016 19:00:17 -0400 Subject: [PATCH] Pipeline configuration added to mock exchange --- rtbkit/examples/mock_exchange_connector.h | 2 ++ rtbkit/examples/rtbkit_integration_test.cc | 7 +++++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/rtbkit/examples/mock_exchange_connector.h b/rtbkit/examples/mock_exchange_connector.h index e49395ad1..2774e1469 100644 --- a/rtbkit/examples/mock_exchange_connector.h +++ b/rtbkit/examples/mock_exchange_connector.h @@ -46,6 +46,8 @@ struct MockExchangeConnector : public HttpExchangeConnector { getParam(parameters, bindHost, "bindHost"); getParam(parameters, performNameLookup, "performNameLookup"); getParam(parameters, backlog, "connectionBacklog"); + + configurePipeline(parameters["pipeline"]); } void start() { diff --git a/rtbkit/examples/rtbkit_integration_test.cc b/rtbkit/examples/rtbkit_integration_test.cc index 536971bd0..159d0132e 100644 --- a/rtbkit/examples/rtbkit_integration_test.cc +++ b/rtbkit/examples/rtbkit_integration_test.cc @@ -315,7 +315,7 @@ struct Components for (auto& connector : exchangeConnectors) { connector->enableUntil(Date::positiveInfinity()); - + connector->configurePipeline(Json::Value::null); int port = connector->init(ports, "localhost", 2 /* threads */); exchangePorts.push_back(port); } @@ -429,7 +429,10 @@ int main(int argc, char ** argv) NetworkAddress bids(components.exchangePorts[i % components.exchangePorts.size()]); NetworkAddress wins(components.winStreamPort); NetworkAddress events(components.eventStreamPort); - exchange.add(new MockBidSource(bids, nBidRequestsPerThread), new MockWinSource(wins), new MockEventSource(events)); + + exchange.add(new MockBidSource(bids, nBidRequestsPerThread), + new MockWinSource(wins), + new MockEventSource(events)); } // Dump the budget stats while we wait for the test to finish. Only the