cros: Slightly more mock out-of-lining.

chrome/browser/chromeos/login/owner_manager_unittest.cc

 // Copyright (c) 2011 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 "chrome/browser/chromeos/login/owner_manager.h"
 #include "chrome/browser/chromeos/login/owner_manager_unittest.h"
 
 #include <string>
 
 #include "base/file_path.h"
@@ skipping 10 matching lines @@
 using ::crypto::RSAPrivateKey;
 using ::testing::DoAll;
 using ::testing::Eq;
 using ::testing::Invoke;
 using ::testing::Return;
 using ::testing::SetArgumentPointee;
 using ::testing::_;
 
 namespace chromeos {
 
+
+void MockKeyLoadObserver::Observe(NotificationType type,
+                                  const NotificationSource& source,
+                                  const NotificationDetails& details) {
+  LOG(INFO) << "Observed key fetch event";
+  if (type == NotificationType::OWNER_KEY_FETCH_ATTEMPT_SUCCEEDED) {
+    EXPECT_TRUE(success_expected_);
+    observed_ = true;
+    if (quit_on_observe_)
+      MessageLoop::current()->Quit();
+  } else if (type == NotificationType::OWNER_KEY_FETCH_ATTEMPT_FAILED) {
+    EXPECT_FALSE(success_expected_);
+    observed_ = true;
+    if (quit_on_observe_)
+      MessageLoop::current()->Quit();
+  }
+}
+
+void MockKeyUser::OnKeyOpComplete(const OwnerManager::KeyOpCode return_code,
+                                  const std::vector<uint8>& payload) {
+  EXPECT_EQ(expected_, return_code);
+  if (quit_on_callback_)
+    MessageLoop::current()->Quit();
+}
+
+
 class OwnerManagerTest : public ::testing::Test {
  public:
   OwnerManagerTest()
       : message_loop_(MessageLoop::TYPE_UI),
         ui_thread_(BrowserThread::UI, &message_loop_),
         file_thread_(BrowserThread::FILE),
         mock_(new MockKeyUtils),
         injector_(mock_) /* injector_ takes ownership of mock_ */ {
   }
   virtual ~OwnerManagerTest() {}
@@ skipping 212 matching lines @@
       BrowserThread::FILE, FROM_HERE,
       NewRunnableMethod(manager.get(),
                         &OwnerManager::Sign,
                         BrowserThread::UI,
                         data,
                         &delegate));
   message_loop_.Run();
 }
 
 }  // namespace chromeos