Index: chrome/browser/extensions/api/dns/dns_apitest.cc |
diff --git a/chrome/browser/extensions/api/dns/dns_apitest.cc b/chrome/browser/extensions/api/dns/dns_apitest.cc |
index 0918425cc89143207b5ab7a3e1e4b1ea54c64b0e..52448cac3fe238fd9114f2d63233843244b1f31c 100644 |
--- a/chrome/browser/extensions/api/dns/dns_apitest.cc |
+++ b/chrome/browser/extensions/api/dns/dns_apitest.cc |
@@ -3,35 +3,18 @@ |
// found in the LICENSE file. |
#include "base/memory/ref_counted.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "chrome/browser/extensions/api/dns/mock_host_resolver_creator.h" |
#include "chrome/browser/extensions/extension_apitest.h" |
-#include "chrome/browser/extensions/extension_function_test_utils.h" |
-#include "extensions/browser/api/dns/dns_api.h" |
#include "extensions/browser/api/dns/host_resolver_wrapper.h" |
-#include "extensions/common/switches.h" |
-#include "net/base/net_errors.h" |
-#include "net/base/net_util.h" |
+#include "extensions/browser/api/dns/mock_host_resolver_creator.h" |
#include "net/dns/mock_host_resolver.h" |
-using extension_function_test_utils::CreateEmptyExtension; |
-using extension_function_test_utils::RunFunctionAndReturnSingleResult; |
- |
-namespace { |
- |
class DnsApiTest : public ExtensionApiTest { |
public: |
- DnsApiTest() : resolver_event_(true, false), |
- resolver_creator_(new extensions::MockHostResolverCreator()) { |
- } |
- |
- virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
- ExtensionApiTest::SetUpCommandLine(command_line); |
- command_line->AppendSwitch( |
- extensions::switches::kEnableExperimentalExtensionApis); |
- } |
+ DnsApiTest() : resolver_creator_(new extensions::MockHostResolverCreator()) {} |
+ private: |
virtual void SetUpOnMainThread() OVERRIDE { |
+ ExtensionApiTest::SetUpOnMainThread(); |
extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting( |
resolver_creator_->CreateMockHostResolver()); |
} |
@@ -40,69 +23,15 @@ class DnsApiTest : public ExtensionApiTest { |
extensions::HostResolverWrapper::GetInstance()-> |
SetHostResolverForTesting(NULL); |
resolver_creator_->DeleteMockHostResolver(); |
+ ExtensionApiTest::CleanUpOnMainThread(); |
} |
- private: |
- base::WaitableEvent resolver_event_; |
- |
// The MockHostResolver asserts that it's used on the same thread on which |
// it's created, which is actually a stronger rule than its real counterpart. |
// But that's fine; it's good practice. |
scoped_refptr<extensions::MockHostResolverCreator> resolver_creator_; |
}; |
-} // namespace |
- |
-IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsResolveIPLiteral) { |
- scoped_refptr<extensions::DnsResolveFunction> resolve_function( |
- new extensions::DnsResolveFunction()); |
- scoped_refptr<extensions::Extension> empty_extension(CreateEmptyExtension()); |
- |
- resolve_function->set_extension(empty_extension.get()); |
- resolve_function->set_has_callback(true); |
- |
- scoped_ptr<base::Value> result(RunFunctionAndReturnSingleResult( |
- resolve_function.get(), "[\"127.0.0.1\"]", browser())); |
- ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType()); |
- base::DictionaryValue *value = |
- static_cast<base::DictionaryValue*>(result.get()); |
- |
- int resultCode; |
- EXPECT_TRUE(value->GetInteger("resultCode", &resultCode)); |
- EXPECT_EQ(net::OK, resultCode); |
- |
- std::string address; |
- EXPECT_TRUE(value->GetString("address", &address)); |
- EXPECT_EQ("127.0.0.1", address); |
-} |
- |
-IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsResolveHostname) { |
- scoped_refptr<extensions::DnsResolveFunction> resolve_function( |
- new extensions::DnsResolveFunction()); |
- scoped_refptr<extensions::Extension> empty_extension(CreateEmptyExtension()); |
- |
- resolve_function->set_extension(empty_extension.get()); |
- resolve_function->set_has_callback(true); |
- |
- std::string function_arguments("[\""); |
- function_arguments += extensions::MockHostResolverCreator::kHostname; |
- function_arguments += "\"]"; |
- scoped_ptr<base::Value> result( |
- RunFunctionAndReturnSingleResult(resolve_function.get(), |
- function_arguments, browser())); |
- ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType()); |
- base::DictionaryValue *value = |
- static_cast<base::DictionaryValue*>(result.get()); |
- |
- int resultCode; |
- EXPECT_TRUE(value->GetInteger("resultCode", &resultCode)); |
- EXPECT_EQ(net::OK, resultCode); |
- |
- std::string address; |
- EXPECT_TRUE(value->GetString("address", &address)); |
- EXPECT_EQ(extensions::MockHostResolverCreator::kAddress, address); |
-} |
- |
IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsExtension) { |
ASSERT_TRUE(RunExtensionTest("dns/api")) << message_; |
} |