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