/* -*- 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 <iostream>

#include "Exceptions.hpp"
#include "Log.hpp"
#include "Unit.hpp"
#include "UnitHTTP.hpp"
#include "helpers.hpp"

using namespace helpers;

class UnitStorage : public UnitWSD
{
    enum {
        PHASE_LOAD,             // load the document
        PHASE_FILTER,           // throw filter exception
        PHASE_RE_LOAD,          // re-load the document
    } _phase;
    std::unique_ptr<UnitWebSocket> _ws;
public:
    UnitStorage() :
        _phase(PHASE_LOAD)
    {
    }

    virtual bool filterLoad(const std::string &/* sessionId */,
                            const std::string &/* jailId */,
                            bool &/* result */)
    {
        if (_phase == PHASE_FILTER)
        {
            _phase = PHASE_RE_LOAD;
            LOG_INF("Throwing low disk space exception.");
            throw StorageSpaceLowException("test: low disk space");
        }
        return false;
    }

    void loadDocument()
    {
        std::string docPath;
        std::string docURL;
        getDocumentPathAndURL("empty.odt", docPath, docURL);
        _ws = std::unique_ptr<UnitWebSocket>(new UnitWebSocket(docURL));
        assert(_ws.get());
    }

    virtual void invokeTest()
    {
        switch (_phase)
        {
        case PHASE_LOAD:
            _phase = PHASE_FILTER;
            loadDocument();
            break;
        case PHASE_FILTER:
            break;
        case PHASE_RE_LOAD:
            loadDocument();
            _ws.reset();
            exitTest(TEST_OK);
            break;
        }
    }
};

UnitBase *unit_create_wsd(void)
{
    return new UnitStorage();
}

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */