Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(71)

Side by Side Diff: chrome/browser/extensions/api/dns/dns_apitest.cc

Issue 394103004: Move DnsApiTest.DnsResolveIPLiteral and DnsApiTest.DnsResolveHostname to app_shell_browsertests. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "base/memory/ref_counted.h" 5 #include "base/memory/ref_counted.h"
6 #include "base/memory/scoped_ptr.h"
7 #include "chrome/browser/extensions/api/dns/mock_host_resolver_creator.h"
8 #include "chrome/browser/extensions/extension_apitest.h" 6 #include "chrome/browser/extensions/extension_apitest.h"
9 #include "chrome/browser/extensions/extension_function_test_utils.h"
10 #include "extensions/browser/api/dns/dns_api.h"
11 #include "extensions/browser/api/dns/host_resolver_wrapper.h" 7 #include "extensions/browser/api/dns/host_resolver_wrapper.h"
12 #include "extensions/common/switches.h" 8 #include "extensions/browser/api/dns/mock_host_resolver_creator.h"
13 #include "net/base/net_errors.h"
14 #include "net/base/net_util.h"
15 #include "net/dns/mock_host_resolver.h" 9 #include "net/dns/mock_host_resolver.h"
16 10
17 using extension_function_test_utils::CreateEmptyExtension;
18 using extension_function_test_utils::RunFunctionAndReturnSingleResult;
19
20 namespace {
21
22 class DnsApiTest : public ExtensionApiTest { 11 class DnsApiTest : public ExtensionApiTest {
23 public: 12 public:
24 DnsApiTest() : resolver_event_(true, false), 13 DnsApiTest() : resolver_creator_(new extensions::MockHostResolverCreator()) {}
25 resolver_creator_(new extensions::MockHostResolverCreator()) {
26 }
27 14
28 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { 15 private:
29 ExtensionApiTest::SetUpCommandLine(command_line);
30 command_line->AppendSwitch(
31 extensions::switches::kEnableExperimentalExtensionApis);
32 }
33
34 virtual void SetUpOnMainThread() OVERRIDE { 16 virtual void SetUpOnMainThread() OVERRIDE {
17 ExtensionApiTest::SetUpOnMainThread();
35 extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting( 18 extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting(
36 resolver_creator_->CreateMockHostResolver()); 19 resolver_creator_->CreateMockHostResolver());
37 } 20 }
38 21
39 virtual void CleanUpOnMainThread() OVERRIDE { 22 virtual void CleanUpOnMainThread() OVERRIDE {
40 extensions::HostResolverWrapper::GetInstance()-> 23 extensions::HostResolverWrapper::GetInstance()->
41 SetHostResolverForTesting(NULL); 24 SetHostResolverForTesting(NULL);
42 resolver_creator_->DeleteMockHostResolver(); 25 resolver_creator_->DeleteMockHostResolver();
26 ExtensionApiTest::CleanUpOnMainThread();
43 } 27 }
44 28
45 private:
46 base::WaitableEvent resolver_event_;
47
48 // The MockHostResolver asserts that it's used on the same thread on which 29 // The MockHostResolver asserts that it's used on the same thread on which
49 // it's created, which is actually a stronger rule than its real counterpart. 30 // it's created, which is actually a stronger rule than its real counterpart.
50 // But that's fine; it's good practice. 31 // But that's fine; it's good practice.
51 scoped_refptr<extensions::MockHostResolverCreator> resolver_creator_; 32 scoped_refptr<extensions::MockHostResolverCreator> resolver_creator_;
52 }; 33 };
53 34
54 } // namespace
55
56 IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsResolveIPLiteral) {
57 scoped_refptr<extensions::DnsResolveFunction> resolve_function(
58 new extensions::DnsResolveFunction());
59 scoped_refptr<extensions::Extension> empty_extension(CreateEmptyExtension());
60
61 resolve_function->set_extension(empty_extension.get());
62 resolve_function->set_has_callback(true);
63
64 scoped_ptr<base::Value> result(RunFunctionAndReturnSingleResult(
65 resolve_function.get(), "[\"127.0.0.1\"]", browser()));
66 ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType());
67 base::DictionaryValue *value =
68 static_cast<base::DictionaryValue*>(result.get());
69
70 int resultCode;
71 EXPECT_TRUE(value->GetInteger("resultCode", &resultCode));
72 EXPECT_EQ(net::OK, resultCode);
73
74 std::string address;
75 EXPECT_TRUE(value->GetString("address", &address));
76 EXPECT_EQ("127.0.0.1", address);
77 }
78
79 IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsResolveHostname) {
80 scoped_refptr<extensions::DnsResolveFunction> resolve_function(
81 new extensions::DnsResolveFunction());
82 scoped_refptr<extensions::Extension> empty_extension(CreateEmptyExtension());
83
84 resolve_function->set_extension(empty_extension.get());
85 resolve_function->set_has_callback(true);
86
87 std::string function_arguments("[\"");
88 function_arguments += extensions::MockHostResolverCreator::kHostname;
89 function_arguments += "\"]";
90 scoped_ptr<base::Value> result(
91 RunFunctionAndReturnSingleResult(resolve_function.get(),
92 function_arguments, browser()));
93 ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType());
94 base::DictionaryValue *value =
95 static_cast<base::DictionaryValue*>(result.get());
96
97 int resultCode;
98 EXPECT_TRUE(value->GetInteger("resultCode", &resultCode));
99 EXPECT_EQ(net::OK, resultCode);
100
101 std::string address;
102 EXPECT_TRUE(value->GetString("address", &address));
103 EXPECT_EQ(extensions::MockHostResolverCreator::kAddress, address);
104 }
105
106 IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsExtension) { 35 IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsExtension) {
107 ASSERT_TRUE(RunExtensionTest("dns/api")) << message_; 36 ASSERT_TRUE(RunExtensionTest("dns/api")) << message_;
108 } 37 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698