1 files changed, 116 insertions, 0 deletions
diff --git a/src/bin/perfdhcp/tests/receiver_unittest.cc b/src/bin/perfdhcp/tests/receiver_unittest.cc
new file mode 100644
index 0000000..1a1cb68
--- /dev/null
+++ b/src/bin/perfdhcp/tests/receiver_unittest.cc
@@ -0,0 +1,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());
+}