Pipeline configuration added to mock exchange

rtbkit · Apr 28, 2016 · 0ed0953 · 0ed0953
1 parent b514580
commit 0ed0953
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git 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
@@ -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