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

Added UDP test.

Browse Source
This commit is contained in:
Matias Fontanini
2011-09-15 10:21:51 -03:00
parent ff21304852
commit 46c9a21a72
1 changed files with 120 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

120
tests/src/udp.cpp Normal file
View File

@@ -0,0 +1,120 @@
#include <gtest/gtest.h>
#include <stdint.h>
#include "udp.h"
#include "pdu.h"
using namespace std;
using namespace Tins;
TEST(UDP, DefaultContructor) {
UDP udp;
EXPECT_EQ(udp.dport(), 0);
EXPECT_EQ(udp.sport(), 0);
EXPECT_FALSE(udp.inner_pdu());
}
TEST(UDP, DPort) {
UDP udp;
uint16_t port = 0x1234;
udp.dport(port);
ASSERT_EQ(udp.dport(), port);
}
TEST(UDP, SPort) {
UDP udp;
uint16_t port = 0x1234;
udp.sport(port);
ASSERT_EQ(udp.sport(), port);
}
TEST(UDP, Length) {
UDP udp;
uint16_t length = 0x1234;
udp.length(length);
ASSERT_EQ(udp.length(), length);
}
TEST(UDP, PDUType) {
UDP udp;
EXPECT_EQ(udp.pdu_type(), PDU::UDP);
}
TEST(UDP, CopyConstructor) {
UDP udp1;
udp1.dport(0x1234);
udp1.sport(0x4321);
udp1.length(0xdead);
UDP udp2(udp1);
EXPECT_EQ(udp2.sport(), udp1.sport());
EXPECT_EQ(udp2.dport(), udp1.dport());
EXPECT_EQ(udp2.length(), udp1.length());
}
TEST(UDP, CopyAssignmentOperator) {
UDP udp1;
udp1.dport(0x1234);
udp1.sport(0x4321);
udp1.length(0xdead);
UDP udp2 = udp1;
EXPECT_EQ(udp2.sport(), udp1.sport());
EXPECT_EQ(udp2.dport(), udp1.dport());
EXPECT_EQ(udp2.length(), udp1.length());
}
TEST(UDP, Clone) {
UDP udp1;
uint16_t sport = 0x1234, dport = 0x4321, length = 0xdead;
udp1.dport(dport);
udp1.sport(sport);
udp1.length(length);
UDP *udp2 = static_cast<UDP*>(udp1.clone_pdu());
ASSERT_TRUE(udp2);
EXPECT_EQ(udp2->sport(), sport);
EXPECT_EQ(udp2->dport(), dport);
EXPECT_EQ(udp2->length(), length);
EXPECT_EQ(udp2->pdu_type(), PDU::UDP);
delete udp2;
}
TEST(UDP, Serialize) {
UDP udp1;
uint16_t sport = 0x1234, dport = 0x4321, length = 0xdead;
udp1.dport(dport);
udp1.sport(sport);
udp1.length(length);
uint32_t size;
uint8_t *buffer = udp1.serialize(size);
ASSERT_TRUE(buffer);
delete[] buffer;
}
TEST(UDP, BufferClone) {
UDP udp1;
uint16_t sport = 0x1234, dport = 0x4321, length = 0xdead;
udp1.dport(dport);
udp1.sport(sport);
udp1.length(length);
uint32_t size;
uint8_t *buffer = udp1.serialize(size);
ASSERT_TRUE(buffer);
UDP udp2(buffer, size);
EXPECT_EQ(udp2.sport(), udp1.sport());
EXPECT_EQ(udp2.dport(), udp1.dport());
EXPECT_EQ(udp2.length(), udp1.length());
EXPECT_EQ(udp2.pdu_type(), PDU::UDP);
delete[] buffer;
}