summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--test/UnitOAuth.cpp106
-rw-r--r--test/WopiTestServer.hpp115
2 files changed, 135 insertions, 86 deletions
diff --git a/test/UnitOAuth.cpp b/test/UnitOAuth.cpp
index 7a52c1ee2..baf05c0e9 100644
--- a/test/UnitOAuth.cpp
+++ b/test/UnitOAuth.cpp
@@ -9,24 +9,18 @@
#include "config.h"
-//#include "Exceptions.hpp"
+#include "WopiTestServer.hpp"
#include "Log.hpp"
#include "Unit.hpp"
#include "UnitHTTP.hpp"
#include "helpers.hpp"
-#include <Poco/JSON/Object.h>
-#include <Poco/LocalDateTime.h>
-#include <Poco/DateTimeFormat.h>
-#include <Poco/DateTimeFormatter.h>
#include <Poco/Net/HTTPRequest.h>
#include <Poco/Net/OAuth20Credentials.h>
#include <Poco/Util/LayeredConfiguration.h>
-using Poco::DateTimeFormatter;
-using Poco::DateTimeFormat;
using Poco::Net::OAuth20Credentials;
-class UnitOAuth : public UnitWSD
+class UnitOAuth : public WopiTestServer
{
enum class Phase
{
@@ -46,6 +40,7 @@ public:
{
}
+ /// The actual assert of the authentication.
void assertRequest(const Poco::Net::HTTPRequest& request, int fileIndex)
{
// check that the request contains the Authorization: header
@@ -68,91 +63,30 @@ public:
}
}
- /// Here we act as a WOPI server, so that we have a server that responds to
- /// the wopi requests without additional expensive setup.
- virtual bool handleHttpRequest(const Poco::Net::HTTPRequest& request, std::shared_ptr<StreamSocket>& socket) override
+ void assertCheckFileInfoRequest(const Poco::Net::HTTPRequest& request) override
{
- static const std::string hello("Hello, world");
-
- Poco::URI uriReq(request.getURI());
- LOG_INF("Fake wopi host request: " << uriReq.toString());
+ std::string path = Poco::URI(request.getURI()).getPath();
+ assertRequest(request, (path == "/wopi/files/0")? 0: 1);
+ }
- // CheckFileInfo
- if (uriReq.getPath() == "/wopi/files/0" || uriReq.getPath() == "/wopi/files/1")
+ void assertGetFileRequest(const Poco::Net::HTTPRequest& request) override
+ {
+ std::string path = Poco::URI(request.getURI()).getPath();
+ if (path == "/wopi/files/0/contents")
{
- LOG_INF("Fake wopi host request, handling CheckFileInfo: " << uriReq.getPath());
-
- assertRequest(request, (uriReq.getPath() == "/wopi/files/0")? 0: 1);
-
- Poco::LocalDateTime now;
- Poco::JSON::Object::Ptr fileInfo = new Poco::JSON::Object();
- fileInfo->set("BaseFileName", "hello.txt");
- fileInfo->set("Size", hello.size());
- fileInfo->set("Version", "1.0");
- fileInfo->set("OwnerId", "test");
- fileInfo->set("UserId", "test");
- fileInfo->set("UserFriendlyName", "test");
- fileInfo->set("UserCanWrite", "true");
- fileInfo->set("PostMessageOrigin", "localhost");
- fileInfo->set("LastModifiedTime", DateTimeFormatter::format(now, DateTimeFormat::ISO8601_FORMAT));
-
- std::ostringstream jsonStream;
- fileInfo->stringify(jsonStream);
- std::string responseString = jsonStream.str();
-
- const std::string mimeType = "application/json; charset=utf-8";
-
- std::ostringstream oss;
- oss << "HTTP/1.1 200 OK\r\n"
- << "Last-Modified: " << Poco::DateTimeFormatter::format(Poco::Timestamp(), Poco::DateTimeFormat::HTTP_FORMAT) << "\r\n"
- << "User-Agent: " << WOPI_AGENT_STRING << "\r\n"
- << "Content-Length: " << responseString.size() << "\r\n"
- << "Content-Type: " << mimeType << "\r\n"
- << "\r\n"
- << responseString;
-
- socket->send(oss.str());
- socket->shutdown();
-
- return true;
+ assertRequest(request, 0);
+ _finishedToken = true;
}
- // GetFile
- else if (uriReq.getPath() == "/wopi/files/0/contents" || uriReq.getPath() == "/wopi/files/1/contents")
+ else
{
- LOG_INF("Fake wopi host request, handling GetFile: " << uriReq.getPath());
-
- if (uriReq.getPath() == "/wopi/files/0/contents")
- {
- assertRequest(request, 0);
- _finishedToken = true;
- }
- else
- {
- assertRequest(request, 1);
- _finishedHeader = true;
- }
-
- const std::string mimeType = "text/plain; charset=utf-8";
-
- std::ostringstream oss;
- oss << "HTTP/1.1 200 OK\r\n"
- << "Last-Modified: " << Poco::DateTimeFormatter::format(Poco::Timestamp(), Poco::DateTimeFormat::HTTP_FORMAT) << "\r\n"
- << "User-Agent: " << WOPI_AGENT_STRING << "\r\n"
- << "Content-Length: " << hello.size() << "\r\n"
- << "Content-Type: " << mimeType << "\r\n"
- << "\r\n"
- << hello;
-
- socket->send(oss.str());
- socket->shutdown();
-
- if (_finishedToken && _finishedHeader)
- exitTest(TestResult::Ok);
-
- return true;
+ assertRequest(request, 1);
+ _finishedHeader = true;
}
+ }
- return false;
+ bool wopiServerFinish() override
+ {
+ return _finishedToken && _finishedHeader;
}
void invokeTest() override
diff --git a/test/WopiTestServer.hpp b/test/WopiTestServer.hpp
new file mode 100644
index 000000000..63cf4c7f4
--- /dev/null
+++ b/test/WopiTestServer.hpp
@@ -0,0 +1,115 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
+/*
+ * This file is part of the LibreOffice project.
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+#include "config.h"
+
+#include "Log.hpp"
+#include "Unit.hpp"
+#include "UnitHTTP.hpp"
+#include <Poco/DateTimeFormat.h>
+#include <Poco/DateTimeFormatter.h>
+#include <Poco/JSON/Object.h>
+#include <Poco/Net/HTTPRequest.h>
+#include <Poco/URI.h>
+
+class WopiTestServer : public UnitWSD
+{
+public:
+ WopiTestServer() : UnitWSD()
+ {
+ }
+
+ virtual void assertCheckFileInfoRequest(const Poco::Net::HTTPRequest& request) = 0;
+
+ virtual void assertGetFileRequest(const Poco::Net::HTTPRequest& request) = 0;
+
+ virtual bool wopiServerFinish() = 0;
+
+protected:
+ /// Here we act as a WOPI server, so that we have a server that responds to
+ /// the wopi requests without additional expensive setup.
+ virtual bool handleHttpRequest(const Poco::Net::HTTPRequest& request, std::shared_ptr<StreamSocket>& socket) override
+ {
+ static const std::string hello("Hello, world");
+
+ Poco::URI uriReq(request.getURI());
+ LOG_INF("Fake wopi host request: " << uriReq.toString());
+
+ // CheckFileInfo
+ if (uriReq.getPath() == "/wopi/files/0" || uriReq.getPath() == "/wopi/files/1")
+ {
+ LOG_INF("Fake wopi host request, handling CheckFileInfo: " << uriReq.getPath());
+
+ assertCheckFileInfoRequest(request);
+
+ Poco::LocalDateTime now;
+ Poco::JSON::Object::Ptr fileInfo = new Poco::JSON::Object();
+ fileInfo->set("BaseFileName", "hello.txt");
+ fileInfo->set("Size", hello.size());
+ fileInfo->set("Version", "1.0");
+ fileInfo->set("OwnerId", "test");
+ fileInfo->set("UserId", "test");
+ fileInfo->set("UserFriendlyName", "test");
+ fileInfo->set("UserCanWrite", "true");
+ fileInfo->set("PostMessageOrigin", "localhost");
+ fileInfo->set("LastModifiedTime", Poco::DateTimeFormatter::format(now, Poco::DateTimeFormat::ISO8601_FORMAT));
+
+ std::ostringstream jsonStream;
+ fileInfo->stringify(jsonStream);
+ std::string responseString = jsonStream.str();
+
+ const std::string mimeType = "application/json; charset=utf-8";
+
+ std::ostringstream oss;
+ oss << "HTTP/1.1 200 OK\r\n"
+ << "Last-Modified: " << Poco::DateTimeFormatter::format(Poco::Timestamp(), Poco::DateTimeFormat::HTTP_FORMAT) << "\r\n"
+ << "User-Agent: " << WOPI_AGENT_STRING << "\r\n"
+ << "Content-Length: " << responseString.size() << "\r\n"
+ << "Content-Type: " << mimeType << "\r\n"
+ << "\r\n"
+ << responseString;
+
+ socket->send(oss.str());
+ socket->shutdown();
+
+ return true;
+ }
+ // GetFile
+ else if (uriReq.getPath() == "/wopi/files/0/contents" || uriReq.getPath() == "/wopi/files/1/contents")
+ {
+ LOG_INF("Fake wopi host request, handling GetFile: " << uriReq.getPath());
+
+ assertGetFileRequest(request);
+
+ const std::string mimeType = "text/plain; charset=utf-8";
+
+ std::ostringstream oss;
+ oss << "HTTP/1.1 200 OK\r\n"
+ << "Last-Modified: " << Poco::DateTimeFormatter::format(Poco::Timestamp(), Poco::DateTimeFormat::HTTP_FORMAT) << "\r\n"
+ << "User-Agent: " << WOPI_AGENT_STRING << "\r\n"
+ << "Content-Length: " << hello.size() << "\r\n"
+ << "Content-Type: " << mimeType << "\r\n"
+ << "\r\n"
+ << hello;
+
+ socket->send(oss.str());
+ socket->shutdown();
+
+ if (wopiServerFinish())
+ exitTest(TestResult::Ok);
+
+ return true;
+ }
+
+ return false;
+ }
+
+};
+
+/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
generated by cgit v1.2.3 (git 2.39.1) at 2024-12-05 02:08:40 +0000