blob: 40250055b5471df0ee4ec483460bf1486ced0f0f [file] [log] [blame]
// Copyright (c) 2012 The Chromium OS 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 "shill/resolver.h"
#include <base/files/file_util.h>
#include <base/files/scoped_temp_dir.h>
#include <base/stl_util.h>
#include <base/strings/stringprintf.h>
#include <gtest/gtest.h>
#include "shill/mock_control.h"
using base::FilePath;
using std::string;
using std::vector;
using testing::Test;
namespace shill {
namespace {
const char kNameServer0[] = "8.8.8.8";
const char kNameServer1[] = "8.8.9.9";
const char kSearchDomain0[] = "chromium.org";
const char kSearchDomain1[] = "google.com";
const char kSearchDomain2[] = "crbug.com";
const char kExpectedOutput[] =
"nameserver 8.8.8.8\n"
"nameserver 8.8.9.9\n"
"search chromium.org google.com\n"
"options single-request timeout:1 attempts:5\n";
const char kExpectedIgnoredSearchOutput[] =
"nameserver 8.8.8.8\n"
"nameserver 8.8.9.9\n"
"search google.com\n"
"options single-request timeout:1 attempts:5\n";
} // namespace
class ResolverTest : public Test {
public:
ResolverTest() : resolver_(Resolver::GetInstance()) {}
virtual void SetUp() {
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
path_ = temp_dir_.path().Append("resolver");
resolver_->set_path(path_);
}
virtual void TearDown() {
resolver_->set_path(FilePath("")); // Don't try to save the store.
ASSERT_TRUE(temp_dir_.Delete());
}
protected:
string ReadFile();
base::ScopedTempDir temp_dir_;
Resolver* resolver_;
FilePath path_;
};
string ResolverTest::ReadFile() {
string data;
EXPECT_TRUE(base::ReadFileToString(resolver_->path_, &data));
return data;
}
TEST_F(ResolverTest, NonEmpty) {
EXPECT_FALSE(base::PathExists(path_));
EXPECT_TRUE(resolver_->ClearDNS());
MockControl control;
vector<string> dns_servers;
vector<string> domain_search;
dns_servers.push_back(kNameServer0);
dns_servers.push_back(kNameServer1);
domain_search.push_back(kSearchDomain0);
domain_search.push_back(kSearchDomain1);
EXPECT_TRUE(resolver_->SetDNSFromLists(dns_servers, domain_search));
EXPECT_TRUE(base::PathExists(path_));
EXPECT_EQ(kExpectedOutput, ReadFile());
EXPECT_TRUE(resolver_->ClearDNS());
}
TEST_F(ResolverTest, Empty) {
EXPECT_FALSE(base::PathExists(path_));
MockControl control;
vector<string> dns_servers;
vector<string> domain_search;
EXPECT_TRUE(resolver_->SetDNSFromLists(dns_servers, domain_search));
EXPECT_FALSE(base::PathExists(path_));
}
TEST_F(ResolverTest, IgnoredSearchList) {
EXPECT_FALSE(base::PathExists(path_));
EXPECT_TRUE(resolver_->ClearDNS());
MockControl control;
vector<string> dns_servers;
vector<string> domain_search;
dns_servers.push_back(kNameServer0);
dns_servers.push_back(kNameServer1);
domain_search.push_back(kSearchDomain0);
domain_search.push_back(kSearchDomain1);
vector<string> ignored_search;
ignored_search.push_back(kSearchDomain0);
ignored_search.push_back(kSearchDomain2);
resolver_->set_ignored_search_list(ignored_search);
EXPECT_TRUE(resolver_->SetDNSFromLists(dns_servers, domain_search));
EXPECT_TRUE(base::PathExists(path_));
EXPECT_EQ(kExpectedIgnoredSearchOutput, ReadFile());
EXPECT_TRUE(resolver_->ClearDNS());
}
} // namespace shill