Index: apps/shell/browser/dns_apitest.cc |
diff --git a/apps/shell/browser/dns_apitest.cc b/apps/shell/browser/dns_apitest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..076e713e197ceac56c09bad58b677a365a29fc86 |
--- /dev/null |
+++ b/apps/shell/browser/dns_apitest.cc |
@@ -0,0 +1,127 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
Yoyo Zhou
2014/07/16 02:30:25
I put this file in apps/shell/browser to avoid int
James Cook
2014/07/16 17:15:31
I think the time has come to move app_shell out of
Yoyo Zhou
2014/07/18 02:45:30
I'll move to src/extensions/shell after this CL. F
tfarina
2014/07/18 02:47:37
extensions/shell :) Cool!
|
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "apps/shell/test/shell_test.h" |
+#include "base/memory/ref_counted.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/values.h" |
mmenke
2014/07/16 14:05:54
You're missing the header for OVERRIDE. Suggest j
Yoyo Zhou
2014/07/18 02:45:30
Done.
|
+#include "extensions/browser/api/dns/dns_api.h" |
+#include "extensions/browser/api/dns/host_resolver_wrapper.h" |
+#include "extensions/browser/api/dns/mock_host_resolver_creator.h" |
+#include "extensions/browser/api_test_utils.h" |
+#include "extensions/browser/extension_function_dispatcher.h" |
+#include "extensions/common/extension.h" |
+#include "extensions/common/extension_builder.h" |
+#include "net/base/net_errors.h" |
+#include "net/dns/mock_host_resolver.h" |
mmenke
2014/07/16 14:05:54
Are you actual depending on this? Looks like you
Yoyo Zhou
2014/07/18 02:45:30
It's used in line 42.
mmenke
2014/07/18 21:32:30
Are you sure? resolver_creator_->CreateMockHostRe
Yoyo Zhou
2014/07/22 00:10:30
It turns out this is because the header was needed
|
+ |
+using extensions::ExtensionFunctionDispatcher; |
+using extensions::api_test_utils::RunFunctionAndReturnSingleResult; |
+ |
+namespace { |
+ |
+class TestFunctionDispatcherDelegate |
+ : public ExtensionFunctionDispatcher::Delegate { |
+ public: |
+ TestFunctionDispatcherDelegate() {} |
+ virtual ~TestFunctionDispatcherDelegate() {} |
+ |
+ // NULL implementation |
+ private: |
mmenke
2014/07/16 14:05:54
DISALLOW_COPY_AND_ASSIGN (And include the requisit
Yoyo Zhou
2014/07/18 02:45:30
Done.
|
+}; |
+} |
mmenke
2014/07/16 14:05:54
nit: Linebreak before ending namespace.
mmenke
2014/07/16 14:05:54
nit: "} // namespace"
Yoyo Zhou
2014/07/18 02:45:30
Done.
|
+ |
+class DnsApiTest : public apps::AppShellTest { |
+ public: |
+ DnsApiTest() : resolver_creator_(new extensions::MockHostResolverCreator()) {} |
+ |
+ private: |
+ virtual void SetUpOnMainThread() OVERRIDE { |
+ apps::AppShellTest::SetUpOnMainThread(); |
+ extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting( |
+ resolver_creator_->CreateMockHostResolver()); |
+ } |
+ |
+ virtual void TearDownOnMainThread() OVERRIDE { |
+ extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting( |
+ NULL); |
+ resolver_creator_->DeleteMockHostResolver(); |
+ apps::AppShellTest::TearDownOnMainThread(); |
+ } |
+ |
+ // 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_; |
+}; |
+ |
+IN_PROC_BROWSER_TEST_F(DnsApiTest, DnsResolveIPLiteral) { |
+ scoped_refptr<extensions::DnsResolveFunction> resolve_function( |
+ new extensions::DnsResolveFunction()); |
+ scoped_refptr<extensions::Extension> empty_extension( |
+ extensions::ExtensionBuilder() |
+ .SetManifest(extensions::DictionaryBuilder().Set("name", "Test").Set( |
+ "version", "1.0")) |
+ .Build()); |
+ |
+ resolve_function->set_extension(empty_extension.get()); |
James Cook
2014/07/16 17:15:31
The old code has set_has_callback(true) after this
Yoyo Zhou
2014/07/18 02:45:30
No. Don't know why it works without it, though. I'
|
+ |
+ TestFunctionDispatcherDelegate delegate; |
Yoyo Zhou
2014/07/16 02:30:25
Some of this setup (like here) is more difficult t
|
+ scoped_ptr<ExtensionFunctionDispatcher> dispatcher( |
+ new ExtensionFunctionDispatcher(browser_context(), &delegate)); |
+ |
+ scoped_ptr<base::Value> result( |
+ RunFunctionAndReturnSingleResult(resolve_function.get(), |
+ "[\"127.0.0.1\"]", |
+ browser_context(), |
+ dispatcher.Pass())); |
+ ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType()); |
+ base::DictionaryValue* value = |
+ static_cast<base::DictionaryValue*>(result.get()); |
mmenke
2014/07/16 14:05:54
No need for a cast, just use:
base::DictionaryVal
Yoyo Zhou
2014/07/18 02:45:31
Done.
|
+ |
+ int resultCode; |
mmenke
2014/07/16 14:05:54
nit: result_code.
Yoyo Zhou
2014/07/18 02:45:30
Done.
|
+ EXPECT_TRUE(value->GetInteger("resultCode", &resultCode)); |
+ EXPECT_EQ(net::OK, resultCode); |
mmenke
2014/07/16 14:05:54
We're exposing Chrome network error codes to exten
Yoyo Zhou
2014/07/18 02:45:30
Yes, in lots of places already; see https://code.g
mmenke
2014/07/18 21:32:30
I'm not really concerned about more stuff dependin
Yoyo Zhou
2014/07/22 00:10:30
This happens in other APIs too, I think, e.g. http
|
+ |
+ 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( |
+ extensions::ExtensionBuilder() |
+ .SetManifest(extensions::DictionaryBuilder().Set("name", "Test").Set( |
+ "version", "1.0")) |
+ .Build()); |
+ |
+ resolve_function->set_extension(empty_extension.get()); |
+ resolve_function->set_has_callback(true); |
+ |
+ TestFunctionDispatcherDelegate delegate; |
+ scoped_ptr<ExtensionFunctionDispatcher> dispatcher( |
+ new ExtensionFunctionDispatcher(browser_context(), &delegate)); |
+ |
+ std::string function_arguments("[\""); |
+ function_arguments += extensions::MockHostResolverCreator::kHostname; |
+ function_arguments += "\"]"; |
+ scoped_ptr<base::Value> result( |
+ RunFunctionAndReturnSingleResult(resolve_function.get(), |
+ function_arguments, |
+ browser_context(), |
+ dispatcher.Pass())); |
+ ASSERT_EQ(base::Value::TYPE_DICTIONARY, result->GetType()); |
+ base::DictionaryValue* value = |
+ static_cast<base::DictionaryValue*>(result.get()); |
James Cook
2014/07/16 17:15:31
ditto
Yoyo Zhou
2014/07/18 02:45:30
Done.
|
+ |
+ int resultCode; |
James Cook
2014/07/16 17:15:31
ditto
Yoyo Zhou
2014/07/18 02:45:30
Done.
|
+ 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); |
+} |