Adding parsing for partition and persistence attributes to the <browser> tag

content/renderer/browser_plugin/browser_plugin_browsertest.cc

 // Copyright (c) 2012 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 "content/renderer/browser_plugin/browser_plugin_browsertest.h"
 
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "content/common/browser_plugin_messages.h"
@@ skipping 12 matching lines @@
 namespace {
 const char kHTMLForBrowserPluginObject[] =
   "<object id='browserplugin' width='640px' height='480px'"
   "  src='foo' type='%s'>";
 
 std::string GetHTMLForBrowserPluginObject() {
   return StringPrintf(kHTMLForBrowserPluginObject,
                       content::kBrowserPluginNewMimeType);
 }
 
+const char kHTMLForPartitionedPluginObject[] =
+  "<object id='browserplugin' width='640px' height='480px'"
+  "  src='foo' type='%s' partition='someid'>";
+
+const char kHTMLForPartitionedPersistedPluginObject[] =
+  "<object id='browserplugin' width='640px' height='480px'"
+  "  src='foo' type='%s' partition='someid' persist>";
+
 }
 
 namespace content {
 
 BrowserPluginTest::BrowserPluginTest() {}
 
 BrowserPluginTest::~BrowserPluginTest() {}
 
 void BrowserPluginTest::SetUp() {
   GetContentClient()->set_renderer_for_testing(&content_renderer_client_);
@@ skipping 14 matching lines @@
   if (value.IsEmpty() || !value->IsString())
     return std::string();
 
   v8::Local<v8::String> v8_str = value->ToString();
   int length = v8_str->Utf8Length() + 1;
   scoped_array<char> str(new char[length]);
   v8_str->WriteUtf8(str.get(), length);
   return str.get();
 }
 
+bool BrowserPluginTest::ExecuteScriptAndReturnBool(
+    const std::string& script,
+    bool& result) {
+  v8::Handle<v8::Value>  value = GetMainFrame()->executeScriptAndReturnValue(
+      WebKit::WebScriptSource(WebKit::WebString::fromUTF8(script.c_str())));
+  if (value.IsEmpty() || !value->IsBoolean())
+    return false;
+
+  result = value->BooleanValue();
+  return true;
+}
+
 // This test verifies that an initial resize occurs when we instantiate the
 // browser plugin. This test also verifies that the browser plugin is waiting
 // for a BrowserPluginMsg_UpdateRect in response. We issue an UpdateRect, and
 // we observe an UpdateRect_ACK, with the resize_pending_ reset, indiciating
 // that the BrowserPlugin is not waiting for any more UpdateRects to
 // satisfy its resize request.
 TEST_F(BrowserPluginTest, InitialResize) {
   LoadHTML(GetHTMLForBrowserPluginObject().c_str());
   // Verify that the information based on ResizeGuest is correct, and
   // use its TransportDIB::Id to paint.
@@ skipping 235 matching lines @@
   browser_plugin->DidNavigate(GURL(kGoogleURL));
   EXPECT_EQ(kGoogleURL, ExecuteScriptAndReturnString("url"));
 
   ExecuteJavaScript(kRemoveEventListener);
   browser_plugin->DidNavigate(GURL(kGoogleNewsURL));
   // The URL variable should not change because we've removed the event
   // listener.
   EXPECT_EQ(kGoogleURL, ExecuteScriptAndReturnString("url"));
 }
 
+// Verify that the 'partition' attribute on the browser plugin is parsed
+// correctly.
+TEST_F(BrowserPluginTest, PartitionAttribute) {
+  std::string html = StringPrintf(kHTMLForPartitionedPluginObject,
+                                  content::kBrowserPluginNewMimeType);
+  LoadHTML(html.c_str());
+
+  std::string partition_value = ExecuteScriptAndReturnString(
+      "document.getElementById('browserplugin').partition");
+  EXPECT_EQ("someid", partition_value);
+
+  // Ensure that setting the partition attribute, once it has been set, has
+  // no effect.
+  ExecuteJavaScript(
+      "document.getElementById('browserplugin').partition = 'bar'");
+
+  partition_value = ExecuteScriptAndReturnString(
+      "document.getElementById('browserplugin').partition");
+  EXPECT_EQ("someid", partition_value);
+}
+
+// Verify that the 'persist' attribute on the browser plugin is parsed
+// correctly.
+TEST_F(BrowserPluginTest, PersistAttribute) {
+  std::string html = StringPrintf(kHTMLForPartitionedPersistedPluginObject,
+                                  content::kBrowserPluginNewMimeType);
+  LoadHTML(html.c_str());
+
+  bool persist_value = false;
+  // Check that the value has been properly parsed.
+  EXPECT_TRUE(ExecuteScriptAndReturnBool(
+      "document.getElementById('browserplugin').persist", persist_value));
+  EXPECT_TRUE(persist_value);
+
+  // Ensure that setting the value, once it is initialized, has no effect.
+  ExecuteJavaScript(
+      "document.getElementById('browserplugin').persist = false");
+  EXPECT_TRUE(ExecuteScriptAndReturnBool(
+      "document.getElementById('browserplugin').persist", persist_value));
+  EXPECT_TRUE(persist_value);
+
+  // Verify that missing persist attribute defaults to false.
+  html = StringPrintf(kHTMLForPartitionedPluginObject,
+                      content::kBrowserPluginNewMimeType);
+  LoadHTML(html.c_str());
+
+  EXPECT_TRUE(ExecuteScriptAndReturnBool(
+      "document.getElementById('browserplugin').persist", persist_value));
+  EXPECT_FALSE(persist_value);
+}
+
+// Test to verify that after the first navigation, the partition and persist
+// properties cannot be modified.
+TEST_F(BrowserPluginTest, ImmutableAttributesAfterNavigation) {
+  const char kHtml[] =
+    "<object id='browserplugin' width='640px' height='480px' type='%s'>";
+  std::string html = StringPrintf(kHtml, content::kBrowserPluginNewMimeType);
+  LoadHTML(html.c_str());
+
+  ExecuteJavaScript(
+      "document.getElementById('browserplugin').partition = 'storage'");
+  std::string partition_value = ExecuteScriptAndReturnString(
+      "document.getElementById('browserplugin').partition");
+  EXPECT_STREQ("storage", partition_value.c_str());
+
+  std::string src_value = ExecuteScriptAndReturnString(
+      "document.getElementById('browserplugin').src");
+  EXPECT_STREQ("", src_value.c_str());
+
+  ExecuteJavaScript("document.getElementById('browserplugin').src = 'bar'");
+  {
+    const IPC::Message* msg =
+        browser_plugin_manager()->sink().GetUniqueMessageMatching(
+            BrowserPluginHostMsg_NavigateOrCreateGuest::ID);
+    ASSERT_TRUE(msg);
+
+    int instance_id;
+    long long frame_id;
+    std::string src;
+    BrowserPluginHostMsg_NavigateOrCreateGuest::Read(
+        msg,
+        &instance_id,
+        &frame_id,
+        &src);
+    EXPECT_STREQ("bar", src.c_str());
+  }
+
+  ExecuteJavaScript(
+      "document.getElementById('browserplugin').partition = 'someid'");
+  partition_value = ExecuteScriptAndReturnString(
+      "document.getElementById('browserplugin').partition");
+  EXPECT_STREQ("storage", partition_value.c_str());
+
+  bool persist_value = false;
+  ExecuteJavaScript(
+      "document.getElementById('browserplugin').persist = true");
+  EXPECT_TRUE(ExecuteScriptAndReturnBool(
+      "document.getElementById('browserplugin').persist", persist_value));
+  EXPECT_FALSE(persist_value);
+}
+
 }  // namespace content