OLD | NEW |
| (Empty) |
1 // Copyright 2016 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 "services/shell/public/cpp/shell_test.h" | |
6 | |
7 #include "base/memory/ptr_util.h" | |
8 #include "base/message_loop/message_loop.h" | |
9 #include "base/run_loop.h" | |
10 #include "services/shell/background/background_shell.h" | |
11 #include "services/shell/public/cpp/service.h" | |
12 | |
13 namespace shell { | |
14 namespace test { | |
15 | |
16 ShellTestClient::ShellTestClient(ShellTest* test) : test_(test) {} | |
17 ShellTestClient::~ShellTestClient() {} | |
18 | |
19 void ShellTestClient::OnStart(Connector* connector, const Identity& identity, | |
20 uint32_t id) { | |
21 test_->OnStartCalled(connector, identity.name(), identity.user_id(), id); | |
22 } | |
23 | |
24 ShellTest::ShellTest() {} | |
25 ShellTest::ShellTest(const std::string& test_name) : test_name_(test_name) {} | |
26 ShellTest::~ShellTest() {} | |
27 | |
28 void ShellTest::InitTestName(const std::string& test_name) { | |
29 DCHECK(test_name_.empty()); | |
30 test_name_ = test_name; | |
31 } | |
32 | |
33 std::unique_ptr<Service> ShellTest::CreateService() { | |
34 return base::WrapUnique(new ShellTestClient(this)); | |
35 } | |
36 | |
37 std::unique_ptr<base::MessageLoop> ShellTest::CreateMessageLoop() { | |
38 return base::WrapUnique(new base::MessageLoop); | |
39 } | |
40 | |
41 void ShellTest::OnStartCalled(Connector* connector, | |
42 const std::string& name, | |
43 const std::string& user_id, | |
44 uint32_t id) { | |
45 DCHECK_EQ(connector_, connector); | |
46 initialize_name_ = name; | |
47 initialize_instance_id_ = id; | |
48 initialize_userid_ = user_id; | |
49 initialize_called_.Run(); | |
50 } | |
51 | |
52 void ShellTest::SetUp() { | |
53 service_ = CreateService(); | |
54 message_loop_ = CreateMessageLoop(); | |
55 background_shell_.reset(new shell::BackgroundShell); | |
56 background_shell_->Init(nullptr); | |
57 | |
58 // Create the shell connection. We don't proceed until we get our | |
59 // Service's OnStart() method is called. | |
60 base::RunLoop run_loop; | |
61 base::MessageLoop::ScopedNestableTaskAllower allow( | |
62 base::MessageLoop::current()); | |
63 initialize_called_ = run_loop.QuitClosure(); | |
64 | |
65 shell_connection_.reset(new ShellConnection( | |
66 service_.get(), | |
67 background_shell_->CreateServiceRequest(test_name_))); | |
68 connector_ = shell_connection_->connector(); | |
69 | |
70 run_loop.Run(); | |
71 } | |
72 | |
73 void ShellTest::TearDown() { | |
74 shell_connection_.reset(); | |
75 background_shell_.reset(); | |
76 message_loop_.reset(); | |
77 service_.reset(); | |
78 } | |
79 | |
80 } // namespace test | |
81 } // namespace shell | |
OLD | NEW |