sapling/eden/fs/testharness/FakeObjectStore.h
Chad Austin bb1cccac89 introduce a variable-width RootId type that identifies the root of an EdenFS checkout's contents
Summary:
Backing stores differentiate between individual tree objects and the
root of a checkout. For example, Git and Mercurial roots are commit
hashes. Allow EdenFS to track variable-width roots to better support
arbitrary backing stores.

Reviewed By: genevievehelsel

Differential Revision: D28619584

fbshipit-source-id: d94f1ecd21a0c416c1b4933341c70deabf386496
2021-06-07 17:25:31 -07:00

65 lines
2.0 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/model/Blob.h"
#include "eden/fs/model/Hash.h"
#include "eden/fs/model/RootId.h"
#include "eden/fs/model/Tree.h"
#include "eden/fs/store/BlobMetadata.h"
#include "eden/fs/store/IObjectStore.h"
#include "eden/fs/store/ImportPriority.h"
#include "eden/fs/store/ObjectFetchContext.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 final : public IObjectStore {
public:
FakeObjectStore();
~FakeObjectStore() override;
void addTree(Tree&& tree);
void addBlob(Blob&& blob);
void setTreeForCommit(const RootId& commitID, Tree&& tree);
folly::Future<std::shared_ptr<const Tree>> getRootTree(
const RootId& commitID,
ObjectFetchContext& context =
ObjectFetchContext::getNullContext()) const override;
folly::Future<std::shared_ptr<const Tree>> getTree(
const Hash& id,
ObjectFetchContext& context =
ObjectFetchContext::getNullContext()) const override;
folly::Future<std::shared_ptr<const Blob>> getBlob(
const Hash& id,
ObjectFetchContext& context =
ObjectFetchContext::getNullContext()) const override;
folly::Future<folly::Unit> prefetchBlobs(
const std::vector<Hash>& ids,
ObjectFetchContext& context =
ObjectFetchContext::getNullContext()) const override;
size_t getAccessCount(const Hash& hash) const;
private:
std::unordered_map<RootId, Tree> commits_;
std::unordered_map<Hash, Tree> trees_;
std::unordered_map<Hash, Blob> blobs_;
mutable std::unordered_map<RootId, size_t> commitAccessCounts_;
mutable std::unordered_map<Hash, size_t> accessCounts_;
};
} // namespace eden
} // namespace facebook