Implement unlimitedStorage content capability

chrome/browser/extensions/content_capabilities_browsertest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <string>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/files/file_util.h"
 #include "base/path_service.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/test_extension_dir.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/crx_file/id_util.h"
 #include "content/public/test/browser_test_utils.h"
 #include "extensions/common/extension_builder.h"
 #include "extensions/common/manifest_handlers/content_capabilities_handler.h"
 #include "extensions/common/switches.h"
 #include "extensions/common/url_pattern.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
+#include "storage/browser/quota/special_storage_policy.h"
 
 using extensions::DictionaryBuilder;
 using extensions::Extension;
 using extensions::ExtensionBuilder;
 using extensions::ListBuilder;
 
 class ContentCapabilitiesTest : public ExtensionApiTest {
  protected:
   void SetUpCommandLine(base::CommandLine* command_line) override {
     ExtensionApiTest::SetUpCommandLine(command_line);
@@ skipping 59 matching lines @@
     embedded_test_server()->ServeFilesFromDirectory(
         test_data.AppendASCII("extensions/content_capabilities"));
     ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
     host_resolver()->AddRule("*", embedded_test_server()->base_url().host());
   }
 
   // Run some script in the context of the given origin and in the presence of
   // the given extension. This is used to wrap calls into the JS test functions
   // defined by
   // $(DIR_TEST_DATA)/extensions/content_capabilities/capability_tests.js.
-  bool TestContentCapability(const Extension* extension,
-                             const char* origin,
-                             const char* code) {
+  testing::AssertionResult TestScriptResult(const Extension* extension,
+                                            const char* origin,
+                                            const char* code) {
     ui_test_utils::NavigateToURL(browser(), GetTestURLFor(origin));
     bool result = false;
-    CHECK(content::ExecuteScriptAndExtractBool(web_contents(), code, &result));
-    return result;
+    if (!content::ExecuteScriptAndExtractBool(web_contents(), code, &result))
+      return testing::AssertionFailure() << "Could not execute test script.";
+    if (!result)
+      return testing::AssertionFailure();
+    return testing::AssertionSuccess();
   }
 
-  bool CanReadClipboard(const Extension* extension, const char* origin) {
-    return TestContentCapability(extension, origin, "tests.canReadClipboard()");
+  testing::AssertionResult CanReadClipboard(const Extension* extension,
+                                            const char* origin) {
+    return TestScriptResult(extension, origin, "tests.canReadClipboard()");
   }
 
-  bool CanWriteClipboard(const Extension* extension, const char* origin) {
-    return TestContentCapability(extension, origin,
-                                 "tests.canWriteClipboard()");
+  testing::AssertionResult CanWriteClipboard(const Extension* extension,
+                                             const char* origin) {
+    return TestScriptResult(extension, origin, "tests.canWriteClipboard()");
+  }
+
+  testing::AssertionResult HasUnlimitedStorage(const Extension* extension,
+                                               const char* origin) {
+    if (profile()->GetSpecialStoragePolicy()->IsStorageUnlimited(
+        GetTestURLFor(origin)))
+      return testing::AssertionSuccess();
+    return testing::AssertionFailure();
   }
 
  private:
   extensions::TestExtensionDir test_extension_dir_;
 };
 
 IN_PROC_BROWSER_TEST_F(ContentCapabilitiesTest, NoCapabilities) {
   InitializeTestServer();
   scoped_refptr<const Extension> extension = LoadExtensionWithCapabilities(
       MakeJSONList("http://foo.example.com/*"), MakeJSONList());
   EXPECT_FALSE(CanReadClipboard(extension.get(), "foo.example.com"));
   EXPECT_FALSE(CanWriteClipboard(extension.get(), "foo.example.com"));
+  EXPECT_FALSE(HasUnlimitedStorage(extension.get(), "foo.example.com"));
 }
 
 IN_PROC_BROWSER_TEST_F(ContentCapabilitiesTest, ClipboardRead) {
   InitializeTestServer();
   scoped_refptr<const Extension> extension = LoadExtensionWithCapabilities(
       MakeJSONList("http://foo.example.com/*"), MakeJSONList("clipboardRead"));
   EXPECT_TRUE(CanReadClipboard(extension.get(), "foo.example.com"));
   EXPECT_FALSE(CanReadClipboard(extension.get(), "bar.example.com"));
   EXPECT_FALSE(CanWriteClipboard(extension.get(), "foo.example.com"));
 }
@@ skipping 10 matching lines @@
 IN_PROC_BROWSER_TEST_F(ContentCapabilitiesTest, ClipboardReadWrite) {
   InitializeTestServer();
   scoped_refptr<const Extension> extension = LoadExtensionWithCapabilities(
       MakeJSONList("http://foo.example.com/*"),
       MakeJSONList("clipboardRead", "clipboardWrite"));
   EXPECT_TRUE(CanReadClipboard(extension.get(), "foo.example.com"));
   EXPECT_TRUE(CanWriteClipboard(extension.get(), "foo.example.com"));
   EXPECT_FALSE(CanReadClipboard(extension.get(), "bar.example.com"));
   EXPECT_FALSE(CanWriteClipboard(extension.get(), "bar.example.com"));
 }
+
+IN_PROC_BROWSER_TEST_F(ContentCapabilitiesTest, UnlimitedStorage) {
+  InitializeTestServer();
+  scoped_refptr<const Extension> extension = LoadExtensionWithCapabilities(
+      MakeJSONList("http://foo.example.com/*"),
+      MakeJSONList("unlimitedStorage"));
+  EXPECT_TRUE(HasUnlimitedStorage(extension.get(), "foo.example.com"));
+  EXPECT_FALSE(HasUnlimitedStorage(extension.get(), "bar.example.com"));
+}