1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
// 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 "extensions/browser/mojo/keep_alive_impl.h"
#include "base/message_loop/message_loop.h"
#include "base/run_loop.h"
#include "content/public/browser/notification_service.h"
#include "extensions/browser/extensions_test.h"
#include "extensions/browser/process_manager.h"
#include "extensions/common/extension_builder.h"
namespace extensions {
class KeepAliveTest : public ExtensionsTest {
public:
KeepAliveTest()
: notification_service_(content::NotificationService::Create()) {}
~KeepAliveTest() override {}
void SetUp() override {
ExtensionsTest::SetUp();
message_loop_.reset(new base::MessageLoop);
extension_ =
ExtensionBuilder()
.SetManifest(
DictionaryBuilder()
.Set("name", "app")
.Set("version", "1")
.Set("manifest_version", 2)
.Set("app",
DictionaryBuilder().Set(
"background",
DictionaryBuilder().Set(
"scripts",
ListBuilder().Append("background.js")))))
.SetID("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
.Build();
}
void TearDown() override {
message_loop_.reset();
ExtensionsTest::TearDown();
}
void WaitUntilLazyKeepAliveChanges() {
int initial_keep_alive_count = GetKeepAliveCount();
while (GetKeepAliveCount() == initial_keep_alive_count) {
base::RunLoop().RunUntilIdle();
}
}
void CreateKeepAlive(mojo::InterfaceRequest<KeepAlive> request) {
KeepAliveImpl::Create(browser_context(), extension_.get(), request.Pass());
}
const Extension* extension() { return extension_.get(); }
int GetKeepAliveCount() {
return ProcessManager::Get(browser_context())
->GetLazyKeepaliveCount(extension());
}
private:
scoped_ptr<base::MessageLoop> message_loop_;
scoped_ptr<content::NotificationService> notification_service_;
scoped_refptr<const Extension> extension_;
DISALLOW_COPY_AND_ASSIGN(KeepAliveTest);
};
TEST_F(KeepAliveTest, Basic) {
mojo::InterfacePtr<KeepAlive> keep_alive;
CreateKeepAlive(mojo::GetProxy(&keep_alive));
EXPECT_EQ(1, GetKeepAliveCount());
keep_alive.reset();
WaitUntilLazyKeepAliveChanges();
EXPECT_EQ(0, GetKeepAliveCount());
}
TEST_F(KeepAliveTest, TwoKeepAlives) {
mojo::InterfacePtr<KeepAlive> keep_alive;
CreateKeepAlive(mojo::GetProxy(&keep_alive));
EXPECT_EQ(1, GetKeepAliveCount());
mojo::InterfacePtr<KeepAlive> other_keep_alive;
CreateKeepAlive(mojo::GetProxy(&other_keep_alive));
EXPECT_EQ(2, GetKeepAliveCount());
keep_alive.reset();
WaitUntilLazyKeepAliveChanges();
EXPECT_EQ(1, GetKeepAliveCount());
other_keep_alive.reset();
WaitUntilLazyKeepAliveChanges();
EXPECT_EQ(0, GetKeepAliveCount());
}
} // namespace extensions
|
