| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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 "chromeos/dbus/flimflam_client_unittest_base.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/json/json_writer.h" | |
| 9 #include "base/values.h" | |
| 10 #include "dbus/message.h" | |
| 11 #include "dbus/object_path.h" | |
| 12 #include "dbus/values_util.h" | |
| 13 #include "testing/gmock/include/gmock/gmock.h" | |
| 14 #include "testing/gtest/include/gtest/gtest.h" | |
| 15 #include "third_party/cros_system_api/dbus/service_constants.h" | |
| 16 | |
| 17 using ::testing::_; | |
| 18 using ::testing::Invoke; | |
| 19 using ::testing::Return; | |
| 20 | |
| 21 namespace chromeos { | |
| 22 | |
| 23 namespace { | |
| 24 | |
| 25 // Runs the given task. This function is used to implement the mock bus. | |
| 26 void RunTask(const tracked_objects::Location& from_here, | |
| 27 const base::Closure& task) { | |
| 28 task.Run(); | |
| 29 } | |
| 30 | |
| 31 } // namespace | |
| 32 | |
| 33 FlimflamClientUnittestBase::MockClosure::MockClosure() {} | |
| 34 | |
| 35 FlimflamClientUnittestBase::MockClosure::~MockClosure() {} | |
| 36 | |
| 37 base::Closure FlimflamClientUnittestBase::MockClosure::GetCallback() { | |
| 38 return base::Bind(&MockClosure::Run, base::Unretained(this)); | |
| 39 } | |
| 40 | |
| 41 | |
| 42 FlimflamClientUnittestBase::MockErrorCallback::MockErrorCallback() {} | |
| 43 | |
| 44 FlimflamClientUnittestBase::MockErrorCallback::~MockErrorCallback() {} | |
| 45 | |
| 46 FlimflamClientHelper::ErrorCallback | |
| 47 FlimflamClientUnittestBase::MockErrorCallback::GetCallback() { | |
| 48 return base::Bind(&MockErrorCallback::Run, base::Unretained(this)); | |
| 49 } | |
| 50 | |
| 51 | |
| 52 FlimflamClientUnittestBase::FlimflamClientUnittestBase( | |
| 53 const std::string& interface_name, | |
| 54 const dbus::ObjectPath& object_path) | |
| 55 : interface_name_(interface_name), | |
| 56 object_path_(object_path), | |
| 57 response_(NULL) { | |
| 58 } | |
| 59 | |
| 60 FlimflamClientUnittestBase::~FlimflamClientUnittestBase() { | |
| 61 } | |
| 62 | |
| 63 void FlimflamClientUnittestBase::SetUp() { | |
| 64 // Create a mock bus. | |
| 65 dbus::Bus::Options options; | |
| 66 options.bus_type = dbus::Bus::SYSTEM; | |
| 67 mock_bus_ = new dbus::MockBus(options); | |
| 68 | |
| 69 // Create a mock proxy. | |
| 70 mock_proxy_ = new dbus::MockObjectProxy( | |
| 71 mock_bus_.get(), | |
| 72 flimflam::kFlimflamServiceName, | |
| 73 object_path_); | |
| 74 | |
| 75 // Set an expectation so mock_proxy's CallMethodAndBlock() will use | |
| 76 // OnCallMethodAndBlock() to return responses. | |
| 77 EXPECT_CALL(*mock_proxy_, CallMethodAndBlock(_, _)) | |
| 78 .WillRepeatedly(Invoke( | |
| 79 this, &FlimflamClientUnittestBase::OnCallMethodAndBlock)); | |
| 80 | |
| 81 // Set an expectation so mock_proxy's CallMethod() will use OnCallMethod() | |
| 82 // to return responses. | |
| 83 EXPECT_CALL(*mock_proxy_, CallMethod(_, _, _)) | |
| 84 .WillRepeatedly(Invoke(this, &FlimflamClientUnittestBase::OnCallMethod)); | |
| 85 | |
| 86 // Set an expectation so mock_proxy's CallMethodWithErrorCallback() will use | |
| 87 // OnCallMethodWithErrorCallback() to return responses. | |
| 88 EXPECT_CALL(*mock_proxy_, CallMethodWithErrorCallback(_, _, _, _)) | |
| 89 .WillRepeatedly(Invoke( | |
| 90 this, &FlimflamClientUnittestBase::OnCallMethodWithErrorCallback)); | |
| 91 | |
| 92 // Set an expectation so mock_proxy's ConnectToSignal() will use | |
| 93 // OnConnectToSignal() to run the callback. | |
| 94 EXPECT_CALL(*mock_proxy_, ConnectToSignal( | |
| 95 interface_name_, | |
| 96 flimflam::kMonitorPropertyChanged, _, _)) | |
| 97 .WillRepeatedly(Invoke(this, | |
| 98 &FlimflamClientUnittestBase::OnConnectToSignal)); | |
| 99 | |
| 100 // Set an expectation so mock_bus's GetObjectProxy() for the given | |
| 101 // service name and the object path will return mock_proxy_. | |
| 102 EXPECT_CALL(*mock_bus_, GetObjectProxy(flimflam::kFlimflamServiceName, | |
| 103 object_path_)) | |
| 104 .WillOnce(Return(mock_proxy_.get())); | |
| 105 | |
| 106 // Set an expectation so mock_bus's PostTaskToDBusThread() will run the | |
| 107 // given task. | |
| 108 EXPECT_CALL(*mock_bus_, PostTaskToDBusThread(_, _)) | |
| 109 .WillRepeatedly(Invoke(&RunTask)); | |
| 110 | |
| 111 // ShutdownAndBlock() will be called in TearDown(). | |
| 112 EXPECT_CALL(*mock_bus_, ShutdownAndBlock()).WillOnce(Return()); | |
| 113 } | |
| 114 | |
| 115 void FlimflamClientUnittestBase::TearDown() { | |
| 116 mock_bus_->ShutdownAndBlock(); | |
| 117 } | |
| 118 | |
| 119 void FlimflamClientUnittestBase::PrepareForMethodCall( | |
| 120 const std::string& method_name, | |
| 121 const ArgumentCheckCallback& argument_checker, | |
| 122 dbus::Response* response) { | |
| 123 expected_method_name_ = method_name; | |
| 124 argument_checker_ = argument_checker; | |
| 125 response_ = response; | |
| 126 } | |
| 127 | |
| 128 void FlimflamClientUnittestBase::SendPropertyChangedSignal( | |
| 129 dbus::Signal* signal) { | |
| 130 ASSERT_FALSE(property_changed_handler_.is_null()); | |
| 131 property_changed_handler_.Run(signal); | |
| 132 } | |
| 133 | |
| 134 // static | |
| 135 void FlimflamClientUnittestBase::ExpectPropertyChanged( | |
| 136 const std::string& expected_name, | |
| 137 const base::Value* expected_value, | |
| 138 const std::string& name, | |
| 139 const base::Value& value) { | |
| 140 EXPECT_EQ(expected_name, name); | |
| 141 EXPECT_TRUE(expected_value->Equals(&value)); | |
| 142 } | |
| 143 | |
| 144 // static | |
| 145 void FlimflamClientUnittestBase::ExpectNoArgument(dbus::MessageReader* reader) { | |
| 146 EXPECT_FALSE(reader->HasMoreData()); | |
| 147 } | |
| 148 | |
| 149 // static | |
| 150 void FlimflamClientUnittestBase::ExpectStringArgument( | |
| 151 const std::string& expected_string, | |
| 152 dbus::MessageReader* reader) { | |
| 153 std::string str; | |
| 154 ASSERT_TRUE(reader->PopString(&str)); | |
| 155 EXPECT_EQ(expected_string, str); | |
| 156 EXPECT_FALSE(reader->HasMoreData()); | |
| 157 } | |
| 158 | |
| 159 // static | |
| 160 void FlimflamClientUnittestBase::ExpectValueArgument( | |
| 161 const base::Value* expected_value, | |
| 162 dbus::MessageReader* reader) { | |
| 163 scoped_ptr<base::Value> value(dbus::PopDataAsValue(reader)); | |
| 164 ASSERT_TRUE(value.get()); | |
| 165 EXPECT_TRUE(value->Equals(expected_value)); | |
| 166 EXPECT_FALSE(reader->HasMoreData()); | |
| 167 } | |
| 168 | |
| 169 // static | |
| 170 void FlimflamClientUnittestBase::ExpectStringAndValueArguments( | |
| 171 const std::string& expected_string, | |
| 172 const base::Value* expected_value, | |
| 173 dbus::MessageReader* reader) { | |
| 174 std::string str; | |
| 175 ASSERT_TRUE(reader->PopString(&str)); | |
| 176 EXPECT_EQ(expected_string, str); | |
| 177 scoped_ptr<base::Value> value(dbus::PopDataAsValue(reader)); | |
| 178 ASSERT_TRUE(value.get()); | |
| 179 EXPECT_TRUE(value->Equals(expected_value)); | |
| 180 EXPECT_FALSE(reader->HasMoreData()); | |
| 181 } | |
| 182 | |
| 183 // static | |
| 184 void FlimflamClientUnittestBase::ExpectNoResultValue( | |
| 185 DBusMethodCallStatus call_status) { | |
| 186 EXPECT_EQ(DBUS_METHOD_CALL_SUCCESS, call_status); | |
| 187 } | |
| 188 | |
| 189 // static | |
| 190 void FlimflamClientUnittestBase::ExpectObjectPathResult( | |
| 191 const dbus::ObjectPath& expected_result, | |
| 192 DBusMethodCallStatus call_status, | |
| 193 const dbus::ObjectPath& result) { | |
| 194 EXPECT_EQ(DBUS_METHOD_CALL_SUCCESS, call_status); | |
| 195 EXPECT_EQ(expected_result, result); | |
| 196 } | |
| 197 | |
| 198 // static | |
| 199 void FlimflamClientUnittestBase::ExpectDictionaryValueResult( | |
| 200 const base::DictionaryValue* expected_result, | |
| 201 DBusMethodCallStatus call_status, | |
| 202 const base::DictionaryValue& result) { | |
| 203 EXPECT_EQ(DBUS_METHOD_CALL_SUCCESS, call_status); | |
| 204 std::string expected_result_string; | |
| 205 base::JSONWriter::Write(expected_result, &expected_result_string); | |
| 206 std::string result_string; | |
| 207 base::JSONWriter::Write(&result, &result_string); | |
| 208 EXPECT_EQ(expected_result_string, result_string); | |
| 209 } | |
| 210 | |
| 211 void FlimflamClientUnittestBase::OnConnectToSignal( | |
| 212 const std::string& interface_name, | |
| 213 const std::string& signal_name, | |
| 214 const dbus::ObjectProxy::SignalCallback& signal_callback, | |
| 215 const dbus::ObjectProxy::OnConnectedCallback& on_connected_callback) { | |
| 216 property_changed_handler_ = signal_callback; | |
| 217 const bool success = true; | |
| 218 message_loop_.PostTask(FROM_HERE, | |
| 219 base::Bind(on_connected_callback, | |
| 220 interface_name, | |
| 221 signal_name, | |
| 222 success)); | |
| 223 } | |
| 224 | |
| 225 void FlimflamClientUnittestBase::OnCallMethod( | |
| 226 dbus::MethodCall* method_call, | |
| 227 int timeout_ms, | |
| 228 const dbus::ObjectProxy::ResponseCallback& response_callback) { | |
| 229 EXPECT_EQ(interface_name_, method_call->GetInterface()); | |
| 230 EXPECT_EQ(expected_method_name_, method_call->GetMember()); | |
| 231 dbus::MessageReader reader(method_call); | |
| 232 argument_checker_.Run(&reader); | |
| 233 message_loop_.PostTask(FROM_HERE, | |
| 234 base::Bind(response_callback, response_)); | |
| 235 } | |
| 236 | |
| 237 void FlimflamClientUnittestBase::OnCallMethodWithErrorCallback( | |
| 238 dbus::MethodCall* method_call, | |
| 239 int timeout_ms, | |
| 240 const dbus::ObjectProxy::ResponseCallback& response_callback, | |
| 241 const dbus::ObjectProxy::ErrorCallback& error_callback) { | |
| 242 OnCallMethod(method_call, timeout_ms, response_callback); | |
| 243 } | |
| 244 | |
| 245 dbus::Response* FlimflamClientUnittestBase::OnCallMethodAndBlock( | |
| 246 dbus::MethodCall* method_call, | |
| 247 int timeout_ms) { | |
| 248 EXPECT_EQ(interface_name_, method_call->GetInterface()); | |
| 249 EXPECT_EQ(expected_method_name_, method_call->GetMember()); | |
| 250 dbus::MessageReader reader(method_call); | |
| 251 argument_checker_.Run(&reader); | |
| 252 return dbus::Response::FromRawMessage( | |
| 253 dbus_message_copy(response_->raw_message())); | |
| 254 } | |
| 255 | |
| 256 } // namespace chromeos | |
| OLD | NEW |