1// Copyright 2014 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 "base/message_loop/message_loop.h" 6#include "extensions/common/extension.h" 7#include "extensions/common/features/feature.h" 8#include "extensions/renderer/script_context.h" 9#include "extensions/renderer/script_context_set.h" 10#include "testing/gtest/include/gtest/gtest.h" 11#include "third_party/WebKit/public/web/WebFrame.h" 12#include "v8/include/v8.h" 13 14namespace extensions { 15 16TEST(ScriptContextSet, Lifecycle) { 17 base::MessageLoop loop; 18 19 ScriptContextSet context_set; 20 21 v8::Isolate* isolate = v8::Isolate::GetCurrent(); 22 v8::HandleScope handle_scope(isolate); 23 v8::Handle<v8::Context> v8_context(v8::Context::New(isolate)); 24 25 // Dirty hack, but we don't actually need the frame, and this is easier than 26 // creating a whole webview. 27 blink::WebFrame* frame = reinterpret_cast<blink::WebFrame*>(1); 28 const Extension* extension = NULL; 29 ScriptContext* context = new ScriptContext( 30 v8_context, frame, extension, Feature::BLESSED_EXTENSION_CONTEXT); 31 32 context_set.Add(context); 33 EXPECT_EQ(1u, context_set.GetAll().count(context)); 34 EXPECT_EQ(context, context_set.GetByV8Context(context->v8_context())); 35 36 // Adding the same item multiple times should be OK and deduped. 37 context_set.Add(context); 38 EXPECT_EQ(1u, context_set.GetAll().count(context)); 39 40 // GetAll() returns a copy so removing from one should not remove from others. 41 ScriptContextSet::ContextSet set_copy = context_set.GetAll(); 42 EXPECT_EQ(1u, set_copy.count(context)); 43 44 context_set.Remove(context); 45 EXPECT_EQ(0, context_set.size()); 46 EXPECT_FALSE(context_set.GetByV8Context(context->v8_context())); 47 EXPECT_EQ(1u, set_copy.size()); 48 49 // After removal, the context should be marked for destruction. 50 EXPECT_FALSE(context->web_frame()); 51 52 // Run loop to do the actual deletion. 53 loop.RunUntilIdle(); 54} 55 56} // namespace extensions 57