Chromium Code Reviews| 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); |
| +} |