| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROMEOS_CRYPTOHOME_MOCK_ASYNC_METHOD_CALLER_H_ | 5 #ifndef CHROMEOS_CRYPTOHOME_MOCK_ASYNC_METHOD_CALLER_H_ |
| 6 #define CHROMEOS_CRYPTOHOME_MOCK_ASYNC_METHOD_CALLER_H_ | 6 #define CHROMEOS_CRYPTOHOME_MOCK_ASYNC_METHOD_CALLER_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| 11 #include "base/callback.h" | 11 #include "base/callback.h" |
| 12 #include "chromeos/cryptohome/async_method_caller.h" | 12 #include "chromeos/cryptohome/async_method_caller.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
| 14 | 14 |
| 15 namespace cryptohome { | 15 namespace cryptohome { |
| 16 | 16 |
| 17 class MockAsyncMethodCaller : public AsyncMethodCaller { | 17 class MockAsyncMethodCaller : public AsyncMethodCaller { |
| 18 public: | 18 public: |
| 19 static const char kFakeAttestationEnrollRequest[]; | 19 static const char kFakeAttestationEnrollRequest[]; |
| 20 static const char kFakeAttestationCertRequest[]; | 20 static const char kFakeAttestationCertRequest[]; |
| 21 static const char kFakeAttestationCert[]; | 21 static const char kFakeAttestationCert[]; |
| 22 static const char kFakeSanitizedUsername[]; | 22 static const char kFakeSanitizedUsername[]; |
| 23 static const char kFakeChallengeResponse[]; | 23 static const char kFakeChallengeResponse[]; |
| 24 | 24 |
| 25 MockAsyncMethodCaller(); | 25 MockAsyncMethodCaller(); |
| 26 virtual ~MockAsyncMethodCaller(); | 26 virtual ~MockAsyncMethodCaller(); |
| 27 | 27 |
| 28 void SetUp(bool success, MountError return_code); | 28 void SetUp(bool success, MountError return_code); |
| 29 | 29 |
| 30 MOCK_METHOD3(AsyncCheckKey, void(const std::string& user_email, | 30 MOCK_METHOD3(AsyncCheckKey, void(const user_manager::UserID& user_id, |
| 31 const std::string& passhash, | 31 const std::string& passhash, |
| 32 Callback callback)); | 32 Callback callback)); |
| 33 MOCK_METHOD4(AsyncMigrateKey, void(const std::string& user_email, | 33 MOCK_METHOD4(AsyncMigrateKey, void(const user_manager::UserID& user_id, |
| 34 const std::string& old_hash, | 34 const std::string& old_hash, |
| 35 const std::string& new_hash, | 35 const std::string& new_hash, |
| 36 Callback callback)); | 36 Callback callback)); |
| 37 MOCK_METHOD4(AsyncMount, void(const std::string& user_email, | 37 MOCK_METHOD4(AsyncMount, void(const user_manager::UserID& user_id, |
| 38 const std::string& passhash, | 38 const std::string& passhash, |
| 39 int flags, | 39 int flags, |
| 40 Callback callback)); | 40 Callback callback)); |
| 41 MOCK_METHOD4(AsyncAddKey, void(const std::string& user_email, | 41 MOCK_METHOD4(AsyncAddKey, void(const user_manager::UserID& user_id, |
| 42 const std::string& passhash, | 42 const std::string& passhash, |
| 43 const std::string& new_key, | 43 const std::string& new_key, |
| 44 Callback callback)); | 44 Callback callback)); |
| 45 MOCK_METHOD1(AsyncMountGuest, void(Callback callback)); | 45 MOCK_METHOD1(AsyncMountGuest, void(Callback callback)); |
| 46 MOCK_METHOD3(AsyncMountPublic, void(const std::string& public_mount_id, | 46 MOCK_METHOD3(AsyncMountPublic, void(const std::string& public_mount_id, |
| 47 int flags, | 47 int flags, |
| 48 Callback callback)); | 48 Callback callback)); |
| 49 MOCK_METHOD2(AsyncRemove, void(const std::string& user_email, | 49 MOCK_METHOD2(AsyncRemove, void(const user_manager::UserID& user_id, |
| 50 Callback callback)); | 50 Callback callback)); |
| 51 MOCK_METHOD2(AsyncTpmAttestationCreateEnrollRequest, | 51 MOCK_METHOD2(AsyncTpmAttestationCreateEnrollRequest, |
| 52 void(chromeos::attestation::PrivacyCAType pca_type, | 52 void(chromeos::attestation::PrivacyCAType pca_type, |
| 53 const DataCallback& callback)); | 53 const DataCallback& callback)); |
| 54 MOCK_METHOD3(AsyncTpmAttestationEnroll, | 54 MOCK_METHOD3(AsyncTpmAttestationEnroll, |
| 55 void(chromeos::attestation::PrivacyCAType pca_type, | 55 void(chromeos::attestation::PrivacyCAType pca_type, |
| 56 const std::string& pca_response, | 56 const std::string& pca_response, |
| 57 const Callback& callback)); | 57 const Callback& callback)); |
| 58 MOCK_METHOD5( | 58 MOCK_METHOD5( |
| 59 AsyncTpmAttestationCreateCertRequest, | 59 AsyncTpmAttestationCreateCertRequest, |
| 60 void(chromeos::attestation::PrivacyCAType pca_type, | 60 void(chromeos::attestation::PrivacyCAType pca_type, |
| 61 chromeos::attestation::AttestationCertificateProfile profile, | 61 chromeos::attestation::AttestationCertificateProfile profile, |
| 62 const std::string& user_id, | 62 const user_manager::UserID& user_id, |
| 63 const std::string& request_origin, | 63 const std::string& request_origin, |
| 64 const DataCallback& callback)); | 64 const DataCallback& callback)); |
| 65 MOCK_METHOD5(AsyncTpmAttestationFinishCertRequest, | 65 MOCK_METHOD5(AsyncTpmAttestationFinishCertRequest, |
| 66 void(const std::string& pca_response, | 66 void(const std::string& pca_response, |
| 67 chromeos::attestation::AttestationKeyType key_type, | 67 chromeos::attestation::AttestationKeyType key_type, |
| 68 const std::string& user_id, | 68 const user_manager::UserID& user_id, |
| 69 const std::string& key_name, | 69 const std::string& key_name, |
| 70 const DataCallback& callback)); | 70 const DataCallback& callback)); |
| 71 MOCK_METHOD4(TpmAttestationRegisterKey, | 71 MOCK_METHOD4(TpmAttestationRegisterKey, |
| 72 void(chromeos::attestation::AttestationKeyType key_type, | 72 void(chromeos::attestation::AttestationKeyType key_type, |
| 73 const std::string& user_id, | 73 const user_manager::UserID& user_id, |
| 74 const std::string& key_name, | 74 const std::string& key_name, |
| 75 const Callback& callback)); | 75 const Callback& callback)); |
| 76 MOCK_METHOD8( | 76 MOCK_METHOD8( |
| 77 TpmAttestationSignEnterpriseChallenge, | 77 TpmAttestationSignEnterpriseChallenge, |
| 78 void(chromeos::attestation::AttestationKeyType key_type, | 78 void(chromeos::attestation::AttestationKeyType key_type, |
| 79 const std::string& user_id, | 79 const user_manager::UserID& user_id, |
| 80 const std::string& key_name, | 80 const std::string& key_name, |
| 81 const std::string& domain, | 81 const std::string& domain, |
| 82 const std::string& device_id, | 82 const std::string& device_id, |
| 83 chromeos::attestation::AttestationChallengeOptions options, | 83 chromeos::attestation::AttestationChallengeOptions options, |
| 84 const std::string& challenge, | 84 const std::string& challenge, |
| 85 const DataCallback& callback)); | 85 const DataCallback& callback)); |
| 86 MOCK_METHOD5(TpmAttestationSignSimpleChallenge, | 86 MOCK_METHOD5(TpmAttestationSignSimpleChallenge, |
| 87 void(chromeos::attestation::AttestationKeyType key_type, | 87 void(chromeos::attestation::AttestationKeyType key_type, |
| 88 const std::string& user_id, | 88 const user_manager::UserID& user_id, |
| 89 const std::string& key_name, | 89 const std::string& key_name, |
| 90 const std::string& challenge, | 90 const std::string& challenge, |
| 91 const DataCallback& callback)); | 91 const DataCallback& callback)); |
| 92 MOCK_METHOD2(AsyncGetSanitizedUsername, | 92 MOCK_METHOD2(AsyncGetSanitizedUsername, |
| 93 void(const std::string& user, | 93 void(const user_manager::UserID& user_id, |
| 94 const DataCallback& callback)); | 94 const DataCallback& callback)); |
| 95 | 95 |
| 96 private: | 96 private: |
| 97 bool success_; | 97 bool success_; |
| 98 MountError return_code_; | 98 MountError return_code_; |
| 99 | 99 |
| 100 void DoCallback(Callback callback); | 100 void DoCallback(Callback callback); |
| 101 // Default fakes for attestation calls. | 101 // Default fakes for attestation calls. |
| 102 void FakeCreateEnrollRequest(const DataCallback& callback); | 102 void FakeCreateEnrollRequest(const DataCallback& callback); |
| 103 void FakeCreateCertRequest(const DataCallback& callback); | 103 void FakeCreateCertRequest(const DataCallback& callback); |
| 104 void FakeFinishCertRequest(const DataCallback& callback); | 104 void FakeFinishCertRequest(const DataCallback& callback); |
| 105 void FakeGetSanitizedUsername(const DataCallback& callback); | 105 void FakeGetSanitizedUsername(const DataCallback& callback); |
| 106 void FakeEnterpriseChallenge(const DataCallback& callback); | 106 void FakeEnterpriseChallenge(const DataCallback& callback); |
| 107 | 107 |
| 108 DISALLOW_COPY_AND_ASSIGN(MockAsyncMethodCaller); | 108 DISALLOW_COPY_AND_ASSIGN(MockAsyncMethodCaller); |
| 109 }; | 109 }; |
| 110 | 110 |
| 111 } // namespace cryptohome | 111 } // namespace cryptohome |
| 112 | 112 |
| 113 #endif // CHROMEOS_CRYPTOHOME_MOCK_ASYNC_METHOD_CALLER_H_ | 113 #endif // CHROMEOS_CRYPTOHOME_MOCK_ASYNC_METHOD_CALLER_H_ |
| OLD | NEW |