1// Copyright (c) 2010 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#ifndef CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_LOGIN_STATUS_CONSUMER_H_
6#define CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_LOGIN_STATUS_CONSUMER_H_
7#pragma once
8
9#include "chrome/browser/chromeos/login/login_status_consumer.h"
10#include "chrome/common/net/gaia/gaia_auth_consumer.h"
11#include "testing/gmock/include/gmock/gmock.h"
12#include "testing/gtest/include/gtest/gtest.h"
13
14namespace chromeos {
15
16class MockConsumer : public LoginStatusConsumer {
17 public:
18  MockConsumer() {}
19  ~MockConsumer() {}
20  MOCK_METHOD1(OnLoginFailure, void(const LoginFailure& error));
21  MOCK_METHOD4(OnLoginSuccess, void(
22      const std::string& username,
23      const std::string& password,
24      const GaiaAuthConsumer::ClientLoginResult& result,
25      bool pending_requests));
26  MOCK_METHOD0(OnOffTheRecordLoginSuccess, void(void));
27  MOCK_METHOD1(OnPasswordChangeDetected,
28      void(const GaiaAuthConsumer::ClientLoginResult& result));
29
30  // The following functions can be used in gmock Invoke() clauses.
31
32  // Compatible with LoginStatusConsumer::OnOffTheRecordLoginSuccess()
33  static void OnGuestSuccessQuit() {
34    MessageLoop::current()->Quit();
35  }
36
37  static void OnGuestSuccessQuitAndFail() {
38    ADD_FAILURE() << "Guest Login should have failed!";
39    MessageLoop::current()->Quit();
40  }
41
42  // Compatible with LoginStatusConsumer::OnLoginSuccess()
43  static void OnSuccessQuit(
44      const std::string& username,
45      const std::string& password,
46      const GaiaAuthConsumer::ClientLoginResult& credentials,
47      bool pending_requests) {
48    MessageLoop::current()->Quit();
49  }
50
51  static void OnSuccessQuitAndFail(
52      const std::string& username,
53      const std::string& password,
54      const GaiaAuthConsumer::ClientLoginResult& credentials,
55      bool pending_requests) {
56    ADD_FAILURE() << "Login should NOT have succeeded!";
57    MessageLoop::current()->Quit();
58  }
59
60  // Compatible with LoginStatusConsumer::OnLoginFailure()
61  static void OnFailQuit(const LoginFailure& error) {
62    MessageLoop::current()->Quit();
63  }
64
65  static void OnFailQuitAndFail(const LoginFailure& error) {
66    ADD_FAILURE() << "Login should not have failed!";
67    MessageLoop::current()->Quit();
68  }
69
70  // Compatible with LoginStatusConsumer::OnPasswordChangeDetected()
71  static void OnMigrateQuit(
72      const GaiaAuthConsumer::ClientLoginResult& credentials) {
73    MessageLoop::current()->Quit();
74  }
75
76  static void OnMigrateQuitAndFail(
77      const GaiaAuthConsumer::ClientLoginResult& credentials) {
78    ADD_FAILURE() << "Should not have detected a PW change!";
79    MessageLoop::current()->Quit();
80  }
81};
82
83}  // namespace chromeos
84
85#endif  // CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_LOGIN_STATUS_CONSUMER_H_
86