PreloadCheck class for extension pre-install checks

extensions/browser/preload_check_test_util.cc

Index: extensions/browser/preload_check_test_util.cc
diff --git a/extensions/browser/preload_check_test_util.cc b/extensions/browser/preload_check_test_util.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b12bc17950de05927ec82b86c69a8736e5b65232
--- /dev/null
+++ b/extensions/browser/preload_check_test_util.cc
@@ -0,0 +1,50 @@
+// Copyright 2017 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 "extensions/browser/preload_check_test_util.h"
+
+#include "base/bind.h"
+#include "base/bind_helpers.h"
+#include "base/logging.h"
+#include "base/memory/ptr_util.h"
+#include "base/run_loop.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace extensions {
+
+PreloadCheckRunner::PreloadCheckRunner() : called_(false) {}
+PreloadCheckRunner::~PreloadCheckRunner() {}
+
+void PreloadCheckRunner::OnCheckComplete(PreloadCheck::Errors errors) {
+  ASSERT_FALSE(called_);
+  called_ = true;
+  errors_ = errors;
+  if (run_loop_)
+    run_loop_->Quit();
+}
+
+void PreloadCheckRunner::Run(PreloadCheck* check) {
+  check->Start(GetCallback());
+}
+
+void PreloadCheckRunner::RunUntilComplete(PreloadCheck* check) {
+  Run(check);
+  ASSERT_FALSE(called_);
+
+  run_loop_ = base::MakeUnique<base::RunLoop>();
+  run_loop_->Run();
+  ASSERT_TRUE(called_);
+}
+
+PreloadCheck::ResultCallback PreloadCheckRunner::GetCallback() {
+  return base::Bind(&PreloadCheckRunner::OnCheckComplete,
+                    base::Unretained(this));
+}
+
+void PreloadCheckRunner::WaitForIdle() {
+  run_loop_ = base::MakeUnique<base::RunLoop>();
+  run_loop_->RunUntilIdle();
+}
+
+}  // namespace extensions