summaryrefslogtreecommitdiff
path: root/test/fixtures/mock_file_source.hpp
blob: 245e0da0ebe38e2388a61e6084dfb213cb61a38a (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
#ifndef TEST_RESOURCES_MOCK_FILE_SOURCE
#define TEST_RESOURCES_MOCK_FILE_SOURCE

#include <mbgl/storage/file_source.hpp>
#include <mbgl/util/timer.hpp>

#include <string>
#include <unordered_map>

namespace mbgl {

// The MockFileSource is a FileSource that can simulate different
// types of failures and it will work completely offline.
class MockFileSource : public FileSource {
public:
    // Success:
    //     Will reply to every request correctly with valid data.
    //
    // RequestFail:
    //     Will reply with an error to requests that contains
    //     the "match" string on the URL.
    //
    // RequestWithCorruptedData:
    //     Will answer every request successfully but will return
    //     corrupt data on the requests that contains the "match"
    //     string on the URL.
    enum Type {
        Success,
        RequestFail,
        RequestWithCorruptedData
    };

    MockFileSource(Type, const std::string& match);
    ~MockFileSource() override;

    // Function that gets called when a matching resource is enqueued.
    std::function<void (void)> requestEnqueuedCallback;

    // FileSource implementation.
    std::unique_ptr<FileRequest> request(const Resource&, Callback) override;

private:
    void respond(Resource, Callback) const;

    friend class MockFileRequest;

    Type type;
    std::string match;
    std::unordered_map<FileRequest*, std::pair<Resource, Callback>> pending;
    util::Timer timer;
};

}

#endif