summaryrefslogtreecommitdiff
path: root/gentoobrowse-api/unittests/mockDefs.h
blob: 7ea35351fdd1c52f9f9ceda6fabc854c45da3e3e (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
#ifndef MOCKDEFS_H
#define MOCKDEFS_H

#include <pq-mock.h>
#include <visibility.h>
#include <dryice.h>
#include <portage.h>
#include <users.h>
#include <maintenance.h>
#include <notifications.h>
#include <mocks.h>
#include <selectcommandUtil.impl.h>
#include <boost/test/unit_test.hpp>

class MockMailServerImpl : public Gentoo::MockMailServer {
	public:
		void sendEmail(const Gentoo::EmailPtr msg, const Ice::Current &) override;
		Gentoo::Emails getSentEmails(const Ice::Current &) override;

	private:
		Gentoo::Emails sentEmails;
};

class DLL_PUBLIC Service : public IceTray::DryIce, DB::PluginMock<PQ::Mock> {
	public:
		Service();
};

class DLL_PUBLIC Maintenance : public IceTray::DryIce, DB::PluginMock<PQ::Mock> {
	public:
		Maintenance();
};

class DLL_PUBLIC Mail : public IceTray::DryIce {
	public:
		Mail();
};

class DLL_PUBLIC TestClient : public IceTray::DryIceClient {
	public:
		TestClient();

		Gentoo::MaintenancePrxPtr m;
		Gentoo::PortagePrxPtr p;
		Gentoo::NotificationsPrxPtr n;
		Gentoo::MailServerPrxPtr ms;
		Gentoo::UsersPrxPtr u;
};

#define SQL_REQUIRE_EQUAL(sql, type, expected) BOOST_TEST_CONTEXT(sql) { sqlRequireEqual<type>(db, sql, expected); }
template<typename T>
void sqlRequireEqual(DB::ConnectionPtr db, const char * const sql, const T & expected) {
	int assertedRows = 0;
	db->select(sql)->forEachRow<T>([expected,&assertedRows](const auto & n) {
		BOOST_REQUIRE_EQUAL(expected, n);
		assertedRows += 1;
	});
	BOOST_REQUIRE_EQUAL(assertedRows, 1);
}

#endif