Enhanced and refactored the AttestationFlow interface.

chrome/browser/chromeos/attestation/attestation_policy_observer_unittest.cc

 // Copyright (c) 2013 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.
 
 #include "base/bind.h"
 #include "base/message_loop.h"
 #include "base/run_loop.h"
 #include "chrome/browser/chromeos/attestation/attestation_policy_observer.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/settings/cros_settings_names.h"
@@ skipping 83 matching lines @@
 }
 
 TEST_F(AttestationPolicyObserverTest, UnregisteredPolicyClient) {
   policy_client_.SetDMToken("");
   Run();
 }
 
 TEST_F(AttestationPolicyObserverTest, NewCertificate) {
   EXPECT_CALL(cryptohome_client_, TpmAttestationDoesKeyExist(_, _, _))
       .WillOnce(WithArgs<2>(Invoke(DBusCallbackFalse)));
-  EXPECT_CALL(attestation_flow_, GetCertificate(_, _))
-      .WillOnce(WithArgs<1>(Invoke(CertCallbackSuccess)));
+  EXPECT_CALL(attestation_flow_, GetCertificate(_, _, _))
+      .WillOnce(WithArgs<2>(Invoke(CertCallbackSuccess)));
   Run();
 }
 
 TEST_F(AttestationPolicyObserverTest, KeyExists) {
   EXPECT_CALL(cryptohome_client_, TpmAttestationDoesKeyExist(_, _, _))
       .WillOnce(WithArgs<2>(Invoke(DBusCallbackTrue)));
   EXPECT_CALL(cryptohome_client_, TpmAttestationGetCertificate(_, _, _))
       .WillOnce(WithArgs<2>(Invoke(DBusDataCallback)));
   Run();
 }
 
 }  // namespace attestation
 }  // namespace chromeos