| Index: net/dns/mdns_cache_unittest.cc
|
| diff --git a/net/dns/mdns_cache_unittest.cc b/net/dns/mdns_cache_unittest.cc
|
| deleted file mode 100644
|
| index c12ad6b6ec3c830fbb88489aea4d8c41df3ea5f4..0000000000000000000000000000000000000000
|
| --- a/net/dns/mdns_cache_unittest.cc
|
| +++ /dev/null
|
| @@ -1,375 +0,0 @@
|
| -// Copyright (c) 2013 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include <algorithm>
|
| -
|
| -#include "base/bind.h"
|
| -#include "net/dns/dns_response.h"
|
| -#include "net/dns/dns_test_util.h"
|
| -#include "net/dns/mdns_cache.h"
|
| -#include "net/dns/record_parsed.h"
|
| -#include "net/dns/record_rdata.h"
|
| -#include "testing/gmock/include/gmock/gmock.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -using ::testing::Return;
|
| -using ::testing::StrictMock;
|
| -
|
| -namespace net {
|
| -
|
| -static const uint8 kTestResponsesDifferentAnswers[] = {
|
| - // Answer 1
|
| - // ghs.l.google.com in DNS format.
|
| - 3, 'g', 'h', 's',
|
| - 1, 'l',
|
| - 6, 'g', 'o', 'o', 'g', 'l', 'e',
|
| - 3, 'c', 'o', 'm',
|
| - 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 53, // TTL (4 bytes) is 53 seconds.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 121, // RDATA is the IP: 74.125.95.121
|
| -
|
| - // Answer 2
|
| - // Pointer to answer 1
|
| - 0xc0, 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 53, // TTL (4 bytes) is 53 seconds.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 122, // RDATA is the IP: 74.125.95.122
|
| -};
|
| -
|
| -static const uint8 kTestResponsesSameAnswers[] = {
|
| - // Answer 1
|
| - // ghs.l.google.com in DNS format.
|
| - 3, 'g', 'h', 's',
|
| - 1, 'l',
|
| - 6, 'g', 'o', 'o', 'g', 'l', 'e',
|
| - 3, 'c', 'o', 'm',
|
| - 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 53, // TTL (4 bytes) is 53 seconds.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 121, // RDATA is the IP: 74.125.95.121
|
| -
|
| - // Answer 2
|
| - // Pointer to answer 1
|
| - 0xc0, 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 112, // TTL (4 bytes) is 112 seconds.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 121, // RDATA is the IP: 74.125.95.121
|
| -};
|
| -
|
| -static const uint8 kTestResponseTwoRecords[] = {
|
| - // Answer 1
|
| - // ghs.l.google.com in DNS format. (A)
|
| - 3, 'g', 'h', 's',
|
| - 1, 'l',
|
| - 6, 'g', 'o', 'o', 'g', 'l', 'e',
|
| - 3, 'c', 'o', 'm',
|
| - 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 53, // TTL (4 bytes) is 53 seconds.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 121, // RDATA is the IP: 74.125.95.121
|
| -
|
| - // Answer 2
|
| - // ghs.l.google.com in DNS format. (AAAA)
|
| - 3, 'g', 'h', 's',
|
| - 1, 'l',
|
| - 6, 'g', 'o', 'o', 'g', 'l', 'e',
|
| - 3, 'c', 'o', 'm',
|
| - 0x00,
|
| - 0x00, 0x1c, // TYPE is AAA.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 53, // TTL (4 bytes) is 53 seconds.
|
| - 0, 16, // RDLENGTH is 16 bytes.
|
| - 0x4a, 0x7d, 0x4a, 0x7d,
|
| - 0x5f, 0x79, 0x5f, 0x79,
|
| - 0x5f, 0x79, 0x5f, 0x79,
|
| - 0x5f, 0x79, 0x5f, 0x79,
|
| -};
|
| -
|
| -static const uint8 kTestResponsesGoodbyePacket[] = {
|
| - // Answer 1
|
| - // ghs.l.google.com in DNS format. (Goodbye packet)
|
| - 3, 'g', 'h', 's',
|
| - 1, 'l',
|
| - 6, 'g', 'o', 'o', 'g', 'l', 'e',
|
| - 3, 'c', 'o', 'm',
|
| - 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 0, // TTL (4 bytes) is zero.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 121, // RDATA is the IP: 74.125.95.121
|
| -
|
| - // Answer 2
|
| - // ghs.l.google.com in DNS format.
|
| - 3, 'g', 'h', 's',
|
| - 1, 'l',
|
| - 6, 'g', 'o', 'o', 'g', 'l', 'e',
|
| - 3, 'c', 'o', 'm',
|
| - 0x00,
|
| - 0x00, 0x01, // TYPE is A.
|
| - 0x00, 0x01, // CLASS is IN.
|
| - 0, 0, 0, 53, // TTL (4 bytes) is 53 seconds.
|
| - 0, 4, // RDLENGTH is 4 bytes.
|
| - 74, 125, 95, 121, // RDATA is the IP: 74.125.95.121
|
| -};
|
| -
|
| -class RecordRemovalMock {
|
| - public:
|
| - MOCK_METHOD1(OnRecordRemoved, void(const RecordParsed*));
|
| -};
|
| -
|
| -class MDnsCacheTest : public ::testing::Test {
|
| - public:
|
| - MDnsCacheTest()
|
| - : default_time_(base::Time::FromDoubleT(1234.0)) {}
|
| - virtual ~MDnsCacheTest() {}
|
| -
|
| - protected:
|
| - base::Time default_time_;
|
| - StrictMock<RecordRemovalMock> record_removal_;
|
| - MDnsCache cache_;
|
| -};
|
| -
|
| -// Test a single insert, corresponding lookup, and unsuccessful lookup.
|
| -TEST_F(MDnsCacheTest, InsertLookupSingle) {
|
| - DnsRecordParser parser(kT1ResponseDatagram, sizeof(kT1ResponseDatagram),
|
| - sizeof(dns_protocol::Header));
|
| - parser.SkipQuestion();
|
| -
|
| - scoped_ptr<const RecordParsed> record1;
|
| - scoped_ptr<const RecordParsed> record2;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - record2 = RecordParsed::CreateFrom(&parser, default_time_);
|
| -
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record1.Pass()));
|
| -
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record2.Pass()));
|
| -
|
| - cache_.FindDnsRecords(ARecordRdata::kType, "ghs.l.google.com", &results,
|
| - default_time_);
|
| -
|
| - EXPECT_EQ(1u, results.size());
|
| - EXPECT_EQ(default_time_, results.front()->time_created());
|
| -
|
| - EXPECT_EQ("ghs.l.google.com", results.front()->name());
|
| -
|
| - results.clear();
|
| - cache_.FindDnsRecords(PtrRecordRdata::kType, "ghs.l.google.com", &results,
|
| - default_time_);
|
| -
|
| - EXPECT_EQ(0u, results.size());
|
| -}
|
| -
|
| -// Test that records expire when their ttl has passed.
|
| -TEST_F(MDnsCacheTest, Expiration) {
|
| - DnsRecordParser parser(kT1ResponseDatagram, sizeof(kT1ResponseDatagram),
|
| - sizeof(dns_protocol::Header));
|
| - parser.SkipQuestion();
|
| - scoped_ptr<const RecordParsed> record1;
|
| - scoped_ptr<const RecordParsed> record2;
|
| -
|
| - std::vector<const RecordParsed*> results;
|
| - const RecordParsed* record_to_be_deleted;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - base::TimeDelta ttl1 = base::TimeDelta::FromSeconds(record1->ttl());
|
| -
|
| - record2 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - base::TimeDelta ttl2 = base::TimeDelta::FromSeconds(record2->ttl());
|
| - record_to_be_deleted = record2.get();
|
| -
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record1.Pass()));
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record2.Pass()));
|
| -
|
| - cache_.FindDnsRecords(ARecordRdata::kType, "ghs.l.google.com", &results,
|
| - default_time_);
|
| -
|
| - EXPECT_EQ(1u, results.size());
|
| -
|
| - EXPECT_EQ(default_time_ + ttl2, cache_.next_expiration());
|
| -
|
| -
|
| - cache_.FindDnsRecords(ARecordRdata::kType, "ghs.l.google.com", &results,
|
| - default_time_ + ttl2);
|
| -
|
| - EXPECT_EQ(0u, results.size());
|
| -
|
| - EXPECT_CALL(record_removal_, OnRecordRemoved(record_to_be_deleted));
|
| -
|
| - cache_.CleanupRecords(default_time_ + ttl2, base::Bind(
|
| - &RecordRemovalMock::OnRecordRemoved, base::Unretained(&record_removal_)));
|
| -
|
| - // To make sure that we've indeed removed them from the map, check no funny
|
| - // business happens once they're deleted for good.
|
| -
|
| - EXPECT_EQ(default_time_ + ttl1, cache_.next_expiration());
|
| - cache_.FindDnsRecords(ARecordRdata::kType, "ghs.l.google.com", &results,
|
| - default_time_ + ttl2);
|
| -
|
| - EXPECT_EQ(0u, results.size());
|
| -}
|
| -
|
| -// Test that a new record replacing one with the same identity (name/rrtype for
|
| -// unique records) causes the cache to output a "record changed" event.
|
| -TEST_F(MDnsCacheTest, RecordChange) {
|
| - DnsRecordParser parser(kTestResponsesDifferentAnswers,
|
| - sizeof(kTestResponsesDifferentAnswers),
|
| - 0);
|
| -
|
| - scoped_ptr<const RecordParsed> record1;
|
| - scoped_ptr<const RecordParsed> record2;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - record2 = RecordParsed::CreateFrom(&parser, default_time_);
|
| -
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record1.Pass()));
|
| - EXPECT_EQ(MDnsCache::RecordChanged,
|
| - cache_.UpdateDnsRecord(record2.Pass()));
|
| -}
|
| -
|
| -// Test that a new record replacing an otherwise identical one already in the
|
| -// cache causes the cache to output a "no change" event.
|
| -TEST_F(MDnsCacheTest, RecordNoChange) {
|
| - DnsRecordParser parser(kTestResponsesSameAnswers,
|
| - sizeof(kTestResponsesSameAnswers),
|
| - 0);
|
| -
|
| - scoped_ptr<const RecordParsed> record1;
|
| - scoped_ptr<const RecordParsed> record2;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - record2 = RecordParsed::CreateFrom(&parser, default_time_ +
|
| - base::TimeDelta::FromSeconds(1));
|
| -
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record1.Pass()));
|
| - EXPECT_EQ(MDnsCache::NoChange, cache_.UpdateDnsRecord(record2.Pass()));
|
| -}
|
| -
|
| -// Test that the next expiration time of the cache is updated properly on record
|
| -// insertion.
|
| -TEST_F(MDnsCacheTest, RecordPreemptExpirationTime) {
|
| - DnsRecordParser parser(kTestResponsesSameAnswers,
|
| - sizeof(kTestResponsesSameAnswers),
|
| - 0);
|
| -
|
| - scoped_ptr<const RecordParsed> record1;
|
| - scoped_ptr<const RecordParsed> record2;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - record2 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - base::TimeDelta ttl1 = base::TimeDelta::FromSeconds(record1->ttl());
|
| - base::TimeDelta ttl2 = base::TimeDelta::FromSeconds(record2->ttl());
|
| -
|
| - EXPECT_EQ(base::Time(), cache_.next_expiration());
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record2.Pass()));
|
| - EXPECT_EQ(default_time_ + ttl2, cache_.next_expiration());
|
| - EXPECT_EQ(MDnsCache::NoChange, cache_.UpdateDnsRecord(record1.Pass()));
|
| - EXPECT_EQ(default_time_ + ttl1, cache_.next_expiration());
|
| -}
|
| -
|
| -// Test that the cache handles mDNS "goodbye" packets correctly, not adding the
|
| -// records to the cache if they are not already there, and eventually removing
|
| -// records from the cache if they are.
|
| -TEST_F(MDnsCacheTest, GoodbyePacket) {
|
| - DnsRecordParser parser(kTestResponsesGoodbyePacket,
|
| - sizeof(kTestResponsesGoodbyePacket),
|
| - 0);
|
| -
|
| - scoped_ptr<const RecordParsed> record_goodbye;
|
| - scoped_ptr<const RecordParsed> record_hello;
|
| - scoped_ptr<const RecordParsed> record_goodbye2;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record_goodbye = RecordParsed::CreateFrom(&parser, default_time_);
|
| - record_hello = RecordParsed::CreateFrom(&parser, default_time_);
|
| - parser = DnsRecordParser(kTestResponsesGoodbyePacket,
|
| - sizeof(kTestResponsesGoodbyePacket),
|
| - 0);
|
| - record_goodbye2 = RecordParsed::CreateFrom(&parser, default_time_);
|
| -
|
| - base::TimeDelta ttl = base::TimeDelta::FromSeconds(record_hello->ttl());
|
| -
|
| - EXPECT_EQ(base::Time(), cache_.next_expiration());
|
| - EXPECT_EQ(MDnsCache::NoChange, cache_.UpdateDnsRecord(record_goodbye.Pass()));
|
| - EXPECT_EQ(base::Time(), cache_.next_expiration());
|
| - EXPECT_EQ(MDnsCache::RecordAdded,
|
| - cache_.UpdateDnsRecord(record_hello.Pass()));
|
| - EXPECT_EQ(default_time_ + ttl, cache_.next_expiration());
|
| - EXPECT_EQ(MDnsCache::NoChange,
|
| - cache_.UpdateDnsRecord(record_goodbye2.Pass()));
|
| - EXPECT_EQ(default_time_ + base::TimeDelta::FromSeconds(1),
|
| - cache_.next_expiration());
|
| -}
|
| -
|
| -TEST_F(MDnsCacheTest, AnyRRType) {
|
| - DnsRecordParser parser(kTestResponseTwoRecords,
|
| - sizeof(kTestResponseTwoRecords),
|
| - 0);
|
| -
|
| - scoped_ptr<const RecordParsed> record1;
|
| - scoped_ptr<const RecordParsed> record2;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - record2 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record1.Pass()));
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record2.Pass()));
|
| -
|
| - cache_.FindDnsRecords(0, "ghs.l.google.com", &results, default_time_);
|
| -
|
| - EXPECT_EQ(2u, results.size());
|
| - EXPECT_EQ(default_time_, results.front()->time_created());
|
| -
|
| - EXPECT_EQ("ghs.l.google.com", results[0]->name());
|
| - EXPECT_EQ("ghs.l.google.com", results[1]->name());
|
| - EXPECT_EQ(dns_protocol::kTypeA,
|
| - std::min(results[0]->type(), results[1]->type()));
|
| - EXPECT_EQ(dns_protocol::kTypeAAAA,
|
| - std::max(results[0]->type(), results[1]->type()));
|
| -}
|
| -
|
| -TEST_F(MDnsCacheTest, RemoveRecord) {
|
| - DnsRecordParser parser(kT1ResponseDatagram, sizeof(kT1ResponseDatagram),
|
| - sizeof(dns_protocol::Header));
|
| - parser.SkipQuestion();
|
| -
|
| - scoped_ptr<const RecordParsed> record1;
|
| - std::vector<const RecordParsed*> results;
|
| -
|
| - record1 = RecordParsed::CreateFrom(&parser, default_time_);
|
| - EXPECT_EQ(MDnsCache::RecordAdded, cache_.UpdateDnsRecord(record1.Pass()));
|
| -
|
| - cache_.FindDnsRecords(dns_protocol::kTypeCNAME, "codereview.chromium.org",
|
| - &results, default_time_);
|
| -
|
| - EXPECT_EQ(1u, results.size());
|
| -
|
| - scoped_ptr<const RecordParsed> record_out =
|
| - cache_.RemoveRecord(results.front());
|
| -
|
| - EXPECT_EQ(record_out.get(), results.front());
|
| -
|
| - cache_.FindDnsRecords(dns_protocol::kTypeCNAME, "codereview.chromium.org",
|
| - &results, default_time_);
|
| -
|
| - EXPECT_EQ(0u, results.size());
|
| -}
|
| -
|
| -} // namespace net
|
|
|