133 lines
3.6 KiB
C++
133 lines
3.6 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
|
|
/*
|
|
* Copyright the Collabora Online contributors.
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* 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 <iostream>
|
|
|
|
#include <Common.hpp>
|
|
#include <Protocol.hpp>
|
|
#include <Unit.hpp>
|
|
#include <Util.hpp>
|
|
#include <FileUtil.hpp>
|
|
#include <helpers.hpp>
|
|
|
|
#include <Poco/Net/HTTPServerRequest.h>
|
|
#include <Poco/Net/HTMLForm.h>
|
|
#include <Poco/Net/StringPartSource.h>
|
|
#include <Poco/Util/LayeredConfiguration.h>
|
|
|
|
// Inside the WSD process
|
|
class UnitConvert : public UnitWSD
|
|
{
|
|
bool _workerStarted;
|
|
std::thread _worker;
|
|
|
|
public:
|
|
UnitConvert()
|
|
: UnitWSD("UnitConvert")
|
|
, _workerStarted(false)
|
|
{
|
|
setHasKitHooks();
|
|
setTimeout(std::chrono::hours(1));
|
|
}
|
|
|
|
~UnitConvert()
|
|
{
|
|
LOG_INF("Joining test worker thread\n");
|
|
_worker.join();
|
|
}
|
|
|
|
void configure(Poco::Util::LayeredConfiguration& config) override
|
|
{
|
|
UnitWSD::configure(config);
|
|
|
|
config.setBool("ssl.enable", true);
|
|
config.setInt("per_document.limit_load_secs", 30);
|
|
config.setBool("storage.filesystem[@allow]", false);
|
|
}
|
|
|
|
void sendConvertTo(std::unique_ptr<Poco::Net::HTTPClientSession>& session, const std::string& filename)
|
|
{
|
|
Poco::Net::HTTPRequest request(Poco::Net::HTTPRequest::HTTP_POST, "/cool/convert-to/pdf");
|
|
Poco::Net::HTMLForm form;
|
|
form.setEncoding(Poco::Net::HTMLForm::ENCODING_MULTIPART);
|
|
form.set("format", "txt");
|
|
form.addPart("data", new Poco::Net::StringPartSource("Hello World Content", "text/plain", filename));
|
|
form.prepareSubmit(request);
|
|
form.write(session->sendRequest(request));
|
|
}
|
|
|
|
bool checkConvertTo(std::unique_ptr<Poco::Net::HTTPClientSession>& session)
|
|
{
|
|
Poco::Net::HTTPResponse response;
|
|
try {
|
|
session->receiveResponse(response);
|
|
} catch (...) {
|
|
return false;
|
|
}
|
|
|
|
return response.getStatus() == Poco::Net::HTTPResponse::HTTPStatus::HTTP_OK;
|
|
}
|
|
|
|
void invokeWSDTest() override
|
|
{
|
|
if (_workerStarted)
|
|
return;
|
|
_workerStarted = true;
|
|
std::cerr << "Starting thread ...\n";
|
|
_worker = std::thread([this]{
|
|
std::cerr << "Now started thread ...\n";
|
|
std::unique_ptr<Poco::Net::HTTPClientSession> session(helpers::createSession(Poco::URI(helpers::getTestServerURI())));
|
|
session->setTimeout(Poco::Timespan(30, 0)); // 30 seconds.
|
|
|
|
sendConvertTo(session, "foo.txt");
|
|
if(!checkConvertTo(session))
|
|
{
|
|
exitTest(TestResult::Failed);
|
|
return;
|
|
}
|
|
|
|
sendConvertTo(session, "test___á.txt");
|
|
if(!checkConvertTo(session))
|
|
{
|
|
exitTest(TestResult::Failed);
|
|
return;
|
|
}
|
|
|
|
exitTest(TestResult::Ok);
|
|
});
|
|
}
|
|
};
|
|
|
|
// Inside the forkit & kit processes
|
|
class UnitKitConvert : public UnitKit
|
|
{
|
|
public:
|
|
UnitKitConvert()
|
|
: UnitKit("UnitKitConvert")
|
|
{
|
|
setTimeout(std::chrono::hours(1));
|
|
}
|
|
};
|
|
|
|
UnitBase *unit_create_wsd(void)
|
|
{
|
|
return new UnitConvert();
|
|
}
|
|
|
|
UnitBase *unit_create_kit(void)
|
|
{
|
|
return new UnitKitConvert();
|
|
}
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|