summaryrefslogtreecommitdiff
path: root/test/UnitWOPIHttpHeaders.cpp
blob: acf2d747146fd6de82614ccbc375e3fa6cd31c50 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/* -*- 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 "WopiTestServer.hpp"
#include <Log.hpp>
#include <Unit.hpp>
#include <UnitHTTP.hpp>
#include <helpers.hpp>

#include <Poco/Net/HTTPRequest.h>

class UnitWopiHttpHeaders : public WopiTestServer
{
    enum class Phase
    {
        Load,
        Polling
    } _phase;

protected:
    void assertCheckFileInfoRequest(const Poco::Net::HTTPRequest& request) override
    {
        assertHeaders(request);
    }

    void assertGetFileRequest(const Poco::Net::HTTPRequest& request) override
    {
        assertHeaders(request);
        exitTest(TestResult::Ok); //TODO: Remove when we add put/rename cases.
    }

    void assertPutFileRequest(const Poco::Net::HTTPRequest& request) override
    {
        assertHeaders(request);
        exitTest(TestResult::Ok);
    }

    void assertPutRelativeFileRequest(const Poco::Net::HTTPRequest& request) override
    {
        assertHeaders(request);
        exitTest(TestResult::Ok);
    }

    void assertRenameFileRequest(const Poco::Net::HTTPRequest& request) override
    {
        assertHeaders(request);
        exitTest(TestResult::Ok);
    }

    void assertHeaders(const Poco::Net::HTTPRequest& request) const
    {
        static const std::map<std::string, std::string> Headers{
            { "Authorization", "Bearer xyz123abc456vwc789z" },
            { "X-Requested-With", "XMLHttpRequest" },
        };

        for (const auto& pair : Headers)
        {
            LOK_ASSERT_MESSAGE("Request must have [" + pair.first + "]", request.has(pair.first));
            LOK_ASSERT_EQUAL(pair.second, request[pair.first]);
        }
    }

public:
    UnitWopiHttpHeaders()
        : _phase(Phase::Load)
    {
    }

    void invokeTest() override
    {
        constexpr char testName[] = "UnitWopiHttpHeaders";

        switch (_phase)
        {
            case Phase::Load:
            {
                // Technically, having an empty line in the header
                // is invalid (it signifies the end of headers), but
                // this is to illustrate that we are able to overcome
                // such issues and generate valid headers.
                const std::string params
                    = "access_header=Authorization%3A%2520Bearer%"
                      "2520xyz123abc456vwc789z%250D%250A%250D%250AX-Requested-With%"
                      "3A%2520XMLHttpRequest&reuse_cookies=language%3Den-us%3AK%3DGS1&permission="
                      "edit";

                initWebsocket("/wopi/files/0?" + params);

                helpers::sendTextFrame(*getWs()->getLOOLWebSocket(), "load url=" + getWopiSrc(),
                                       testName);
                SocketPoll::wakeupWorld();

                _phase = Phase::Polling;
                break;
            }
            case Phase::Polling:
            {
                // Just wait for the results.
                break;
            }
        }
    }
};

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

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