1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "components/translate/core/browser/translate_script.h"
6
7 #include "base/bind.h"
8 #include "base/command_line.h"
9 #include "base/strings/stringprintf.h"
10 #include "components/translate/core/browser/translate_download_manager.h"
11 #include "components/translate/core/common/translate_switches.h"
12 #include "net/base/load_flags.h"
13 #include "net/base/url_util.h"
14 #include "net/http/http_request_headers.h"
15 #include "net/url_request/test_url_fetcher_factory.h"
16 #include "testing/gtest/include/gtest/gtest.h"
17 #include "url/gurl.h"
18
19 class TranslateScriptTest : public testing::Test {
20 public:
TranslateScriptTest()21 TranslateScriptTest() : testing::Test() {}
22
23 protected:
SetUp()24 virtual void SetUp() {
25 script_.reset(new TranslateScript);
26 DCHECK(script_.get());
27 TranslateDownloadManager::GetInstance()->set_application_locale("en");
28 }
29
TearDown()30 virtual void TearDown() {
31 script_.reset();
32 }
33
Request()34 void Request() {
35 script_->Request(
36 base::Bind(&TranslateScriptTest::OnComplete, base::Unretained(this)));
37 }
38
GetTestURLFetcher()39 net::TestURLFetcher* GetTestURLFetcher() {
40 return url_fetcher_factory_.GetFetcherByID(TranslateScript::kFetcherId);
41 }
42
43 private:
OnComplete(bool success,const std::string & script)44 void OnComplete(bool success, const std::string& script) {
45 }
46
47 scoped_ptr<TranslateScript> script_;
48 net::TestURLFetcherFactory url_fetcher_factory_;
49
50 DISALLOW_COPY_AND_ASSIGN(TranslateScriptTest);
51 };
52
TEST_F(TranslateScriptTest,CheckScriptParameters)53 TEST_F(TranslateScriptTest, CheckScriptParameters) {
54 Request();
55 net::TestURLFetcher* fetcher = GetTestURLFetcher();
56 ASSERT_TRUE(fetcher);
57
58 GURL expected_url(TranslateScript::kScriptURL);
59 GURL url = fetcher->GetOriginalURL();
60 EXPECT_TRUE(url.is_valid());
61 EXPECT_EQ(expected_url.GetOrigin().spec(), url.GetOrigin().spec());
62 EXPECT_EQ(expected_url.path(), url.path());
63
64 int load_flags = fetcher->GetLoadFlags();
65 EXPECT_EQ(net::LOAD_DO_NOT_SEND_COOKIES,
66 load_flags & net::LOAD_DO_NOT_SEND_COOKIES);
67 EXPECT_EQ(net::LOAD_DO_NOT_SAVE_COOKIES,
68 load_flags & net::LOAD_DO_NOT_SAVE_COOKIES);
69
70 std::string expected_extra_headers =
71 base::StringPrintf("%s\r\n\r\n", TranslateScript::kRequestHeader);
72 net::HttpRequestHeaders extra_headers;
73 fetcher->GetExtraRequestHeaders(&extra_headers);
74 EXPECT_EQ(expected_extra_headers, extra_headers.ToString());
75
76 std::string always_use_ssl;
77 net::GetValueForKeyInQuery(
78 url, TranslateScript::kAlwaysUseSslQueryName, &always_use_ssl);
79 EXPECT_EQ(std::string(TranslateScript::kAlwaysUseSslQueryValue),
80 always_use_ssl);
81
82 std::string callback;
83 net::GetValueForKeyInQuery(
84 url, TranslateScript::kCallbackQueryName, &callback);
85 EXPECT_EQ(std::string(TranslateScript::kCallbackQueryValue), callback);
86
87 #if !defined(OS_IOS)
88 // iOS does not have specific loaders for the isolated world.
89 std::string css_loader_callback;
90 net::GetValueForKeyInQuery(
91 url, TranslateScript::kCssLoaderCallbackQueryName, &css_loader_callback);
92 EXPECT_EQ(std::string(TranslateScript::kCssLoaderCallbackQueryValue),
93 css_loader_callback);
94
95 std::string javascript_loader_callback;
96 net::GetValueForKeyInQuery(
97 url,
98 TranslateScript::kJavascriptLoaderCallbackQueryName,
99 &javascript_loader_callback);
100 EXPECT_EQ(std::string(TranslateScript::kJavascriptLoaderCallbackQueryValue),
101 javascript_loader_callback);
102 #endif // !defined(OS_IOS)
103 }
104
TEST_F(TranslateScriptTest,CheckScriptURL)105 TEST_F(TranslateScriptTest, CheckScriptURL) {
106 const std::string script_url("http://www.tamurayukari.com/mero-n.js");
107 CommandLine* command_line = CommandLine::ForCurrentProcess();
108 command_line->AppendSwitchASCII(translate::switches::kTranslateScriptURL,
109 script_url);
110
111 Request();
112 net::TestURLFetcher* fetcher = GetTestURLFetcher();
113 ASSERT_TRUE(fetcher);
114
115 GURL expected_url(script_url);
116 GURL url = fetcher->GetOriginalURL();
117 EXPECT_TRUE(url.is_valid());
118 EXPECT_EQ(expected_url.GetOrigin().spec(), url.GetOrigin().spec());
119 EXPECT_EQ(expected_url.path(), url.path());
120 }
121