Index: chromeos/attestation/mock_attestation_flow.h |
diff --git a/chromeos/attestation/mock_attestation_flow.h b/chromeos/attestation/mock_attestation_flow.h |
index c9dc9d57892483ed6160737b5284b10ed55d2d49..59462c5410daeb494f6375be2139582cc0ac490d 100644 |
--- a/chromeos/attestation/mock_attestation_flow.h |
+++ b/chromeos/attestation/mock_attestation_flow.h |
@@ -2,6 +2,9 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
+#ifndef CHROMEOS_ATTESTATION_MOCK_ATTESTATION_FLOW_H_ |
+#define CHROMEOS_ATTESTATION_MOCK_ATTESTATION_FLOW_H_ |
+ |
#include "chromeos/attestation/attestation_flow.h" |
#include "base/basictypes.h" |
@@ -57,5 +60,16 @@ class MockObserver { |
MOCK_METHOD2(MockCertificateCallback, void(bool, const std::string&)); |
}; |
+class MockAttestationFlow : public AttestationFlow { |
+ public: |
+ MockAttestationFlow(); |
+ virtual ~MockAttestationFlow(); |
+ |
+ MOCK_METHOD2(GetCertificate, void(const std::string&, |
+ const CertificateCallback&)); |
+}; |
+ |
} // namespace attestation |
} // namespace chromeos |
+ |
+#endif // CHROMEOS_ATTESTATION_MOCK_ATTESTATION_FLOW_H_ |