blob: ee76a9a0decd90b3c5b9a807412d44d3f1a64dc8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright (c) 2012 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 "net/dns/dns_config_service.h"
#include "base/bind.h"
#include "base/compiler_specific.h"
#include "base/memory/scoped_ptr.h"
#include "base/message_loop.h"
#include "base/test/test_timeouts.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace net {
namespace {
class DnsConfigServiceTest : public testing::Test,
public DnsConfigService::Observer {
public:
void OnConfigChanged(const DnsConfig& config) OVERRIDE {
EXPECT_FALSE(config.Equals(last_config_)) <<
"Config must be different from last call.";
last_config_ = config;
if (quit_on_config_)
MessageLoop::current()->Quit();
}
void Timeout() const {
MessageLoop::current()->Quit();
}
protected:
// test::Test methods
virtual void SetUp() OVERRIDE {
service_.reset(new DnsConfigService());
quit_on_config_ = false;
}
DnsConfig last_config_;
bool quit_on_config_;
// Service under test.
scoped_ptr<DnsConfigService> service_;
};
} // namespace
TEST_F(DnsConfigServiceTest, NotifyOnChange) {
service_->AddObserver(this);
service_->Watch();
DnsConfig config;
IPAddressNumber ip;
ASSERT_TRUE(ParseIPLiteralToNumber("1.2.3.4", &ip));
config.nameservers.push_back(IPEndPoint(ip, 53));
ASSERT_TRUE(config.IsValid());
service_->OnConfigRead(config);
EXPECT_FALSE(last_config_.Equals(config)) <<
"Unexpected notification with incomplete config.";
DnsHosts hosts;
ParseHosts("127.0.0.1 localhost", &hosts);
ASSERT_FALSE(hosts.empty());
service_->OnHostsRead(hosts);
EXPECT_TRUE(last_config_.hosts == hosts);
EXPECT_TRUE(last_config_.EqualsIgnoreHosts(config));
DnsConfig complete_config = config;
complete_config.hosts = hosts;
EXPECT_TRUE(last_config_.Equals(complete_config));
// Check if it ignores no-change in config or hosts.
service_->OnConfigRead(config);
service_->OnHostsRead(hosts);
// Check if it provides config and hosts after AddObserver;
service_->RemoveObserver(this);
last_config_ = DnsConfig();
service_->AddObserver(this);
EXPECT_TRUE(last_config_.Equals(complete_config));
}
#if defined(OS_POSIX) || defined(OS_WIN)
// This is really an integration test.
TEST_F(DnsConfigServiceTest, GetSystemConfig) {
scoped_ptr<DnsConfigService> service(DnsConfigService::CreateSystemService());
// Quit the loop after timeout unless cancelled
const base::TimeDelta kTimeout = TestTimeouts::action_timeout();
base::WeakPtrFactory<DnsConfigServiceTest> factory_(this);
MessageLoop::current()->PostDelayedTask(
FROM_HERE,
base::Bind(&DnsConfigServiceTest::Timeout, factory_.GetWeakPtr()),
kTimeout);
service->AddObserver(this);
service->Watch();
quit_on_config_ = true;
MessageLoop::current()->Run();
ASSERT_TRUE(last_config_.IsValid()) << "Did not receive DnsConfig in " <<
kTimeout.InMilliseconds() << "ms";
}
#endif // OS_POSIX || OS_WIN
} // namespace net
|
