Enhanced and refactored the AttestationFlow interface.

chromeos/attestation/mock_attestation_flow.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // 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"
@@ skipping 47 matching lines @@
   virtual ~MockObserver();
 
   MOCK_METHOD2(MockCertificateCallback, void(bool, const std::string&));
 };
 
 class MockAttestationFlow : public AttestationFlow {
  public:
   MockAttestationFlow();
   virtual ~MockAttestationFlow();
 
-  MOCK_METHOD2(GetCertificate, void(const std::string&,
+  MOCK_METHOD3(GetCertificate, void(AttestationCertificateProfile,
+                                    bool,
                                     const CertificateCallback&));
 };
 
 }  // namespace attestation
 }  // namespace chromeos
 
 #endif  // CHROMEOS_ATTESTATION_MOCK_ATTESTATION_FLOW_H_