mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 23:38:50 +03:00
fbdb46f5cb
Reviewed By: chadaustin Differential Revision: D17872966 fbshipit-source-id: cd60a364a2146f0dadbeca693b1d4a5d7c97ff63
55 lines
1.6 KiB
C++
55 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This software may be used and distributed according to the terms of the
|
|
* GNU General Public License version 2.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <unordered_map>
|
|
#include "eden/fs/store/IObjectStore.h"
|
|
|
|
#include "eden/fs/model/Blob.h"
|
|
#include "eden/fs/model/Hash.h"
|
|
#include "eden/fs/model/Tree.h"
|
|
#include "eden/fs/store/BlobMetadata.h"
|
|
|
|
namespace facebook {
|
|
namespace eden {
|
|
|
|
/**
|
|
* Fake implementation of IObjectStore that allows the data to be injected
|
|
* directly. This is designed to be used for unit tests.
|
|
*/
|
|
class FakeObjectStore : public IObjectStore {
|
|
public:
|
|
FakeObjectStore();
|
|
~FakeObjectStore() override;
|
|
|
|
void addTree(Tree&& tree);
|
|
void addBlob(Blob&& blob);
|
|
void setTreeForCommit(const Hash& commitID, Tree&& tree);
|
|
|
|
folly::Future<std::shared_ptr<const Tree>> getTree(
|
|
const Hash& id) const override;
|
|
folly::Future<std::shared_ptr<const Blob>> getBlob(
|
|
const Hash& id) const override;
|
|
folly::Future<std::shared_ptr<const Tree>> getTreeForCommit(
|
|
const Hash& commitID) const override;
|
|
folly::Future<BlobMetadata> getBlobMetadata(const Hash& id) const override;
|
|
folly::Future<folly::Unit> prefetchBlobs(
|
|
const std::vector<Hash>& ids) const override;
|
|
|
|
size_t getAccessCount(const Hash& hash) const;
|
|
|
|
private:
|
|
std::unordered_map<Hash, Tree> trees_;
|
|
std::unordered_map<Hash, Blob> blobs_;
|
|
std::unordered_map<Hash, Tree> commits_;
|
|
std::unordered_map<Hash, BlobMetadata> blobMetadata_;
|
|
mutable std::unordered_map<Hash, size_t> accessCounts_;
|
|
};
|
|
} // namespace eden
|
|
} // namespace facebook
|