Package filtering related test code

2 files changed, 57 insertions, 2 deletions

diff --git a/zenserver/testing/httptest.cpp b/zenserver/testing/httptest.cpp
index 653f76046..d1955ca27 100644
--- a/zenserver/testing/httptest.cpp
+++ b/zenserver/testing/httptest.cpp
@@ -36,4 +36,46 @@ HttpTestingService::HandleRequest(HttpServerRequest& Request)
 	m_Router.HandleRequest(Request);
 }
 
+//////////////////////////////////////////////////////////////////////////
+
+Ref<IHttpPackageHandler>
+HttpTestingService::HandlePackageRequest(HttpServerRequest& HttpServiceRequest)
+{
+	return new PackageHandler(HttpServiceRequest);
+}
+
+HttpTestingService::PackageHandler::PackageHandler(const HttpServerRequest& Request)
+{
+	ZEN_UNUSED(Request);
+}
+
+HttpTestingService::PackageHandler::~PackageHandler()
+{
+}
+
+void
+HttpTestingService::PackageHandler::FilterOffer(std::vector<IoHash>& OfferCids)
+{
+	ZEN_UNUSED(OfferCids);
+	// No-op
+	return;
+}
+void
+HttpTestingService::PackageHandler::OnBeginChunks()
+{
+}
+
+zen::IoBuffer
+HttpTestingService::PackageHandler::CreateTarget(const IoHash& Cid, uint64_t StorageSize)
+{
+	ZEN_UNUSED(Cid, StorageSize);
+
+	return {};
+}
+
+void
+HttpTestingService::PackageHandler::OnEndChunks()
+{
+}
+
 }  // namespace zen
diff --git a/zenserver/testing/httptest.h b/zenserver/testing/httptest.h
index 18652db71..2f3482abc 100644
--- a/zenserver/testing/httptest.h
+++ b/zenserver/testing/httptest.h
@@ -17,8 +17,21 @@ public:
 	HttpTestingService();
 	~HttpTestingService();
 
-	virtual const char* BaseUri() const override;
-	virtual void		HandleRequest(HttpServerRequest& Request) override;
+	virtual const char*				 BaseUri() const override;
+	virtual void					 HandleRequest(HttpServerRequest& Request) override;
+	virtual Ref<IHttpPackageHandler> HandlePackageRequest(HttpServerRequest& HttpServiceRequest) override;
+
+	class PackageHandler : public IHttpPackageHandler
+	{
+	public:
+		PackageHandler(const HttpServerRequest& Request);
+		~PackageHandler();
+
+		void	 FilterOffer(std::vector<IoHash>& OfferCids) override;
+		void	 OnBeginChunks() override;
+		IoBuffer CreateTarget(const IoHash& Cid, uint64_t StorageSize) override;
+		void	 OnEndChunks() override;
+	};
 
 private:
 	HttpRequestRouter m_Router;