Commit 92d94434 authored by Leo Lai's avatar Leo Lai Committed by Commit Bot

remove FakeCryptohomeClient from AttestationFlow unittests.

It is not necessary anymore after attestation's GetKeyInfo is used.

BUG=b:158955123
TEST=chromeos_unittests.

Change-Id: I4d288c8de7227b94941da116267a3253f4a71439
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2513950
Commit-Queue: Leo Lai <cylai@google.com>
Reviewed-by: default avatarMaksim Ivanov <emaxx@chromium.org>
Cr-Commit-Position: refs/heads/master@{#823884}
parent 0434aa04
...@@ -17,7 +17,6 @@ ...@@ -17,7 +17,6 @@
#include "chromeos/cryptohome/cryptohome_parameters.h" #include "chromeos/cryptohome/cryptohome_parameters.h"
#include "chromeos/cryptohome/mock_async_method_caller.h" #include "chromeos/cryptohome/mock_async_method_caller.h"
#include "chromeos/dbus/attestation/attestation_client.h" #include "chromeos/dbus/attestation/attestation_client.h"
#include "chromeos/dbus/cryptohome/fake_cryptohome_client.h"
#include "components/account_id/account_id.h" #include "components/account_id/account_id.h"
#include "testing/gmock/include/gmock/gmock.h" #include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h" #include "testing/gtest/include/gtest/gtest.h"
...@@ -78,7 +77,6 @@ TEST_F(AttestationFlowTest, GetCertificate) { ...@@ -78,7 +77,6 @@ TEST_F(AttestationFlowTest, GetCertificate) {
// Verify the order of calls in a sequence. // Verify the order of calls in a sequence.
Sequence flow_order; Sequence flow_order;
FakeCryptohomeClient client;
// Set the enrollment status as `false` so the full enrollment flow is // Set the enrollment status as `false` so the full enrollment flow is
// triggered. // triggered.
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
...@@ -156,7 +154,8 @@ TEST_F(AttestationFlowTest, GetCertificate) { ...@@ -156,7 +154,8 @@ TEST_F(AttestationFlowTest, GetCertificate) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, account_id, flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, account_id,
"fake_origin", true, std::string() /* key_name */, "fake_origin", true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -169,7 +168,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Ecc) { ...@@ -169,7 +168,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Ecc) {
// Verify the order of calls in a sequence. // Verify the order of calls in a sequence.
Sequence flow_order; Sequence flow_order;
FakeCryptohomeClient client;
// Set the enrollment status as `false` so the full enrollment flow is // Set the enrollment status as `false` so the full enrollment flow is
// triggered. // triggered.
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
...@@ -247,8 +245,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Ecc) { ...@@ -247,8 +245,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Ecc) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface), AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
::attestation::KEY_TYPE_ECC); std::move(proxy_interface), ::attestation::KEY_TYPE_ECC);
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, account_id, flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, account_id,
"fake_origin", true, std::string() /* key_name */, "fake_origin", true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -259,7 +257,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Not_Prepared) { ...@@ -259,7 +257,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Not_Prepared) {
// Verify the order of calls in a sequence. // Verify the order of calls in a sequence.
Sequence flow_order; Sequence flow_order;
FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -336,7 +333,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Not_Prepared) { ...@@ -336,7 +333,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Not_Prepared) {
base::Unretained(&observer))); base::Unretained(&observer)));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.set_retry_delay(base::TimeDelta::FromMilliseconds(30)); flow.set_retry_delay(base::TimeDelta::FromMilliseconds(30));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, account_id, flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, account_id,
"fake_origin", true, std::string() /* key_name */, "fake_origin", true, std::string() /* key_name */,
...@@ -349,7 +347,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Prepared) { ...@@ -349,7 +347,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Prepared) {
StrictMock<cryptohome::MockAsyncMethodCaller> async_caller; StrictMock<cryptohome::MockAsyncMethodCaller> async_caller;
async_caller.SetUp(false, cryptohome::MOUNT_ERROR_NONE); async_caller.SetUp(false, cryptohome::MOUNT_ERROR_NONE);
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -373,7 +370,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Prepared) { ...@@ -373,7 +370,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Prepared) {
base::Unretained(&observer))); base::Unretained(&observer)));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.set_ready_timeout(base::TimeDelta::FromMilliseconds(20)); flow.set_ready_timeout(base::TimeDelta::FromMilliseconds(20));
flow.set_retry_delay(base::TimeDelta::FromMilliseconds(6)); flow.set_retry_delay(base::TimeDelta::FromMilliseconds(6));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(),
...@@ -387,7 +385,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Confirm_Prepared) { ...@@ -387,7 +385,6 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Confirm_Prepared) {
StrictMock<cryptohome::MockAsyncMethodCaller> async_caller; StrictMock<cryptohome::MockAsyncMethodCaller> async_caller;
async_caller.SetUp(false, cryptohome::MOUNT_ERROR_NONE); async_caller.SetUp(false, cryptohome::MOUNT_ERROR_NONE);
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -412,7 +409,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Confirm_Prepared) { ...@@ -412,7 +409,8 @@ TEST_F(AttestationFlowTest, GetCertificate_Attestation_Never_Confirm_Prepared) {
base::Unretained(&observer))); base::Unretained(&observer)));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.set_ready_timeout(base::TimeDelta::FromMilliseconds(20)); flow.set_ready_timeout(base::TimeDelta::FromMilliseconds(20));
flow.set_retry_delay(base::TimeDelta::FromMilliseconds(6)); flow.set_retry_delay(base::TimeDelta::FromMilliseconds(6));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(),
...@@ -428,7 +426,6 @@ TEST_F(AttestationFlowTest, GetCertificate_NoEK) { ...@@ -428,7 +426,6 @@ TEST_F(AttestationFlowTest, GetCertificate_NoEK) {
EXPECT_CALL(async_caller, AsyncTpmAttestationCreateEnrollRequest(_, _)) EXPECT_CALL(async_caller, AsyncTpmAttestationCreateEnrollRequest(_, _))
.Times(1); .Times(1);
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -449,7 +446,8 @@ TEST_F(AttestationFlowTest, GetCertificate_NoEK) { ...@@ -449,7 +446,8 @@ TEST_F(AttestationFlowTest, GetCertificate_NoEK) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -462,7 +460,6 @@ TEST_F(AttestationFlowTest, GetCertificate_EKRejected) { ...@@ -462,7 +460,6 @@ TEST_F(AttestationFlowTest, GetCertificate_EKRejected) {
EXPECT_CALL(async_caller, AsyncTpmAttestationCreateEnrollRequest(_, _)) EXPECT_CALL(async_caller, AsyncTpmAttestationCreateEnrollRequest(_, _))
.Times(1); .Times(1);
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -488,7 +485,8 @@ TEST_F(AttestationFlowTest, GetCertificate_EKRejected) { ...@@ -488,7 +485,8 @@ TEST_F(AttestationFlowTest, GetCertificate_EKRejected) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -507,7 +505,6 @@ TEST_F(AttestationFlowTest, GetCertificate_FailEnroll) { ...@@ -507,7 +505,6 @@ TEST_F(AttestationFlowTest, GetCertificate_FailEnroll) {
AsyncTpmAttestationEnroll(_, fake_enroll_response, _)) AsyncTpmAttestationEnroll(_, fake_enroll_response, _))
.WillOnce(WithArgs<2>(Invoke(AsyncCallbackFalse))); .WillOnce(WithArgs<2>(Invoke(AsyncCallbackFalse)));
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -533,7 +530,8 @@ TEST_F(AttestationFlowTest, GetCertificate_FailEnroll) { ...@@ -533,7 +530,8 @@ TEST_F(AttestationFlowTest, GetCertificate_FailEnroll) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -569,8 +567,6 @@ TEST_F(AttestationFlowTest, GetMachineCertificateAlreadyEnrolled) { ...@@ -569,8 +567,6 @@ TEST_F(AttestationFlowTest, GetMachineCertificateAlreadyEnrolled) {
kEnterpriseMachineKey, _)) kEnterpriseMachineKey, _))
.Times(1); .Times(1);
chromeos::FakeCryptohomeClient client;
std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>()); std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>());
proxy->DeferToFake(true); proxy->DeferToFake(true);
EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault()); EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault());
...@@ -590,7 +586,8 @@ TEST_F(AttestationFlowTest, GetMachineCertificateAlreadyEnrolled) { ...@@ -590,7 +586,8 @@ TEST_F(AttestationFlowTest, GetMachineCertificateAlreadyEnrolled) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_MACHINE_CERTIFICATE, EmptyAccountId(), flow.GetCertificate(PROFILE_ENTERPRISE_MACHINE_CERTIFICATE, EmptyAccountId(),
"", true, std::string() /* key_name */, "", true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -625,8 +622,6 @@ TEST_F(AttestationFlowTest, GetEnrollmentCertificateAlreadyEnrolled) { ...@@ -625,8 +622,6 @@ TEST_F(AttestationFlowTest, GetEnrollmentCertificateAlreadyEnrolled) {
kEnterpriseEnrollmentKey, _)) kEnterpriseEnrollmentKey, _))
.Times(1); .Times(1);
chromeos::FakeCryptohomeClient client;
std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>()); std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>());
proxy->DeferToFake(true); proxy->DeferToFake(true);
EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault()); EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault());
...@@ -646,7 +641,8 @@ TEST_F(AttestationFlowTest, GetEnrollmentCertificateAlreadyEnrolled) { ...@@ -646,7 +641,8 @@ TEST_F(AttestationFlowTest, GetEnrollmentCertificateAlreadyEnrolled) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_ENROLLMENT_CERTIFICATE, flow.GetCertificate(PROFILE_ENTERPRISE_ENROLLMENT_CERTIFICATE,
EmptyAccountId(), "", true, std::string() /* key_name */, EmptyAccountId(), "", true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -671,8 +667,6 @@ TEST_F(AttestationFlowTest, GetCertificate_FailCreateCertRequest) { ...@@ -671,8 +667,6 @@ TEST_F(AttestationFlowTest, GetCertificate_FailCreateCertRequest) {
->mutable_certificate_request_reply() ->mutable_certificate_request_reply()
->set_status(::attestation::STATUS_UNEXPECTED_DEVICE_ERROR); ->set_status(::attestation::STATUS_UNEXPECTED_DEVICE_ERROR);
chromeos::FakeCryptohomeClient client;
// We're not expecting any server calls in this case; StrictMock will verify. // We're not expecting any server calls in this case; StrictMock will verify.
std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>()); std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>());
EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault()); EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault());
...@@ -685,7 +679,8 @@ TEST_F(AttestationFlowTest, GetCertificate_FailCreateCertRequest) { ...@@ -685,7 +679,8 @@ TEST_F(AttestationFlowTest, GetCertificate_FailCreateCertRequest) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -699,7 +694,6 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestRejected) { ...@@ -699,7 +694,6 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestRejected) {
->set_enrolled(true); ->set_enrolled(true);
StrictMock<cryptohome::MockAsyncMethodCaller> async_caller; StrictMock<cryptohome::MockAsyncMethodCaller> async_caller;
async_caller.SetUp(true, cryptohome::MOUNT_ERROR_NONE); async_caller.SetUp(true, cryptohome::MOUNT_ERROR_NONE);
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->AllowlistLegacyCreateCertificateRequest( ->AllowlistLegacyCreateCertificateRequest(
...@@ -729,7 +723,8 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestRejected) { ...@@ -729,7 +723,8 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestRejected) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -756,8 +751,6 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestBadRequest) { ...@@ -756,8 +751,6 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestBadRequest) {
->set_pca_request( ->set_pca_request(
cryptohome::MockAsyncMethodCaller::kFakeAttestationCertRequest); cryptohome::MockAsyncMethodCaller::kFakeAttestationCertRequest);
chromeos::FakeCryptohomeClient client;
std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>()); std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>());
proxy->DeferToFake(true); proxy->DeferToFake(true);
EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault()); EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault());
...@@ -781,7 +774,8 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestBadRequest) { ...@@ -781,7 +774,8 @@ TEST_F(AttestationFlowTest, GetCertificate_CertRequestBadRequest) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -792,7 +786,6 @@ TEST_F(AttestationFlowTest, GetCertificate_FailIsEnrolled) { ...@@ -792,7 +786,6 @@ TEST_F(AttestationFlowTest, GetCertificate_FailIsEnrolled) {
// We're not expecting any async calls in this case; StrictMock will verify. // We're not expecting any async calls in this case; StrictMock will verify.
StrictMock<cryptohome::MockAsyncMethodCaller> async_caller; StrictMock<cryptohome::MockAsyncMethodCaller> async_caller;
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->mutable_status_reply() ->mutable_status_reply()
...@@ -810,7 +803,8 @@ TEST_F(AttestationFlowTest, GetCertificate_FailIsEnrolled) { ...@@ -810,7 +803,8 @@ TEST_F(AttestationFlowTest, GetCertificate_FailIsEnrolled) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
true, std::string() /* key_name */, true, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -844,8 +838,6 @@ TEST_F(AttestationFlowTest, GetCertificate_CheckExisting) { ...@@ -844,8 +838,6 @@ TEST_F(AttestationFlowTest, GetCertificate_CheckExisting) {
kEnterpriseUserKey, _)) kEnterpriseUserKey, _))
.Times(1); .Times(1);
chromeos::FakeCryptohomeClient client;
std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>()); std::unique_ptr<MockServerProxy> proxy(new StrictMock<MockServerProxy>());
proxy->DeferToFake(true); proxy->DeferToFake(true);
EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault()); EXPECT_CALL(*proxy, GetType()).WillRepeatedly(DoDefault());
...@@ -865,7 +857,8 @@ TEST_F(AttestationFlowTest, GetCertificate_CheckExisting) { ...@@ -865,7 +857,8 @@ TEST_F(AttestationFlowTest, GetCertificate_CheckExisting) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
false, std::string() /* key_name */, false, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
...@@ -880,7 +873,6 @@ TEST_F(AttestationFlowTest, GetCertificate_AlreadyExists) { ...@@ -880,7 +873,6 @@ TEST_F(AttestationFlowTest, GetCertificate_AlreadyExists) {
// We're not expecting any async calls in this case; StrictMock will verify. // We're not expecting any async calls in this case; StrictMock will verify.
StrictMock<cryptohome::MockAsyncMethodCaller> async_caller; StrictMock<cryptohome::MockAsyncMethodCaller> async_caller;
chromeos::FakeCryptohomeClient client;
chromeos::AttestationClient::Get() chromeos::AttestationClient::Get()
->GetTestInterface() ->GetTestInterface()
->GetMutableKeyInfoReply("", kEnterpriseUserKey) ->GetMutableKeyInfoReply("", kEnterpriseUserKey)
...@@ -898,7 +890,8 @@ TEST_F(AttestationFlowTest, GetCertificate_AlreadyExists) { ...@@ -898,7 +890,8 @@ TEST_F(AttestationFlowTest, GetCertificate_AlreadyExists) {
&MockObserver::MockCertificateCallback, base::Unretained(&observer)); &MockObserver::MockCertificateCallback, base::Unretained(&observer));
std::unique_ptr<ServerProxy> proxy_interface(proxy.release()); std::unique_ptr<ServerProxy> proxy_interface(proxy.release());
AttestationFlow flow(&async_caller, &client, std::move(proxy_interface)); AttestationFlow flow(&async_caller, /*cryptohome_client=*/nullptr,
std::move(proxy_interface));
flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "", flow.GetCertificate(PROFILE_ENTERPRISE_USER_CERTIFICATE, EmptyAccountId(), "",
false, std::string() /* key_name */, false, std::string() /* key_name */,
std::move(mock_callback)); std::move(mock_callback));
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment