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 |