summaryrefslogtreecommitdiff
path: root/test/fixtures/stub_file_source.hpp
blob: 344dff6073e3b0ca7482982506fc2aa4ac3b14ac (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
#ifndef TEST_RESOURCES_STUB_FILE_SOURCE
#define TEST_RESOURCES_STUB_FILE_SOURCE

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

#include <unordered_map>

namespace mbgl {

class StubFileSource : public FileSource {
public:
    StubFileSource();
    ~StubFileSource() override;

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

    // You can set the response callback on a global level by assigning this callback:
    std::function<optional<Response> (const Resource&)> response = [this] (const Resource& resource) {
        return defaultResponse(resource);
    };

    // Or set per-kind responses by setting these callbacks:
    std::function<optional<Response> (const Resource&)> styleResponse;
    std::function<optional<Response> (const Resource&)> sourceResponse;
    std::function<optional<Response> (const Resource&)> tileResponse;
    std::function<optional<Response> (const Resource&)> glyphsResponse;
    std::function<optional<Response> (const Resource&)> spriteJSONResponse;
    std::function<optional<Response> (const Resource&)> spriteImageResponse;

private:
    friend class StubFileRequest;

    // The default behavior is to throw if no per-kind callback has been set.
    optional<Response> defaultResponse(const Resource&);

    std::unordered_map<FileRequest*, std::pair<optional<Response>, Callback>> pending;
    util::Timer timer;
};

}

#endif