// Copyright 2015 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 "chrome/browser/extensions/extension_install_prompt_test_helper.h" #include "base/bind.h" #include "base/callback_helpers.h" #include "testing/gtest/include/gtest/gtest.h" ExtensionInstallPromptTestHelper::ExtensionInstallPromptTestHelper() {} ExtensionInstallPromptTestHelper::ExtensionInstallPromptTestHelper( const base::Closure& quit_closure) : quit_closure_(quit_closure) {} ExtensionInstallPromptTestHelper::~ExtensionInstallPromptTestHelper() {} ExtensionInstallPrompt::DoneCallback ExtensionInstallPromptTestHelper::GetCallback() { return base::Bind(&ExtensionInstallPromptTestHelper::HandleResult, base::Unretained(this)); } ExtensionInstallPrompt::Result ExtensionInstallPromptTestHelper::result() const { if (!result_.get()) { ADD_FAILURE() << "Result was never set!"; return ExtensionInstallPrompt::Result::ACCEPTED; // Avoid crashing. } return *result_; } void ExtensionInstallPromptTestHelper::HandleResult( ExtensionInstallPrompt::Result result) { if (result_.get()) ADD_FAILURE() << "HandleResult() called twice!"; if (!quit_closure_.is_null()) base::ResetAndReturn(&quit_closure_).Run(); result_.reset(new ExtensionInstallPrompt::Result(result)); }