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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// 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/.
#include <config.h>
#include "command_options_helper.h"
#include <dhcp/iface_mgr.h>
#include <exceptions/exceptions.h>
#include "receiver.h"
#include <gtest/gtest.h>
using namespace isc;
using namespace isc::dhcp;
using namespace isc::perfdhcp;
/// \brief FakeReceiverPerfSocket class that mocks PerfSocket.
///
/// It stubs send and receive operations and collects statistics.
class FakeReceiverPerfSocket: public BasePerfSocket {
public:
/// \brief Default constructor for FakeReceiverPerfSocket.
FakeReceiverPerfSocket() :
iface_(boost::make_shared<Iface>("fake", 0)),
sent_cnt_(0),
recv_cnt_(0) {};
IfacePtr iface_; ///< Local fake interface.
int sent_cnt_; ///< Counter of sent packets
int recv_cnt_; ///< Counter of received packets.
/// \brief Simulate receiving DHCPv4 packet.
virtual dhcp::Pkt4Ptr receive4(uint32_t timeout_sec, uint32_t timeout_usec) override {
(void)timeout_sec; // silence compile 'unused parameter' warning;
(void)timeout_usec; // silence compile 'unused parameter' warning;
recv_cnt_++;
// slow down receiving as receiver calls it in a loop thousands of time
// if null is returned
usleep(50);
return(dhcp::Pkt4Ptr());
};
/// \brief Simulate receiving DHCPv6 packet.
virtual dhcp::Pkt6Ptr receive6(uint32_t timeout_sec, uint32_t timeout_usec) override {
(void)timeout_sec; // silence compile 'unused parameter' warning;
(void)timeout_usec; // silence compile 'unused parameter' warning;
recv_cnt_++;
return(dhcp::Pkt6Ptr());
};
/// \brief Simulate sending DHCPv4 packet.
virtual bool send(const dhcp::Pkt4Ptr& pkt) override {
sent_cnt_++;
pkt->updateTimestamp();
return true;
};
/// \brief Simulate sending DHCPv6 packet.
virtual bool send(const dhcp::Pkt6Ptr& pkt) override {
sent_cnt_++;
pkt->updateTimestamp();
return true;
};
/// \brief Override getting interface.
virtual IfacePtr getIface() override { return iface_; }
void reset() {
sent_cnt_ = 0;
recv_cnt_ = 0;
}
};
TEST(Receiver, singleThreaded) {
CommandOptions opt;
CommandOptionsHelper::process(opt, "perfdhcp -g single -l 127.0.0.1 all");
ASSERT_TRUE(opt.isSingleThreaded());
FakeReceiverPerfSocket socket;
Receiver receiver(socket, opt.isSingleThreaded(), opt.getIpVersion());
ASSERT_NO_THROW(receiver.start());
auto pkt = receiver.getPkt();
EXPECT_EQ(pkt, nullptr);
ASSERT_NO_THROW(receiver.stop());
}
TEST(Receiver, multiThreaded) {
CommandOptions opt;
CommandOptionsHelper::process(opt, "perfdhcp -g multi -l 127.0.0.1 all");
ASSERT_FALSE(opt.isSingleThreaded());
FakeReceiverPerfSocket socket;
Receiver receiver(socket, opt.isSingleThreaded(), opt.getIpVersion());
ASSERT_NO_THROW(receiver.start());
auto pkt = receiver.getPkt();
EXPECT_EQ(pkt, nullptr);
ASSERT_NO_THROW(receiver.stop());
}
|