diff options
Diffstat (limited to 'zenserver/testing/httptest.h')
| -rw-r--r-- | zenserver/testing/httptest.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/zenserver/testing/httptest.h b/zenserver/testing/httptest.h index f7ea0c31c..267d59b36 100644 --- a/zenserver/testing/httptest.h +++ b/zenserver/testing/httptest.h @@ -5,6 +5,7 @@ #include <zencore/logging.h> #include <zencore/stats.h> #include <zenhttp/httpserver.h> +#include <zenhttp/websocket.h> #include <atomic> @@ -13,7 +14,7 @@ namespace zen { /** * Test service to facilitate testing the HTTP framework and client interactions */ -class HttpTestingService : public HttpService +class HttpTestingService : public HttpService, public WebSocketService { public: HttpTestingService(); @@ -22,6 +23,7 @@ public: virtual const char* BaseUri() const override; virtual void HandleRequest(HttpServerRequest& Request) override; virtual Ref<IHttpPackageHandler> HandlePackageRequest(HttpServerRequest& HttpServiceRequest) override; + virtual bool HandleMessage(WebSocketId SocketId, const CbPackage& Msg) override; class PackageHandler : public IHttpPackageHandler { |