SysrepoSubscription: Allow subscribing to startup
Change-Id: I06459720b9c3dedcd8f074d9cb218143c695f13a
diff --git a/tests/mock/sysrepo_subscription.cpp b/tests/mock/sysrepo_subscription.cpp
index 5c5fd27..8faffa5 100644
--- a/tests/mock/sysrepo_subscription.cpp
+++ b/tests/mock/sysrepo_subscription.cpp
@@ -55,10 +55,10 @@
DataSupplier::~DataSupplier() = default;
-SysrepoSubscription::SysrepoSubscription(const std::string& moduleName, Recorder* rec)
+SysrepoSubscription::SysrepoSubscription(const std::string& moduleName, Recorder* rec, sr_datastore_t ds)
: m_connection(std::make_shared<sysrepo::Connection>())
{
- m_session = std::make_shared<sysrepo::Session>(m_connection);
+ m_session = std::make_shared<sysrepo::Session>(m_connection, ds);
m_subscription = std::make_shared<sysrepo::Subscribe>(m_session);
sysrepo::ModuleChangeCb cb;
if (rec) {
diff --git a/tests/mock/sysrepo_subscription.hpp b/tests/mock/sysrepo_subscription.hpp
index c3f1f69..fc2e499 100644
--- a/tests/mock/sysrepo_subscription.hpp
+++ b/tests/mock/sysrepo_subscription.hpp
@@ -10,6 +10,7 @@
#include <memory>
#include <optional>
+#include <sysrepo-cpp/Session.hpp>
#include "datastore_access.hpp"
namespace sysrepo {
@@ -35,7 +36,7 @@
class SysrepoSubscription {
public:
- SysrepoSubscription(const std::string& moduleName, Recorder* rec = nullptr);
+ SysrepoSubscription(const std::string& moduleName, Recorder* rec = nullptr, sr_datastore_t ds = SR_DS_RUNNING);
private:
std::shared_ptr<sysrepo::Connection> m_connection;