IE CEEE: Explicitly assign window properties/methods into the script host glo...

ceee/ie/plugin/scripting/content_script_manager_unittest.cc

 // Copyright (c) 2010 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.
 //
 // Content script manager implementation unit tests.
 #include "ceee/ie/plugin/scripting/content_script_manager.h"
 
 #include "base/logging.h"
 #include "base/utf_string_conversions.h"
 #include "ceee/common/initializing_coclass.h"
@@ skipping 205 matching lines @@
 
     function_ = NULL;
     function_keeper_.Release();
 
     // Test for leakage.
     EXPECT_EQ(0, testing::InstanceCountMixinBase::all_instance_count());
   }
 
   // Set up to expect scripting initialization.
   void ExpectScriptInitialization() {
-    EXPECT_CALL(*script_host_,
-        RegisterScriptObject(StrEq(L"window"), _, true))
-            .WillOnce(Return(S_OK));
-
-    EXPECT_CALL(*script_host_,
-        RunScript(testing::StartsWith(L"ceee-content://"), _))
-        .Times(5)
-            .WillRepeatedly(Return(S_OK));
-
-    // Return the mock function for start/end init.
-    EXPECT_CALL(*script_host_, RunExpression(StrEq(L"ceee.startInit_"), _))
-        .WillOnce(
-            DoAll(
-                CopyVariantToArgument<1>(CComVariant(function_keeper_)),
-                Return(S_OK)));
-    EXPECT_CALL(*script_host_, RunExpression(StrEq(L"ceee.endInit_"), _))
-        .WillOnce(
-            DoAll(
-                CopyVariantToArgument<1>(CComVariant(function_keeper_)),
-                Return(S_OK)));
-
     EXPECT_CALL(*frame_host_, GetExtensionId(_)).WillOnce(DoAll(
         SetArgumentPointee<0>(std::wstring(kExtensionId)),
         Return(S_OK)));
 
-    // And expect two invocations.
+    EXPECT_CALL(*script_host_,
+        RunScript(testing::StartsWith(L"ceee-content://"), _))
+        .Times(5).WillRepeatedly(Return(S_OK));
+
+    // Return the mock function for start/end init.
+    EXPECT_CALL(*script_host_, RunExpression(StrEq(L"ceee.startInit_"), _))
+        .WillOnce(DoAll(
+            CopyVariantToArgument<1>(CComVariant(function_keeper_)),
+            Return(S_OK)));
+    EXPECT_CALL(*script_host_, RunExpression(StrEq(L"ceee.endInit_"), _))
+        .WillOnce(DoAll(
+            CopyVariantToArgument<1>(CComVariant(function_keeper_)),
+            Return(S_OK)));
+
+    // Register the window object and initialize its globals.
+    EXPECT_CALL(*script_host_,
+        RegisterScriptObject(StrEq(L"window"), _, false))
+            .WillOnce(Return(S_OK));
+    EXPECT_CALL(*script_host_, RunExpression(StrEq(L"ceee.initGlobals_"), _))
+        .WillOnce(DoAll(
+            CopyVariantToArgument<1>(CComVariant(function_keeper_)),
+            Return(S_OK)));
+
+    // And expect three invocations.
     // TODO(siggi@chromium.org): be more specific?
     EXPECT_CALL(*function_, Invoke(_, _, _, _, _, _, _, _))
-        .Times(2)
-            .WillRepeatedly(Return(S_OK));
+        .Times(3).WillRepeatedly(Return(S_OK));
   }
 
   void ExpectCreateScriptHost(TestingContentScriptManager* manager) {
     EXPECT_CALL(*manager, CreateScriptHost(_))
         .WillOnce(
             DoAll(
                 CopyInterfaceToArgument<0>(script_host_keeper_),
                 Return(S_OK)));
   }
 
@@ skipping 214 matching lines @@
 
   EXPECT_CALL(manager,
       InjectStyleTag(document_, head_node.p, StrEq(kCssContent)))
               .WillOnce(Return(S_OK));
 
   ASSERT_HRESULT_SUCCEEDED(
         manager.ContentScriptManager::InsertCss(kCssContent, document_));
 }
 
 }  // namespace