blob: db13d8040d8d18067add010938a3a82635cfde5a [file] [log] [blame]
Peter Qiu326b6cf2015-09-02 11:11:42 -07001//
2// Copyright (C) 2014 The Android Open Source Project
3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8// http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15//
Peter Qiu376e4042014-11-13 09:40:28 -080016
17#include "apmanager/service.h"
18
19#include <string>
20
21#include <base/strings/string_util.h>
22#include <base/strings/stringprintf.h>
Peter Qiu376e4042014-11-13 09:40:28 -080023#include <chromeos/process_mock.h>
24#include <gmock/gmock.h>
25#include <gtest/gtest.h>
26
Peter Qiu7a420d32015-09-22 11:25:15 -070027#if !defined(__ANDROID__)
28#include <chromeos/dbus/service_constants.h>
29#else
30#include "dbus/apmanager/dbus-constants.h"
31#endif // __ANDROID__
32
Peter Qiu376e4042014-11-13 09:40:28 -080033#include "apmanager/mock_config.h"
Peter Qiubf8e36c2014-12-03 22:59:45 -080034#include "apmanager/mock_dhcp_server.h"
35#include "apmanager/mock_dhcp_server_factory.h"
Peter Qiu77517302015-01-08 16:22:16 -080036#include "apmanager/mock_file_writer.h"
Peter Qiufda548c2015-01-13 14:39:19 -080037#include "apmanager/mock_hostapd_monitor.h"
Peter Qiufb39ba42014-11-21 09:09:59 -080038#include "apmanager/mock_manager.h"
Peter Qiu1dbf9fd2015-01-09 13:36:55 -080039#include "apmanager/mock_process_factory.h"
Peter Qiu376e4042014-11-13 09:40:28 -080040
41using chromeos::ProcessMock;
42using ::testing::_;
43using ::testing::Mock;
44using ::testing::Return;
45using ::testing::SetArgPointee;
46
47namespace {
48 const int kServiceIdentifier = 1;
49 const char kHostapdConfig[] = "ssid=test\n";
David Pursell8b9b2542014-12-02 10:15:35 -080050 const char kBinSleep[] = "/bin/sleep";
Peter Qiu77517302015-01-08 16:22:16 -080051 const char kHostapdConfigFilePath[] =
52 "/var/run/apmanager/hostapd/hostapd-1.conf";
Peter Qiu376e4042014-11-13 09:40:28 -080053} // namespace
54
55namespace apmanager {
56
57class ServiceTest : public testing::Test {
58 public:
Peter Qiu1dbf9fd2015-01-09 13:36:55 -080059 ServiceTest()
Peter Qiu943cf3a2015-02-24 10:59:17 -080060 : dhcp_server_factory_(MockDHCPServerFactory::GetInstance()),
Peter Qiu1dbf9fd2015-01-09 13:36:55 -080061 file_writer_(MockFileWriter::GetInstance()),
Peter Qiufda548c2015-01-13 14:39:19 -080062 process_factory_(MockProcessFactory::GetInstance()),
Peter Qiu943cf3a2015-02-24 10:59:17 -080063 hostapd_monitor_(new MockHostapdMonitor()),
64 service_(&manager_, kServiceIdentifier) {}
Peter Qiu1dbf9fd2015-01-09 13:36:55 -080065
66 virtual void SetUp() {
67 service_.dhcp_server_factory_ = dhcp_server_factory_;
68 service_.file_writer_ = file_writer_;
69 service_.process_factory_ = process_factory_;
Peter Qiufda548c2015-01-13 14:39:19 -080070 service_.hostapd_monitor_.reset(hostapd_monitor_);
Peter Qiu1dbf9fd2015-01-09 13:36:55 -080071 }
Peter Qiu376e4042014-11-13 09:40:28 -080072
73 void StartDummyProcess() {
74 service_.hostapd_process_.reset(new chromeos::ProcessImpl);
David Pursell8b9b2542014-12-02 10:15:35 -080075 service_.hostapd_process_->AddArg(kBinSleep);
Peter Qiu376e4042014-11-13 09:40:28 -080076 service_.hostapd_process_->AddArg("12345");
77 CHECK(service_.hostapd_process_->Start());
78 LOG(INFO) << "DummyProcess: " << service_.hostapd_process_->pid();
79 }
80
81 void SetConfig(Config* config) {
82 service_.config_.reset(config);
83 }
84
85 protected:
Peter Qiufb39ba42014-11-21 09:09:59 -080086 MockManager manager_;
Peter Qiu1dbf9fd2015-01-09 13:36:55 -080087 MockDHCPServerFactory* dhcp_server_factory_;
88 MockFileWriter* file_writer_;
89 MockProcessFactory* process_factory_;
Peter Qiufda548c2015-01-13 14:39:19 -080090 MockHostapdMonitor* hostapd_monitor_;
Peter Qiu943cf3a2015-02-24 10:59:17 -080091 Service service_;
Peter Qiu376e4042014-11-13 09:40:28 -080092};
93
94MATCHER_P(IsServiceErrorStartingWith, message, "") {
95 return arg != nullptr &&
Alex Vakulenkoe8f58982015-06-15 12:53:22 -070096 arg->GetDomain() == chromeos::errors::dbus::kDomain &&
97 arg->GetCode() == kServiceError &&
98 base::EndsWith(arg->GetMessage(), message, false);
Peter Qiu376e4042014-11-13 09:40:28 -080099}
100
101TEST_F(ServiceTest, StartWhenServiceAlreadyRunning) {
102 StartDummyProcess();
103
104 chromeos::ErrorPtr error;
105 EXPECT_FALSE(service_.Start(&error));
106 EXPECT_THAT(error, IsServiceErrorStartingWith("Service already running"));
107}
108
109TEST_F(ServiceTest, StartWhenConfigFileFailed) {
110 MockConfig* config = new MockConfig();
111 SetConfig(config);
112
113 chromeos::ErrorPtr error;
114 EXPECT_CALL(*config, GenerateConfigFile(_, _)).WillOnce(Return(false));
115 EXPECT_FALSE(service_.Start(&error));
116 EXPECT_THAT(error, IsServiceErrorStartingWith(
117 "Failed to generate config file"));
118}
119
120TEST_F(ServiceTest, StartSuccess) {
121 MockConfig* config = new MockConfig();
122 SetConfig(config);
123
Peter Qiubf8e36c2014-12-03 22:59:45 -0800124 // Setup mock DHCP server.
Peter Qiubf8e36c2014-12-03 22:59:45 -0800125 MockDHCPServer* dhcp_server = new MockDHCPServer();
Peter Qiu1dbf9fd2015-01-09 13:36:55 -0800126 // Setup mock process.
127 ProcessMock* process = new ProcessMock();
Peter Qiu77517302015-01-08 16:22:16 -0800128
Peter Qiu376e4042014-11-13 09:40:28 -0800129 std::string config_str(kHostapdConfig);
130 chromeos::ErrorPtr error;
131 EXPECT_CALL(*config, GenerateConfigFile(_, _)).WillOnce(
132 DoAll(SetArgPointee<1>(config_str), Return(true)));
Peter Qiu1dbf9fd2015-01-09 13:36:55 -0800133 EXPECT_CALL(*file_writer_, Write(kHostapdConfigFilePath, kHostapdConfig))
Peter Qiu77517302015-01-08 16:22:16 -0800134 .WillOnce(Return(true));
Peter Qiufb39ba42014-11-21 09:09:59 -0800135 EXPECT_CALL(*config, ClaimDevice()).WillOnce(Return(true));
Peter Qiu1dbf9fd2015-01-09 13:36:55 -0800136 EXPECT_CALL(*process_factory_, CreateProcess()).WillOnce(Return(process));
137 EXPECT_CALL(*process, Start()).WillOnce(Return(true));
138 EXPECT_CALL(*dhcp_server_factory_, CreateDHCPServer(_, _))
Peter Qiubf8e36c2014-12-03 22:59:45 -0800139 .WillOnce(Return(dhcp_server));
140 EXPECT_CALL(*dhcp_server, Start()).WillOnce(Return(true));
Peter Qiu943cf3a2015-02-24 10:59:17 -0800141 EXPECT_CALL(manager_, RequestDHCPPortAccess(_));
Peter Qiufda548c2015-01-13 14:39:19 -0800142 EXPECT_CALL(*hostapd_monitor_, Start());
Peter Qiu376e4042014-11-13 09:40:28 -0800143 EXPECT_TRUE(service_.Start(&error));
144 EXPECT_EQ(nullptr, error);
145}
146
147TEST_F(ServiceTest, StopWhenServiceNotRunning) {
148 chromeos::ErrorPtr error;
149 EXPECT_FALSE(service_.Stop(&error));
150 EXPECT_THAT(error, IsServiceErrorStartingWith(
151 "Service is not currently running"));
152}
153
154TEST_F(ServiceTest, StopSuccess) {
155 StartDummyProcess();
156
Peter Qiufb39ba42014-11-21 09:09:59 -0800157 MockConfig* config = new MockConfig();
158 SetConfig(config);
Peter Qiu376e4042014-11-13 09:40:28 -0800159 chromeos::ErrorPtr error;
Peter Qiufb39ba42014-11-21 09:09:59 -0800160 EXPECT_CALL(*config, ReleaseDevice()).Times(1);
Peter Qiu376e4042014-11-13 09:40:28 -0800161 EXPECT_TRUE(service_.Stop(&error));
162}
163
164} // namespace apmanager