3rd-party/libtins
1
0
Fork 0
mirror of https://github.com/mfontanini/libtins synced 2026-01-23 02:35:57 +01:00
Code Issues Releases Wiki Activity

Parse and serialize MX preference field correctly

Browse Source
This commit is contained in:
Matias Fontanini
2016-01-11 15:48:03 -08:00
parent 75add84741
commit ca56cc10dd
3 changed files with 89 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tests/src/dns.cpp
View File

@@ -118,6 +118,7 @@ TEST_F(DNSTest, ConstructorFromBuffer2) {
}
DNS::resources_type resources = dns.answers();
size_t resource_index = 0;
for(DNS::resources_type::const_iterator it = resources.begin(); it != resources.end(); ++it) {
EXPECT_EQ("google.com", it->dname());
EXPECT_EQ(DNS::MX, it->type());
@@ -130,6 +131,13 @@ TEST_F(DNSTest, ConstructorFromBuffer2) {
it->data() == "alt5.aspmx.l.google.com" ||
it->data() == "aspmx.l.google.com"
);
if (resource_index == 0) {
EXPECT_EQ(50, it->preference());
}
else if (resource_index == 1) {
EXPECT_EQ(40, it->preference());
}
resource_index++;
}
// Add some stuff and see if something gets broken
if(i == 0) {
@@ -450,3 +458,18 @@ TEST_F(DNSTest, ItAintGonnaCorrupt) {
EXPECT_EQ(it->query_class(), DNS::IN);
}
}
TEST_F(DNSTest, MXPreferenceField) {
DNS dns1;
dns1.add_answer(
DNS::Resource("example.com", "mail.example.com", DNS::MX, DNS::IN, 0x762, 42)
);
DNS::serialization_type buffer = dns1.serialize();
DNS dns2(&buffer[0], buffer.size());
DNS::resources_type answers = dns1.answers();
ASSERT_EQ(1, answers.size());
const DNS::Resource& resource = *answers.begin();
EXPECT_EQ(42, resource.preference());
EXPECT_EQ("example.com", resource.dname());
}