PreloadCheck class for extension pre-install checks

extensions/browser/preload_check_test_util.h

+// 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.
+
+#ifndef EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_
+#define EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_
+
+#include <memory>
+
+#include "extensions/browser/preload_check.h"
+
+namespace base {
+class RunLoop;
+}
+
+namespace extensions {
+
+// Provides a callback method for a PreloadCheck that stores its results.
+class PreloadCheckRunner {
+ public:
+  PreloadCheckRunner();
+  virtual ~PreloadCheckRunner();
+
+  // Starts the check, providing OnCheckComplete as the callback.
+  void Run(PreloadCheck* check);
+
+  // Starts the check and waits for its callback to execute.
+  void RunUntilComplete(PreloadCheck* check);
+
+  // Runs the message loop until OnCheckComplete is called.
+  void WaitForCompletion();
+
+  // Runs the message loop until idle. Useful to see whether OnCheckComplete is
+  // called without waiting indefinitely.
+  void WaitForIdle();
+
+  PreloadCheck::ResultCallback GetCallback();
+
+  const PreloadCheck::Errors& errors() const { return errors_; }
+  bool called() const { return called_; }
+
+ private:
+  void OnCheckComplete(PreloadCheck::Errors errors);
+
+  PreloadCheck::Errors errors_;
+  bool called_;
+
+  // Using a RunLoop data member would trigger tricky timing troubles.
+  std::unique_ptr<base::RunLoop> run_loop_;
+
+  DISALLOW_COPY_AND_ASSIGN(PreloadCheckRunner);
+};
+
+}  // namespace extensions
+
+#endif  // EXTENSIONS_BROWSER_PRELOAD_CHECK_TEST_UTIL_H_