3rd-party/libtins
1
0
Fork 0
mirror of https://github.com/mfontanini/libtins synced 2026-01-28 04:34:27 +01:00
Code Issues Releases Wiki Activity

Cleanup tests names and CMake script

Browse Source
This commit is contained in:
Matias Fontanini
2016-10-23 10:21:58 -07:00
parent aaba3dd46a
commit 9dbad2a26f
62 changed files with 70 additions and 189 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
tests/src/pdu_test.cpp Normal file
View File

@@ -0,0 +1,81 @@
#include <gtest/gtest.h>
#include <algorithm>
#include <string>
#include <stdint.h>
#include "ip.h"
#include "tcp.h"
#include "udp.h"
#include "rawpdu.h"
#include "pdu.h"
#include "packet.h"
using namespace std;
using namespace Tins;
class PDUTest : public testing::Test {
public:
};
TEST_F(PDUTest, FindPDU) {
IP ip = IP("192.168.0.1") / TCP(22, 52) / RawPDU("Test");
EXPECT_TRUE(ip.find_pdu<TCP>() != NULL);
EXPECT_TRUE(ip.find_pdu<RawPDU>() != NULL);
EXPECT_FALSE(ip.find_pdu<UDP>() != NULL);
TCP& t1 = ip.rfind_pdu<TCP>();
const TCP& t2 = ip.rfind_pdu<TCP>();
(void)t1;
(void)t2;
EXPECT_THROW(ip.rfind_pdu<UDP>(), pdu_not_found);
}
TEST_F(PDUTest, OperatorConcat) {
std::string raw_payload = "Test";
IP ip = IP("192.168.0.1") / TCP(22, 52) / RawPDU(raw_payload);
EXPECT_EQ(ip.dst_addr(), "192.168.0.1");
ASSERT_TRUE(ip.inner_pdu() != NULL);
TCP* tcp = ip.find_pdu<TCP>();
ASSERT_TRUE(tcp != NULL);
EXPECT_EQ(tcp->dport(), 22);
EXPECT_EQ(tcp->sport(), 52);
ASSERT_TRUE(tcp->inner_pdu() != NULL);
RawPDU* raw = tcp->find_pdu<RawPDU>();
ASSERT_TRUE(raw != NULL);
ASSERT_EQ(raw->payload_size(), raw_payload.size());
EXPECT_TRUE(std::equal(raw_payload.begin(), raw_payload.end(), raw->payload().begin()));
}
TEST_F(PDUTest, OperatorConcatOnPointers) {
std::string raw_payload = "Test";
IP ip = IP("192.168.0.1") / TCP(22, 52);
TCP* tcp = ip.find_pdu<TCP>();
ASSERT_TRUE(tcp != NULL);
tcp /= RawPDU(raw_payload);
RawPDU* raw = ip.find_pdu<RawPDU>();
ASSERT_TRUE(raw != NULL);
ASSERT_EQ(raw->payload_size(), raw_payload.size());
EXPECT_TRUE(std::equal(raw->payload().begin(), raw->payload().end(), raw_payload.begin()));
}
TEST_F(PDUTest, OperatorConcatOnPacket) {
std::string raw_payload = "Test";
Packet packet = IP("192.168.0.1") / TCP(22, 52);
TCP* tcp = packet.pdu()->find_pdu<TCP>();
ASSERT_TRUE(tcp != NULL);
tcp /= RawPDU(raw_payload);
RawPDU* raw = packet.pdu()->find_pdu<RawPDU>();
ASSERT_TRUE(raw != NULL);
ASSERT_EQ(raw->payload_size(), raw_payload.size());
EXPECT_TRUE(std::equal(raw->payload().begin(), raw->payload().end(), raw_payload.begin()));
}
TEST_F(PDUTest, TinsCast) {
PDU* null_pdu = 0;
TCP tcp;
PDU* pdu = &tcp;
EXPECT_EQ(tins_cast<TCP*>(pdu), &tcp);
EXPECT_EQ(tins_cast<const TCP*>(pdu), &tcp);
EXPECT_EQ(tins_cast<TCP*>(null_pdu), null_pdu);
EXPECT_EQ(tins_cast<UDP*>(pdu), null_pdu);
EXPECT_THROW(tins_cast<UDP>(*pdu), bad_tins_cast);
}