Publish

1 files changed, 172 insertions, 0 deletions

diff --git a/persistence/src/tests/proxy/mockprovider.h b/persistence/src/tests/proxy/mockprovider.h
new file mode 100644
index 00000000000..c2fd844a010
--- /dev/null
+++ b/persistence/src/tests/proxy/mockprovider.h
@@ -0,0 +1,172 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#pragma once
+
+#include <vespa/persistence/spi/persistenceprovider.h>
+
+namespace storage {
+namespace spi {
+
+struct MockProvider : PersistenceProvider {
+    enum Function { NONE, INITIALIZE, GET_PARTITION_STATES, LIST_BUCKETS,
+                    SET_CLUSTER_STATE,
+                    SET_ACTIVE_STATE, GET_BUCKET_INFO, PUT, REMOVE_BY_ID,
+                    REMOVE_IF_FOUND, REPLACE_WITH_REMOVE, UPDATE, FLUSH, GET,
+                    CREATE_ITERATOR, ITERATE, DESTROY_ITERATOR, CREATE_BUCKET,
+                    DELETE_BUCKET, GET_MODIFIED_BUCKETS, SPLIT, JOIN, MOVE, MAINTAIN,
+                    REMOVE_ENTRY };
+
+    mutable Function last_called;
+
+    MockProvider() : last_called(NONE) {}
+
+    virtual Result initialize() {
+        last_called = INITIALIZE;
+        return Result();
+    }
+
+    virtual PartitionStateListResult getPartitionStates() const {
+        last_called = GET_PARTITION_STATES;
+        return PartitionStateListResult(PartitionStateList(1u));
+    }
+
+    virtual BucketIdListResult listBuckets(PartitionId id) const {
+        last_called = LIST_BUCKETS;
+        BucketIdListResult::List result;
+        result.push_back(document::BucketId(id));
+        return BucketIdListResult(result);
+    }
+
+    virtual Result setClusterState(const ClusterState &) {
+        last_called = SET_CLUSTER_STATE;
+        return Result();
+    }
+
+    virtual Result setActiveState(const Bucket &,
+                                  BucketInfo::ActiveState) {
+        last_called = SET_ACTIVE_STATE;
+        return Result();
+    }
+
+    virtual BucketInfoResult getBucketInfo(const Bucket &bucket) const {
+        last_called = GET_BUCKET_INFO;
+        return BucketInfoResult(BucketInfo(BucketChecksum(1), 2, 3,
+                                           bucket.getBucketId().getRawId(),
+                                           bucket.getPartition(),
+                                           BucketInfo::READY,
+                                           BucketInfo::ACTIVE));
+    }
+
+    virtual Result put(const Bucket &, Timestamp, const Document::SP&, Context&) {
+        last_called = PUT;
+        return Result();
+    }
+
+    virtual RemoveResult remove(const Bucket &, Timestamp,
+                                const DocumentId &, Context&) {
+        last_called = REMOVE_BY_ID;
+        return RemoveResult(true);
+    }
+
+    virtual RemoveResult removeIfFound(const Bucket &, Timestamp,
+                                       const DocumentId &, Context&) {
+        last_called = REMOVE_IF_FOUND;
+        return RemoveResult(true);
+    }
+
+    virtual RemoveResult replaceWithRemove(const Bucket &, Timestamp,
+                                           const DocumentId &, Context&) {
+        last_called = REPLACE_WITH_REMOVE;
+        return RemoveResult(true);
+    }
+
+    virtual UpdateResult update(const Bucket &, Timestamp timestamp,
+                                const DocumentUpdate::SP&, Context&) {
+        last_called = UPDATE;
+        return UpdateResult(Timestamp(timestamp - 10));
+    }
+
+    virtual Result flush(const Bucket&, Context&) {
+        last_called = FLUSH;
+        return Result();
+    }
+
+    virtual GetResult get(const Bucket &, const document::FieldSet&,
+                          const DocumentId&, Context&) const {
+        last_called = GET;
+        return GetResult(Document::UP(new Document),
+                         Timestamp(6u));
+    }
+
+    virtual CreateIteratorResult createIterator(const Bucket& bucket,
+                                                const document::FieldSet&,
+                                                const Selection&,
+                                                IncludedVersions,
+                                                Context&)
+    {
+        last_called = CREATE_ITERATOR;
+        return CreateIteratorResult(IteratorId(bucket.getPartition()));
+    }
+
+    virtual IterateResult iterate(IteratorId, uint64_t, Context&) const {
+        last_called = ITERATE;
+        IterateResult::List result;
+        result.push_back(DocEntry::LP(new DocEntry(Timestamp(1), 0)));
+        return IterateResult(result, true);
+    }
+
+    virtual Result destroyIterator(IteratorId, Context&) {
+        last_called = DESTROY_ITERATOR;
+        return Result();
+    }
+
+    virtual Result createBucket(const Bucket&, Context&) {
+        last_called = CREATE_BUCKET;
+        return Result();
+    }
+    virtual Result deleteBucket(const Bucket&, Context&) {
+        last_called = DELETE_BUCKET;
+        return Result();
+    }
+
+    virtual BucketIdListResult getModifiedBuckets() const {
+        last_called = GET_MODIFIED_BUCKETS;
+        BucketIdListResult::List list;
+        list.push_back(document::BucketId(2));
+        list.push_back(document::BucketId(3));
+        return BucketIdListResult(list);
+    }
+
+    virtual Result split(const Bucket &, const Bucket &, const Bucket &,
+                         Context&)
+    {
+        last_called = SPLIT;
+        return Result();
+    }
+
+    virtual Result join(const Bucket &, const Bucket &, const Bucket &,
+                        Context&)
+    {
+        last_called = JOIN;
+        return Result();
+    }
+
+    virtual Result move(const Bucket &, PartitionId, Context&) {
+        last_called = MOVE;
+        return Result();
+    }
+
+
+    virtual Result maintain(const Bucket &, MaintenanceLevel) {
+        last_called = MAINTAIN;
+        return Result();
+    }
+
+    virtual Result removeEntry(const Bucket &, Timestamp, Context&) {
+        last_called = REMOVE_ENTRY;
+        return Result();
+    }
+};
+
+}  // namespace spi
+}  // namespace storage
+