summaryrefslogtreecommitdiff
path: root/netfs/unittests/mockDaemon.cpp
blob: 0cb31728690b6631053a1dcf576d11783375b310 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#include "mockDaemon.h"
#include <buffer.h>
#include <definedDirs.h>

const std::filesystem::path MockDaemonHost::TestExportRoot(binDir / UniqueExport::get(getpid()));

MockDaemon::MockDaemon(std::string ep) : NetFSDaemon(), testEndpoint(std::move(ep)) { }

NetFS::Daemon::ConfigurationPtr
MockDaemon::ReadConfiguration(const std::filesystem::path & path) const
{
	auto c = NetFSDaemon::ReadConfiguration(path);
	for (const auto & e : c->Exports) {
		e.second->RootPath = MockDaemonHost::TestExportRoot.string();
	}
	c->Hosts.insert({"unittest", std::make_shared<NetFS::Daemon::Host>(testEndpoint)});
	return c;
}

MockDaemonHost::MockDaemonHost(std::string ep, Ice::StringSeq ps) :
	testEndpoint(std::move(ep)), params(std::move(ps)), daemon(nullptr)
{
	std::filesystem::remove_all(TestExportRoot);
	std::filesystem::create_directories(TestExportRoot);
	restart();
}

MockDaemonHost::~MockDaemonHost()
{
	stop();
	std::filesystem::remove_all(TestExportRoot);
}

void
MockDaemonHost::stop()
{
	if (daemon) {
		daemon->stop();
		delete daemon;
	}
	if (ic) {
		ic->destroy();
	}
}

void
MockDaemonHost::start()
{
	Ice::InitializationData id;
	id.properties = Ice::createProperties();
	id.properties->parseCommandLineOptions("", params);
	ic = Ice::initialize(id);
	ic->getProperties()->setProperty("NetFSD.HostNameOverride", "unittest");
	daemon = new MockDaemon(testEndpoint);
	daemon->start("NetFSDaemonAdapter", ic, {});
}

void
MockDaemonHost::restart()
{
	stop();
	start();
}