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 |