1 files changed, 121 insertions, 0 deletions
diff --git a/src/lib/dhcpsrv/testutils/test_config_backend.h b/src/lib/dhcpsrv/testutils/test_config_backend.h
new file mode 100644
index 0000000..d78f97f
--- /dev/null
+++ b/src/lib/dhcpsrv/testutils/test_config_backend.h
@@ -0,0 +1,121 @@
+// Copyright (C) 2018-2019 Internet Systems Consortium, Inc. ("ISC")
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+#ifndef TEST_CONFIG_BACKEND_H
+#define TEST_CONFIG_BACKEND_H
+
+#include <config.h>
+
+#include <database/database_connection.h>
+#include <dhcpsrv/config_backend_dhcp4_mgr.h>
+#include <dhcpsrv/config_backend_dhcp6_mgr.h>
+#include <boost/shared_ptr.hpp>
+#include <boost/lexical_cast.hpp>
+
+namespace isc {
+namespace dhcp {
+namespace test {
+
+/// @brief Type of pointers to stamped elements.
+typedef boost::shared_ptr<data::StampedElement> StampedElementPtr;
+
+/// @brief Base class for implementing fake backends
+template<typename ConfigBackendType>
+class TestConfigBackend : public ConfigBackendType {
+public:
+    /// @brief Constructor
+    ///
+    /// @param params database connection parameters
+    /// @throw BadValue if parameters do not include "type"
+    TestConfigBackend(const db::DatabaseConnection::ParameterMap& params)
+     : connection_(params) {
+        try {
+            db_type_ = connection_.getParameter("type");
+        } catch (...) {
+            isc_throw(BadValue, "Backend parameters must include \"type\"");
+        }
+
+        try {
+            host_ = connection_.getParameter("host");
+        } catch (...) {
+            host_ = "default_host";
+        }
+
+        try {
+            port_ = boost::lexical_cast<uint16_t>(connection_.getParameter("host"));
+        } catch (...) {
+            port_ = 0;
+        }
+    }
+
+    /// @brief virtual Destructor.
+    virtual ~TestConfigBackend(){};
+
+    /// @brief Returns backend type.
+    ///
+    /// @return string db_type name
+    virtual std::string getType() const {
+        return (db_type_);
+    }
+
+    /// @brief Returns backend host.
+    ///
+    /// @return string host
+    virtual std::string getHost() const {
+        return (host_);
+    }
+
+    /// @brief Returns backend port.
+    ///
+    /// @return uint16_t port
+    virtual uint16_t getPort() const {
+        return (port_);
+    }
+
+    /// @brief Returns server tag to be associated with the stored configuration.
+    ///
+    /// @param server_selector Server selector.
+    std::string getServerTag(const db::ServerSelector& server_selector) const {
+        if (server_selector.getType() == db::ServerSelector::Type::ALL) {
+            return ("all");
+        }
+        // Return first tag found.
+        auto tags = server_selector.getTags();
+        if (!tags.empty()) {
+            return (tags.begin()->get());
+        }
+        return ("");
+    }
+
+    /// @brief Merge server tags for a stamped element and a server selector.
+    ///
+    /// @param elem Stamped element to update.
+    /// @param server_selector Server selector.
+    void mergeServerTags(const StampedElementPtr& elem,
+                         const db::ServerSelector& server_selector) const {
+        auto tags = server_selector.getTags();
+        for (auto tag : tags) {
+            elem->setServerTag(tag.get());
+        }
+    }
+
+    /// @brief Fake database connection
+    db::DatabaseConnection connection_;
+
+    /// @brief  Back end type
+    std::string db_type_;
+
+    /// @brief  Back end host
+    std::string host_;
+
+    /// @brief  Back end port
+    uint16_t port_;
+};
+
+} // namespace test
+} // namespace dhcp
+} // namespace isc
+
+#endif // TEST_CONFIG_BACKEND_H