mirror of
https://github.com/mfontanini/libtins
synced 2026-01-26 03:51:35 +01:00
d84f10cf08
Initial code cleanup More code cleanup Cleanup more code Cleanup Dot11 code Fix OSX build issue Cleanup examples Fix ref and pointer declaration syntax Fix braces
98 lines
2.9 KiB
C++
98 lines
2.9 KiB
C++
#include "dot11/dot11_control.h"
|
|
|
|
#ifdef HAVE_DOT11
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "tests/dot11.h"
|
|
|
|
|
|
using namespace std;
|
|
using namespace Tins;
|
|
|
|
typedef Dot11::address_type address_type;
|
|
|
|
class Dot11BlockAckRequestTest : public testing::Test {
|
|
public:
|
|
static const address_type empty_addr, hwaddr;
|
|
static const uint8_t expected_packet[];
|
|
};
|
|
|
|
const uint8_t Dot11BlockAckRequestTest::expected_packet[] = {
|
|
132, 0, 176, 1, 0, 33, 107, 2, 154, 230, 0, 28, 223, 215, 13, 85, 4,
|
|
0, 176, 33
|
|
};
|
|
|
|
void test_equals(const Dot11BlockAckRequest& dot1, const Dot11BlockAckRequest& dot2) {
|
|
EXPECT_EQ(dot1.fragment_number(), dot2.fragment_number());
|
|
EXPECT_EQ(dot1.start_sequence(), dot2.start_sequence());
|
|
EXPECT_EQ(dot1.bar_control(), dot2.bar_control());
|
|
}
|
|
|
|
void test_equals_expected(const Dot11BlockAckRequest& dot11) {
|
|
EXPECT_EQ(dot11.type(), Dot11::CONTROL);
|
|
EXPECT_EQ(dot11.subtype(), Dot11::BLOCK_ACK_REQ);
|
|
EXPECT_EQ(dot11.bar_control(), 4);
|
|
EXPECT_EQ(dot11.start_sequence(), 539);
|
|
EXPECT_EQ(dot11.fragment_number(), 0);
|
|
}
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, Constructor) {
|
|
Dot11BlockAckRequest dot11;
|
|
test_equals_empty(static_cast<const Dot11ControlTA&>(dot11));
|
|
EXPECT_EQ(dot11.subtype(), Dot11::BLOCK_ACK_REQ);
|
|
EXPECT_EQ(dot11.fragment_number(), 0);
|
|
EXPECT_EQ(dot11.start_sequence(), 0);
|
|
EXPECT_EQ(dot11.bar_control(), 0);
|
|
}
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, ConstructorFromBuffer) {
|
|
Dot11BlockAckRequest dot11(expected_packet, sizeof(expected_packet));
|
|
test_equals_expected(dot11);
|
|
}
|
|
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, CopyConstructor) {
|
|
Dot11BlockAckRequest dot1;
|
|
dot1.fragment_number(6);
|
|
dot1.start_sequence(0x294);
|
|
dot1.bar_control(0x9);
|
|
Dot11BlockAckRequest dot2(dot1);
|
|
test_equals(dot1, dot2);
|
|
}
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, CopyAssignmentOperator) {
|
|
Dot11BlockAckRequest dot1;
|
|
dot1.fragment_number(6);
|
|
dot1.start_sequence(0x294);
|
|
dot1.bar_control(0x9);
|
|
Dot11BlockAckRequest dot2;
|
|
dot2 = dot1;
|
|
test_equals(dot1, dot2);
|
|
}
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, ClonePDU) {
|
|
Dot11BlockAckRequest dot1;
|
|
dot1.fragment_number(6);
|
|
dot1.start_sequence(0x294);
|
|
dot1.bar_control(0x9);
|
|
Internals::smart_ptr<Dot11BlockAckRequest>::type dot2(dot1.clone());
|
|
test_equals(dot1, *dot2);
|
|
}
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, FromBytes) {
|
|
Internals::smart_ptr<PDU>::type dot11(Dot11::from_bytes(expected_packet, sizeof(expected_packet)));
|
|
ASSERT_TRUE(dot11.get() != NULL);
|
|
const Dot11BlockAckRequest* inner = dot11->find_pdu<Dot11BlockAckRequest>();
|
|
ASSERT_TRUE(inner != NULL);
|
|
test_equals_expected(*inner);
|
|
}
|
|
|
|
TEST_F(Dot11BlockAckRequestTest, Serialize) {
|
|
Dot11BlockAckRequest pdu(expected_packet, sizeof(expected_packet));
|
|
PDU::serialization_type buffer = pdu.serialize();
|
|
ASSERT_EQ(sizeof(expected_packet), buffer.size());
|
|
EXPECT_TRUE(std::equal(buffer.begin(), buffer.end(), expected_packet));
|
|
}
|
|
|
|
#endif // HAVE_DOT11
|