10529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch// Copyright 2014 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 59ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 6f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "extensions/common/extension.h" 73551c9c881056c480085172ff9840cab31610854Torne (Richard Coles)#include "extensions/common/features/feature.h" 80529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch#include "extensions/renderer/script_context.h" 90529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch#include "extensions/renderer/script_context_set.h" 10116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "gin/public/context_holder.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 127d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "third_party/WebKit/public/web/WebFrame.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "v8/include/v8.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 170529e5d033099cbfc42635f6f6183833b09dff6eBen MurdochTEST(ScriptContextSet, Lifecycle) { 18a93a17c8d99d686bd4a1511e5504e5e6cc9fcadfTorne (Richard Coles) base::MessageLoop loop; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 200529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch ScriptContextSet context_set; 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) v8::Isolate* isolate = v8::Isolate::GetCurrent(); 2358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) v8::HandleScope handle_scope(isolate); 24116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch gin::ContextHolder context_holder(isolate); 25116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch context_holder.SetContext(v8::Context::New(isolate)); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Dirty hack, but we don't actually need the frame, and this is easier than 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // creating a whole webview. 29f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) blink::WebFrame* frame = reinterpret_cast<blink::WebFrame*>(1); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const Extension* extension = NULL; 31116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch ScriptContext* context = 32116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch new ScriptContext(context_holder.context(), 33116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch frame, 34116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch extension, 351320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci Feature::BLESSED_EXTENSION_CONTEXT, 361320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci extension, 37116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch Feature::BLESSED_EXTENSION_CONTEXT); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context_set.Add(context); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, context_set.GetAll().count(context)); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(context, context_set.GetByV8Context(context->v8_context())); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Adding the same item multiple times should be OK and deduped. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context_set.Add(context); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, context_set.GetAll().count(context)); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // GetAll() returns a copy so removing from one should not remove from others. 480529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch ScriptContextSet::ContextSet set_copy = context_set.GetAll(); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, set_copy.count(context)); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context_set.Remove(context); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, context_set.size()); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(context_set.GetByV8Context(context->v8_context())); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, set_copy.size()); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // After removal, the context should be marked for destruction. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(context->web_frame()); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Run loop to do the actual deletion. 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) loop.RunUntilIdle(); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 64