blob: 8f60ebe6676232cc6efe36ef3cc25cdacaed9b9e [file] [log] [blame]
Kun Yi68c81142018-12-18 11:17:14 -08001#pragma once
2
3#include "binarystore.hpp"
Patrick Venture15f0f942020-07-09 09:38:18 -07004#include "binarystore_interface.hpp"
Kun Yi2765b642019-01-16 11:11:24 -08005#include "sys_file.hpp"
Kun Yi68c81142018-12-18 11:17:14 -08006
Patrick Venture15f0f942020-07-09 09:38:18 -07007#include <memory>
8#include <string>
9#include <vector>
10
Kun Yi68c81142018-12-18 11:17:14 -080011#include <gmock/gmock.h>
12
Kun Yi64dc05c2018-12-19 13:19:03 -080013using ::testing::Invoke;
14
Kun Yi68c81142018-12-18 11:17:14 -080015namespace binstore
16{
17
18class MockBinaryStore : public BinaryStoreInterface
19{
20 public:
Kun Yi2765b642019-01-16 11:11:24 -080021 MockBinaryStore(const std::string& baseBlobId,
Patrick Venturee496b2b2020-07-09 13:49:05 -070022 std::unique_ptr<SysFile> file) :
23 real_store_(baseBlobId, std::move(file))
Kun Yi64dc05c2018-12-19 13:19:03 -080024 {
25 // Implemented calls in BinaryStore will be directed to the real object.
26 ON_CALL(*this, getBaseBlobId)
27 .WillByDefault(Invoke(&real_store_, &BinaryStore::getBaseBlobId));
28 ON_CALL(*this, getBlobIds)
29 .WillByDefault(Invoke(&real_store_, &BinaryStore::getBlobIds));
Kun Yi6baa7132019-01-08 21:21:02 -080030 ON_CALL(*this, openOrCreateBlob)
31 .WillByDefault(
32 Invoke(&real_store_, &BinaryStore::openOrCreateBlob));
33 ON_CALL(*this, close)
34 .WillByDefault(Invoke(&real_store_, &BinaryStore::close));
Kun Yi6967b772019-02-22 13:48:12 -080035 ON_CALL(*this, read)
36 .WillByDefault(Invoke(&real_store_, &BinaryStore::read));
37 ON_CALL(*this, write)
38 .WillByDefault(Invoke(&real_store_, &BinaryStore::write));
Kun Yid297c9f2019-01-09 13:52:30 -080039 ON_CALL(*this, commit)
40 .WillByDefault(Invoke(&real_store_, &BinaryStore::commit));
Kun Yi1a25e0d2020-05-11 12:28:53 -070041 ON_CALL(*this, stat)
42 .WillByDefault(Invoke(&real_store_, &BinaryStore::stat));
Kun Yi64dc05c2018-12-19 13:19:03 -080043 }
Kun Yi68c81142018-12-18 11:17:14 -080044 MOCK_CONST_METHOD0(getBaseBlobId, std::string());
Kun Yi68c81142018-12-18 11:17:14 -080045 MOCK_CONST_METHOD0(getBlobIds, std::vector<std::string>());
Kun Yi38146a02018-12-18 21:54:26 -080046 MOCK_METHOD2(openOrCreateBlob, bool(const std::string&, uint16_t));
Kun Yic0adbc32018-12-18 22:35:29 -080047 MOCK_METHOD1(deleteBlob, bool(const std::string&));
Kun Yi68c81142018-12-18 11:17:14 -080048 MOCK_METHOD2(read, std::vector<uint8_t>(uint32_t, uint32_t));
49 MOCK_METHOD2(write, bool(uint32_t, const std::vector<uint8_t>&));
50 MOCK_METHOD0(commit, bool());
51 MOCK_METHOD0(close, bool());
Kun Yi1a25e0d2020-05-11 12:28:53 -070052 MOCK_METHOD1(stat, bool(blobs::BlobMeta* meta));
Kun Yi64dc05c2018-12-19 13:19:03 -080053
54 private:
55 BinaryStore real_store_;
Kun Yi68c81142018-12-18 11:17:14 -080056};
57
58} // namespace binstore