Remove CanChallengePlatform() and CheckPlatformState().

chrome/browser/chromeos/attestation/platform_verification_flow_unittest.cc

 // Copyright 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 <string>
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/testing_pref_service.h"
 #include "base/run_loop.h"
 #include "chrome/browser/chromeos/attestation/attestation_signed_data.pb.h"
 #include "chrome/browser/chromeos/attestation/platform_verification_flow.h"
 #include "chrome/browser/chromeos/login/mock_user_manager.h"
 #include "chrome/browser/chromeos/settings/cros_settings.h"
 #include "chrome/browser/chromeos/settings/device_settings_service.h"
 #include "chrome/browser/chromeos/settings/stub_cros_settings_provider.h"
 #include "chrome/common/pref_names.h"
 #include "chromeos/attestation/mock_attestation_flow.h"
 #include "chromeos/cryptohome/mock_async_method_caller.h"
 #include "chromeos/dbus/fake_cryptohome_client.h"
 #include "chromeos/settings/cros_settings_names.h"
-#include "chromeos/system/mock_statistics_provider.h"
 #include "content/public/test/test_browser_thread.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::DoAll;
 using testing::Invoke;
 using testing::Return;
 using testing::SetArgumentPointee;
 using testing::StrictMock;
 using testing::WithArgs;
@@ skipping 83 matching lines @@
 
 }  // namespace
 
 class PlatformVerificationFlowTest : public ::testing::Test {
  public:
   PlatformVerificationFlowTest()
       : message_loop_(base::MessageLoop::TYPE_UI),
         ui_thread_(content::BrowserThread::UI, &message_loop_),
         certificate_success_(true),
         sign_challenge_success_(true),
-        result_(PlatformVerificationFlow::INTERNAL_ERROR),
-        check_state_result_(false) {}
+        result_(PlatformVerificationFlow::INTERNAL_ERROR) {}
 
   void SetUp() {
     // Configure a user for the mock user manager.
     mock_user_manager_.SetActiveUser(kTestEmail);
 
-    // Configure the statistics provider to report verified mode.
-    EXPECT_CALL(mock_statistics_provider_,
-                GetMachineStatistic(system::kDevSwitchBootMode, _))
-        .WillRepeatedly(DoAll(SetArgumentPointee<1>(std::string("0")),
-                              Return(true)));
-
     // Create a verifier for tests to call.
     verifier_.reset(new PlatformVerificationFlow(&mock_attestation_flow_,
                                                  &mock_async_caller_,
                                                  &fake_cryptohome_client_,
                                                  &mock_user_manager_,
-                                                 &mock_statistics_provider_,
                                                  &fake_delegate_));
 
     // Create callbacks for tests to use with verifier_.
     callback_ = base::Bind(&PlatformVerificationFlowTest::FakeChallengeCallback,
                            base::Unretained(this));
-    check_state_callback_ = base::Bind(
-        &PlatformVerificationFlowTest::FakeCheckStateCallback,
-        base::Unretained(this));
 
     // Configure the test pref service.
     pref_service_.registry()->RegisterBooleanPref(prefs::kEnableDRM, true);
     pref_service_.registry()->RegisterBooleanPref(prefs::kRAConsentFirstTime,
                                                   true);
     pref_service_.registry()->RegisterBooleanPref(prefs::kRAConsentAlways,
                                                   false);
     pref_service_.registry()->RegisterDictionaryPref(prefs::kRAConsentDomains);
     verifier_->set_testing_prefs(&pref_service_);
 
@@ skipping 59 matching lines @@
   void FakeChallengeCallback(PlatformVerificationFlow::Result result,
                              const std::string& salt,
                              const std::string& signature,
                              const std::string& certificate) {
     result_ = result;
     challenge_salt_ = salt;
     challenge_signature_ = signature;
     certificate_ = certificate;
   }
 
-  void FakeCheckStateCallback(bool result) {
-    check_state_result_ = result;
-  }
-
   std::string CreateFakeResponseProto() {
     SignedData pb;
     pb.set_data(kTestSignedData);
     pb.set_signature(kTestSignature);
     std::string serial;
     CHECK(pb.SerializeToString(&serial));
     return serial;
   }
 
  protected:
   base::MessageLoop message_loop_;
   content::TestBrowserThread ui_thread_;
   StrictMock<MockAttestationFlow> mock_attestation_flow_;
   cryptohome::MockAsyncMethodCaller mock_async_caller_;
   CustomFakeCryptohomeClient fake_cryptohome_client_;
   MockUserManager mock_user_manager_;
-  system::MockStatisticsProvider mock_statistics_provider_;
   FakeDelegate fake_delegate_;
   TestingPrefServiceSimple pref_service_;
   CrosSettingsProvider* device_settings_provider_;
   StubCrosSettingsProvider stub_settings_provider_;
   ScopedTestDeviceSettingsService test_device_settings_service_;
   ScopedTestCrosSettings test_cros_settings_;
   scoped_ptr<PlatformVerificationFlow> verifier_;
 
   // Controls result of FakeGetCertificate.
   bool certificate_success_;
 
   // Controls result of FakeSignChallenge.
   bool sign_challenge_success_;
 
   // Callback functions and data.
   PlatformVerificationFlow::ChallengeCallback callback_;
   PlatformVerificationFlow::Result result_;
   std::string challenge_salt_;
   std::string challenge_signature_;
   std::string certificate_;
-  base::Callback<void(bool result)> check_state_callback_;
-  bool check_state_result_;
 };
 
 TEST_F(PlatformVerificationFlowTest, SuccessNoConsent) {
   // Make sure the call will fail if consent is requested.
   fake_delegate_.set_response(PlatformVerificationFlow::CONSENT_RESPONSE_DENY);
   ExpectAttestationFlow();
   verifier_->ChallengePlatformKey(NULL, kTestID, kTestChallenge, callback_);
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(PlatformVerificationFlow::SUCCESS, result_);
   EXPECT_EQ(kTestSignedData, challenge_salt_);
@@ skipping 103 matching lines @@
 
 TEST_F(PlatformVerificationFlowTest, ConsentNoResponse) {
   fake_delegate_.set_response(PlatformVerificationFlow::CONSENT_RESPONSE_NONE);
   pref_service_.SetUserPref(prefs::kRAConsentAlways,
                             new base::FundamentalValue(true));
   verifier_->ChallengePlatformKey(NULL, kTestID, kTestChallenge, callback_);
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(PlatformVerificationFlow::USER_REJECTED, result_);
 }
 
-TEST_F(PlatformVerificationFlowTest, FastCheck) {
-  verifier_->CheckPlatformState(check_state_callback_);
-  base::RunLoop().RunUntilIdle();
-  EXPECT_TRUE(check_state_result_);
-}
-
-TEST_F(PlatformVerificationFlowTest, FastCheckNoStat) {
-  // Configure the stats provider to fail.
-  EXPECT_CALL(mock_statistics_provider_,
-              GetMachineStatistic(system::kDevSwitchBootMode, _))
-      .WillRepeatedly(Return(false));
-
-  verifier_->CheckPlatformState(check_state_callback_);
-  base::RunLoop().RunUntilIdle();
-  EXPECT_FALSE(check_state_result_);
-}
-
-TEST_F(PlatformVerificationFlowTest, FastCheckStatDevMode) {
-  // Configure the stats provider to fail.
-  EXPECT_CALL(mock_statistics_provider_,
-              GetMachineStatistic(system::kDevSwitchBootMode, _))
-      .WillRepeatedly(DoAll(SetArgumentPointee<1>(std::string("1")),
-                            Return(true)));
-
-  verifier_->CheckPlatformState(check_state_callback_);
-  base::RunLoop().RunUntilIdle();
-  EXPECT_FALSE(check_state_result_);
-}
-
-TEST_F(PlatformVerificationFlowTest, FastCheckStatInvalidMode) {
-  // Configure the stats provider to fail.
-  EXPECT_CALL(mock_statistics_provider_,
-              GetMachineStatistic(system::kDevSwitchBootMode, _))
-      .WillRepeatedly(DoAll(SetArgumentPointee<1>(std::string("INVALID")),
-                            Return(true)));
-
-  verifier_->CheckPlatformState(check_state_callback_);
-  base::RunLoop().RunUntilIdle();
-  EXPECT_FALSE(check_state_result_);
-}
-
-TEST_F(PlatformVerificationFlowTest, FastCheckNoAttestation) {
-  fake_cryptohome_client_.set_attestation_prepared(false);
-  verifier_->CheckPlatformState(check_state_callback_);
-  base::RunLoop().RunUntilIdle();
-  EXPECT_FALSE(check_state_result_);
-}
-
-TEST_F(PlatformVerificationFlowTest, FastCheckDBusFailure) {
-  fake_cryptohome_client_.set_call_status(DBUS_METHOD_CALL_FAILURE);
-  verifier_->CheckPlatformState(check_state_callback_);
-  base::RunLoop().RunUntilIdle();
-  EXPECT_FALSE(check_state_result_);
-}
-
 }  // namespace attestation
 }  // namespace chromeos