mirror of
https://github.com/mfontanini/libtins
synced 2026-01-30 05:24:26 +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
94 lines
2.9 KiB
C++
94 lines
2.9 KiB
C++
#include "dot11/dot11_assoc.h"
|
|
|
|
#ifdef HAVE_DOT11
|
|
|
|
#include <gtest/gtest.h>
|
|
#include "tests/dot11_mgmt.h"
|
|
|
|
|
|
using namespace std;
|
|
using namespace Tins;
|
|
|
|
typedef Dot11::address_type address_type;
|
|
|
|
class Dot11AssocRequestTest : public testing::Test {
|
|
public:
|
|
static const address_type empty_addr, hwaddr;
|
|
static const uint8_t expected_packet[];
|
|
//static void test_equals_expected(const Dot11Beacon&dot11);
|
|
};
|
|
|
|
const uint8_t Dot11AssocRequestTest::expected_packet[] = {
|
|
1, 1, 79, 35, 0, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 6, 2, 3, 4, 5, 6, 7,
|
|
0, 0, 21, 32, 243, 146
|
|
};
|
|
|
|
void test_equals(const Dot11AssocRequest& dot1, const Dot11AssocRequest& dot2) {
|
|
test_equals(dot1.capabilities(), dot2.capabilities());
|
|
EXPECT_EQ(dot1.listen_interval(), dot2.listen_interval());
|
|
test_equals(
|
|
static_cast<const Dot11ManagementFrame&>(dot1),
|
|
static_cast<const Dot11ManagementFrame&>(dot2)
|
|
);
|
|
}
|
|
|
|
void test_equals_expected(const Dot11AssocRequest& dot11) {
|
|
test_equals_expected(static_cast<const Dot11ManagementFrame&>(dot11));
|
|
EXPECT_EQ(dot11.listen_interval(), 0x92f3);
|
|
EXPECT_EQ(dot11.subtype(), Dot11::ASSOC_REQ);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, Constructor) {
|
|
Dot11AssocRequest dot11;
|
|
test_equals_empty(static_cast<const Dot11ManagementFrame&>(dot11));
|
|
EXPECT_EQ(dot11.listen_interval(), 0);
|
|
EXPECT_EQ(dot11.subtype(), Dot11::ASSOC_REQ);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, ConstructorFromBuffer) {
|
|
Dot11AssocRequest dot11(expected_packet, sizeof(expected_packet));
|
|
test_equals_expected(dot11);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, CopyConstructor) {
|
|
Dot11AssocRequest dot1(expected_packet, sizeof(expected_packet));
|
|
Dot11AssocRequest dot2(dot1);
|
|
test_equals(dot1, dot2);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, CopyAssignmentOperator) {
|
|
Dot11AssocRequest dot1(expected_packet, sizeof(expected_packet));
|
|
Dot11AssocRequest dot2;
|
|
dot2 = dot1;
|
|
test_equals(dot1, dot2);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, ListenInterval) {
|
|
Dot11AssocRequest dot11;
|
|
dot11.listen_interval(0x92fd);
|
|
EXPECT_EQ(dot11.listen_interval(), 0x92fd);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, ClonePDU) {
|
|
Dot11AssocRequest dot1(expected_packet, sizeof(expected_packet));
|
|
Internals::smart_ptr<Dot11AssocRequest>::type dot2(dot1.clone());
|
|
test_equals(dot1, *dot2);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, FromBytes) {
|
|
Internals::smart_ptr<PDU>::type dot11(Dot11::from_bytes(expected_packet, sizeof(expected_packet)));
|
|
ASSERT_TRUE(dot11.get() != NULL);
|
|
const Dot11AssocRequest* inner = dot11->find_pdu<Dot11AssocRequest>();
|
|
ASSERT_TRUE(inner != NULL);
|
|
test_equals_expected(*inner);
|
|
}
|
|
|
|
TEST_F(Dot11AssocRequestTest, Serialize) {
|
|
Dot11AssocRequest 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
|