Publish

1 files changed, 59 insertions, 0 deletions

diff --git a/persistence/src/tests/proxy/proxy_factory_wrapper.h b/persistence/src/tests/proxy/proxy_factory_wrapper.h
new file mode 100644
index 00000000000..10f251f2beb
--- /dev/null
+++ b/persistence/src/tests/proxy/proxy_factory_wrapper.h
@@ -0,0 +1,59 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#pragma once
+
+#include <vespa/vespalib/util/vstringfmt.h>
+#include <vespa/persistence/conformancetest/conformancetest.h>
+#include <vespa/persistence/proxy/providerstub.h>
+#include <vespa/persistence/proxy/providerproxy.h>
+#include "dummy_provider_factory.h"
+
+namespace storage {
+namespace spi {
+
+/**
+ * Generic wrapper for persistence conformance test factories. This
+ * wrapper will take any other factory and expose a factory interface
+ * that will create persistence instances that communicate with
+ * persistence instances created by the wrapped factory using the RPC
+ * persistence Proxy.
+ **/
+struct ProxyFactoryWrapper : ConformanceTest::PersistenceFactory
+{
+    typedef storage::spi::ConformanceTest::PersistenceFactory Factory;
+    typedef storage::spi::PersistenceProvider                 Provider;
+    typedef storage::spi::ProviderStub                        Server;
+    typedef storage::spi::ProviderProxy                       Client;
+    typedef document::DocumentTypeRepo                        Repo;
+
+    Factory::UP factory;
+    ProxyFactoryWrapper(Factory::UP f) : factory(std::move(f)) {}
+
+    struct Wrapper : Client {
+        DummyProviderFactory::UP provider;
+        Server::UP server;
+        Wrapper(DummyProviderFactory::UP p, Server::UP s, const Repo &repo)
+            : Client(vespalib::make_vespa_string("tcp/localhost:%u", s->getPort()), repo),
+              provider(std::move(p)),
+              server(std::move(s))
+        {}
+    };
+
+    virtual Provider::UP
+    getPersistenceImplementation(const document::DocumentTypeRepo::SP &repo,
+                                 const document::DocumenttypesConfig &typesCfg) {
+        DummyProviderFactory::UP provider(new DummyProviderFactory(factory->getPersistenceImplementation(repo,
+                                                                                                         typesCfg)));
+        Server::UP server(new Server(0, 8, *repo, *provider));
+        return Provider::UP(new Wrapper(std::move(provider), std::move(server), *repo));
+    }
+
+    virtual bool
+    supportsActiveState() const
+    {
+        return factory->supportsActiveState();
+    }
+};
+}  // namespace spi
+}  // namespace storage
+