gview_request_interceptor_unittest.cc revision dc0f95d653279beabeb9817299e2902918ba123e 
1// Copyright (c) 2011 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 <string>
6
7#include "base/message_loop.h"
8#include "chrome/browser/chromeos/gview_request_interceptor.h"
9#include "chrome/common/chrome_paths.h"
10#include "content/browser/plugin_service.h"
11#include "net/base/load_flags.h"
12#include "net/url_request/url_request.h"
13#include "net/url_request/url_request_job.h"
14#include "net/url_request/url_request_test_job.h"
15#include "net/url_request/url_request_test_util.h"
16#include "testing/gtest/include/gtest/gtest.h"
17#include "webkit/glue/plugins/plugin_list.h"
18
19namespace chromeos {
20
21class GViewURLRequestTestJob : public net::URLRequestTestJob {
22 public:
23  explicit GViewURLRequestTestJob(net::URLRequest* request)
24      : net::URLRequestTestJob(request, true) {
25  }
26
27  virtual bool GetMimeType(std::string* mime_type) const {
28    // During the course of a single test, two URLRequestJobs are
29    // created -- the first is for the viewable document URL, and the
30    // second is for the rediected URL.  In order to test the
31    // interceptor, the mime type of the first request must be one of
32    // the supported viewable mime types.  So when the net::URLRequestJob
33    // is a request for one of the test URLs that point to viewable
34    // content, return an appropraite mime type.  Otherwise, return
35    // "text/html".
36    if (request_->url() == GURL("http://foo.com/file.pdf")) {
37      *mime_type = "application/pdf";
38    } else if (request_->url() == GURL("http://foo.com/file.ppt")) {
39      *mime_type = "application/vnd.ms-powerpoint";
40    } else {
41      *mime_type = "text/html";
42    }
43    return true;
44  }
45
46 private:
47  ~GViewURLRequestTestJob() {}
48};
49
50class GViewRequestInterceptorTest : public testing::Test {
51 public:
52  virtual void SetUp() {
53    net::URLRequest::RegisterProtocolFactory("http",
54                                        &GViewRequestInterceptorTest::Factory);
55    interceptor_ = GViewRequestInterceptor::GetInstance();
56    ASSERT_TRUE(PathService::Get(chrome::FILE_PDF_PLUGIN, &pdf_path_));
57  }
58
59  virtual void TearDown() {
60    net::URLRequest::RegisterProtocolFactory("http", NULL);
61    message_loop_.RunAllPending();
62  }
63
64  static net::URLRequestJob* Factory(net::URLRequest* request,
65                                     const std::string& scheme) {
66    return new GViewURLRequestTestJob(request);
67  }
68
69  void RegisterPDFPlugin() {
70    webkit::npapi::WebPluginInfo info;
71    info.path = pdf_path_;
72    info.enabled = webkit::npapi::WebPluginInfo::USER_ENABLED_POLICY_UNMANAGED;
73    webkit::npapi::PluginList::Singleton()->RegisterInternalPlugin(info);
74    webkit::npapi::PluginList::Singleton()->RefreshPlugins();
75  }
76
77  void UnregisterPDFPlugin() {
78    webkit::npapi::PluginList::Singleton()->UnregisterInternalPlugin(pdf_path_);
79    webkit::npapi::PluginList::Singleton()->RefreshPlugins();
80  }
81
82  void SetPDFPluginLoadedState(bool want_loaded, bool* out_is_enabled) {
83    webkit::npapi::WebPluginInfo info;
84    bool is_loaded =
85        webkit::npapi::PluginList::Singleton()->GetPluginInfoByPath(
86            pdf_path_, &info);
87    if (is_loaded && !want_loaded) {
88      UnregisterPDFPlugin();
89      is_loaded = webkit::npapi::PluginList::Singleton()->GetPluginInfoByPath(
90          pdf_path_, &info);
91    } else if (!is_loaded && want_loaded) {
92      // This "loads" the plug-in even if it's not present on the
93      // system - which is OK since we don't actually use it, just
94      // need it to be "enabled" for the test.
95      RegisterPDFPlugin();
96      is_loaded = webkit::npapi::PluginList::Singleton()->GetPluginInfoByPath(
97          pdf_path_, &info);
98    }
99    EXPECT_EQ(want_loaded, is_loaded);
100    *out_is_enabled = webkit::npapi::IsPluginEnabled(info);
101  }
102
103 protected:
104  MessageLoopForIO message_loop_;
105  TestDelegate test_delegate_;
106  net::URLRequest::Interceptor* interceptor_;
107  FilePath pdf_path_;
108};
109
110TEST_F(GViewRequestInterceptorTest, DoNotInterceptHtml) {
111  net::URLRequest request(GURL("http://foo.com/index.html"), &test_delegate_);
112  request.Start();
113  MessageLoop::current()->Run();
114  EXPECT_EQ(0, test_delegate_.received_redirect_count());
115  EXPECT_EQ(GURL("http://foo.com/index.html"), request.url());
116}
117
118TEST_F(GViewRequestInterceptorTest, DoNotInterceptDownload) {
119  net::URLRequest request(GURL("http://foo.com/file.pdf"), &test_delegate_);
120  request.set_load_flags(net::LOAD_IS_DOWNLOAD);
121  request.Start();
122  MessageLoop::current()->Run();
123  EXPECT_EQ(0, test_delegate_.received_redirect_count());
124  EXPECT_EQ(GURL("http://foo.com/file.pdf"), request.url());
125}
126
127TEST_F(GViewRequestInterceptorTest, DoNotInterceptPdfWhenEnabled) {
128  bool enabled;
129  SetPDFPluginLoadedState(true, &enabled);
130
131  if (!enabled) {
132    bool pdf_plugin_enabled =
133        webkit::npapi::PluginList::Singleton()->EnablePlugin(pdf_path_);
134    EXPECT_TRUE(pdf_plugin_enabled);
135  }
136
137  net::URLRequest request(GURL("http://foo.com/file.pdf"), &test_delegate_);
138  request.Start();
139  MessageLoop::current()->Run();
140  EXPECT_EQ(0, test_delegate_.received_redirect_count());
141  EXPECT_EQ(GURL("http://foo.com/file.pdf"), request.url());
142}
143
144TEST_F(GViewRequestInterceptorTest, InterceptPdfWhenDisabled) {
145  bool enabled;
146  SetPDFPluginLoadedState(true, &enabled);
147
148  if (enabled) {
149    bool pdf_plugin_disabled =
150        webkit::npapi::PluginList::Singleton()->DisablePlugin(pdf_path_);
151    EXPECT_TRUE(pdf_plugin_disabled);
152  }
153
154  net::URLRequest request(GURL("http://foo.com/file.pdf"), &test_delegate_);
155  request.Start();
156  MessageLoop::current()->Run();
157  EXPECT_EQ(1, test_delegate_.received_redirect_count());
158  EXPECT_EQ(
159      GURL("http://docs.google.com/gview?url=http%3A//foo.com/file.pdf"),
160      request.url());
161}
162
163TEST_F(GViewRequestInterceptorTest, InterceptPdfWithNoPlugin) {
164  bool enabled;
165  SetPDFPluginLoadedState(false, &enabled);
166
167  net::URLRequest request(GURL("http://foo.com/file.pdf"), &test_delegate_);
168  request.Start();
169  MessageLoop::current()->Run();
170  EXPECT_EQ(1, test_delegate_.received_redirect_count());
171  EXPECT_EQ(GURL("http://docs.google.com/gview?url=http%3A//foo.com/file.pdf"),
172                 request.url());
173}
174
175TEST_F(GViewRequestInterceptorTest, InterceptPowerpoint) {
176  net::URLRequest request(GURL("http://foo.com/file.ppt"), &test_delegate_);
177  request.Start();
178  MessageLoop::current()->Run();
179  EXPECT_EQ(1, test_delegate_.received_redirect_count());
180  EXPECT_EQ(GURL("http://docs.google.com/gview?url=http%3A//foo.com/file.ppt"),
181                 request.url());
182}
183
184}  // namespace chromeos
185