b34de7fad1
With some minor modifications. Change-Id: I5bc06bd96bbf201ccaee87170d052b4ad2a6c778 Signed-off-by: Ashod Nakashian <ashod.nakashian@collabora.co.uk>
95 lines
2.7 KiB
C++
95 lines
2.7 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
|
|
/*
|
|
* 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 "Unit.hpp"
|
|
#include "WopiTestServer.hpp"
|
|
|
|
using namespace helpers;
|
|
|
|
class UnitStorage : public WopiTestServer
|
|
{
|
|
STATE_ENUM(Phase,
|
|
Load, // load the document
|
|
Filter, // throw filter exception
|
|
Reload, // re-load the document
|
|
Done)
|
|
_phase;
|
|
|
|
public:
|
|
UnitStorage()
|
|
: WopiTestServer("UnitStorage")
|
|
, _phase(Phase::Load)
|
|
{
|
|
}
|
|
|
|
bool filterCheckDiskSpace(const std::string& /* path */, bool& newResult) override
|
|
{
|
|
// Fail the disk-space check in Filter phase.
|
|
newResult = _phase != Phase::Filter;
|
|
LOG_TST("Result: " << (newResult ? "success" : "out-of-disk-space"));
|
|
return true;
|
|
}
|
|
|
|
bool onDocumentLoaded(const std::string& message) override
|
|
{
|
|
LOG_TST("Loaded: [" << message << ']');
|
|
|
|
LOK_ASSERT_STATE(_phase, Phase::Done);
|
|
passTest("Loaded successfully");
|
|
|
|
return true;
|
|
}
|
|
|
|
bool onDocumentError(const std::string& message) override
|
|
{
|
|
// This may trigger multiple times.
|
|
LOK_ASSERT_EQUAL_MESSAGE("Expect only documentunloading errors",
|
|
std::string("error: cmd=internal kind=diskfull"), message);
|
|
|
|
LOK_ASSERT_STATE(_phase, Phase::Filter);
|
|
|
|
return true;
|
|
}
|
|
|
|
// When we fail to load, we must destroy the DocBroker.
|
|
void onDocBrokerDestroy(const std::string&) override
|
|
{
|
|
if (_phase == Phase::Filter)
|
|
{
|
|
TRANSITION_STATE(_phase, Phase::Reload);
|
|
}
|
|
}
|
|
|
|
void invokeWSDTest() override
|
|
{
|
|
LOG_TRC("invokeWSDTest: " << toString(_phase));
|
|
switch (_phase)
|
|
{
|
|
case Phase::Load:
|
|
TRANSITION_STATE(_phase, Phase::Filter);
|
|
initWebsocket("/wopi/files/0?access_token=anything");
|
|
WSD_CMD("load url=" + getWopiSrc());
|
|
break;
|
|
case Phase::Filter:
|
|
break;
|
|
case Phase::Reload:
|
|
LOG_TST("Reloading the document");
|
|
TRANSITION_STATE(_phase, Phase::Done);
|
|
initWebsocket("/wopi/files/0?access_token=anything");
|
|
WSD_CMD("load url=" + getWopiSrc());
|
|
break;
|
|
case Phase::Done:
|
|
break;
|
|
}
|
|
}
|
|
};
|
|
|
|
UnitBase* unit_create_wsd(void) { return new UnitStorage(); }
|
|
|
|
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|