1ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 5c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#ifndef CHROME_BROWSER_CHROMEOS_CROS_MOCK_CRYPTOHOME_LIBRARY_H_ 6c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#define CHROME_BROWSER_CHROMEOS_CROS_MOCK_CRYPTOHOME_LIBRARY_H_ 73345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#pragma once 8c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include <string> 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 11c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/chromeos/cros/cryptohome_library.h" 12c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "testing/gmock/include/gmock/gmock.h" 13c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 143345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrickusing ::testing::Invoke; 153345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrickusing ::testing::WithArgs; 163345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrickusing ::testing::_; 173345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochnamespace chromeos { 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochclass MockCryptohomeLibrary : public CryptohomeLibrary { 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch public: 22731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MockCryptohomeLibrary() : outcome_(false), code_(0) { 23731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick } 24731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick virtual ~MockCryptohomeLibrary() {} 25731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick void SetUp(bool outcome, int code) { 26731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick outcome_ = outcome; 27731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick code_ = code; 283345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ON_CALL(*this, AsyncCheckKey(_, _, _)) 293345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick .WillByDefault( 303345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick WithArgs<2>(Invoke(this, &MockCryptohomeLibrary::DoCallback))); 313345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ON_CALL(*this, AsyncMigrateKey(_, _, _, _)) 323345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick .WillByDefault( 333345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick WithArgs<3>(Invoke(this, &MockCryptohomeLibrary::DoCallback))); 343345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ON_CALL(*this, AsyncMount(_, _, _, _)) 353345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick .WillByDefault( 363345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick WithArgs<3>(Invoke(this, &MockCryptohomeLibrary::DoCallback))); 373345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ON_CALL(*this, AsyncMountForBwsi(_)) 383345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick .WillByDefault( 393345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick WithArgs<0>(Invoke(this, &MockCryptohomeLibrary::DoCallback))); 403345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick ON_CALL(*this, AsyncRemove(_, _)) 413345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick .WillByDefault( 423345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick WithArgs<1>(Invoke(this, &MockCryptohomeLibrary::DoCallback))); 43dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen ON_CALL(*this, AsyncDoAutomaticFreeDiskSpaceControl(_)) 44dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen .WillByDefault( 45dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen WithArgs<0>(Invoke(this, &MockCryptohomeLibrary::DoCallback))); 463345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 47c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch MOCK_METHOD2(CheckKey, bool(const std::string& user_email, 48c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::string& passhash)); 493345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD3(AsyncCheckKey, bool(const std::string& user_email, 503345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string& passhash, 513345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick Delegate* callback)); 52c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch MOCK_METHOD3(MigrateKey, bool(const std::string& user_email, 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::string& old_hash, 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch const std::string& new_hash)); 553345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD4(AsyncMigrateKey, bool(const std::string& user_email, 563345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string& old_hash, 573345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string& new_hash, 583345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick Delegate* callback)); 593345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD3(Mount, bool(const std::string& user_email, 603345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string& passhash, 613345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick int* error_code)); 623345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD4(AsyncMount, bool(const std::string& user_email, 633345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string& passhash, 643345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const bool create_if_missing, 653345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick Delegate* callback)); 663345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD1(MountForBwsi, bool(int*)); 673345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD1(AsyncMountForBwsi, bool(Delegate* callback)); 68731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(Unmount, bool(void)); 69c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch MOCK_METHOD1(Remove, bool(const std::string& user_email)); 703345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick MOCK_METHOD2(AsyncRemove, bool(const std::string& user_email, Delegate* d)); 71c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch MOCK_METHOD0(IsMounted, bool(void)); 72c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch MOCK_METHOD0(GetSystemSalt, CryptohomeBlob(void)); 73dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen MOCK_METHOD1(AsyncDoAutomaticFreeDiskSpaceControl, bool(Delegate* callback)); 743345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 75731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(TpmIsReady, bool(void)); 76731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(TpmIsEnabled, bool(void)); 77731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(TpmIsOwned, bool(void)); 78731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(TpmIsBeingOwned, bool(void)); 79731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD1(TpmGetPassword, bool(std::string* password)); 80731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(TpmCanAttemptOwnership, void(void)); 81731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick MOCK_METHOD0(TpmClearStoredPassword, void(void)); 82ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(Pkcs11IsTpmTokenReady, bool(void)); 83ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD2(Pkcs11GetTpmTokenInfo, void(std::string*, std::string*)); 84ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen 85ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD2(InstallAttributesGet, bool(const std::string&, std::string*)); 86ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD2(InstallAttributesSet, bool(const std::string&, 87ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen const std::string&)); 88ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(InstallAttributesCount, int(void)); 89ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(InstallAttributesFinalize, bool(void)); 90ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(InstallAttributesIsReady, bool(void)); 91ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(InstallAttributesIsSecure, bool(void)); 92ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(InstallAttributesIsInvalid, bool(void)); 93ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen MOCK_METHOD0(InstallAttributesIsFirstInstall, bool(void)); 94731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 953345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick void SetAsyncBehavior(bool outcome, int code) { 963345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick outcome_ = outcome; 973345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick code_ = code; 983345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 993345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 1003345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick bool DoCallback(Delegate* d) { 1013345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick d->OnComplete(outcome_, code_); 1023345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return true; 1033345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 1043345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 1053345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick private: 1063345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick bool outcome_; 1073345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick int code_; 1083345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick DISALLOW_COPY_AND_ASSIGN(MockCryptohomeLibrary); 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch}; 110dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} // namespace chromeos 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#endif // CHROME_BROWSER_CHROMEOS_CROS_MOCK_CRYPTOHOME_LIBRARY_H_ 114