Perform navigation policy check on UI thread for --site-per-process.

content/test/test_frame_navigation_observer.h

+// Copyright 2014 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 CONTENT_TEST_TEST_FRAME_NAVIGATION_OBSERVER_H_
+#define CONTENT_TEST_TEST_FRAME_NAVIGATION_OBSERVER_H_
+
+#include <set>
+
+#include "base/callback.h"
+#include "base/compiler_specific.h"
+#include "base/memory/scoped_ptr.h"
+#include "content/public/browser/web_contents_observer.h"
+#include "content/public/test/test_utils.h"
+
+class GURL;
+
+namespace content {
+class FrameTreeNode;
+class RenderFrameHostImpl;
+class WebContents;
+struct LoadCommittedDetails;
+
+// For content_browsertests, which run on the UI thread, run a second
+// MessageLoop and quit when the navigation in a specific frame completes
+// loading.
+class TestFrameNavigationObserver : public WebContentsObserver {
+ public:
+  // Create and register a new TestFrameNavigationObserver which will track
+  // navigations performed in the specified |node| of the frame tree.
+  TestFrameNavigationObserver(FrameTreeNode* node, int number_of_navigations);
+  // As above but waits for one navigation.
+  explicit TestFrameNavigationObserver(FrameTreeNode* node);
+
+  virtual ~TestFrameNavigationObserver();
+
+  // Runs a nested message loop and blocks until the expected number of
+  // navigations are complete.
+  void Wait();
+
+ private:
+  // WebContentsObserver
+  virtual void DidStartProvisionalLoadForFrame(
+      int64 frame_id,
+      int64 parent_frame_id,
+      bool is_main_frame,
+      const GURL& validated_url,
+      bool is_error_page,
+      bool is_iframe_srcdoc,
+      RenderViewHost* render_view_host) OVERRIDE;
+  virtual void DidNavigateAnyFrame(
+      const LoadCommittedDetails& details,
+      const FrameNavigateParams& params) OVERRIDE;
+
+  // The id of the FrameTreeNode in which navigations are peformed.
+  int frame_tree_node_id_;
+
+  // If true the navigation has started.
+  bool navigation_started_;
+
+  // The number of navigations that have been completed.
+  int navigations_completed_;
+
+  // The number of navigations to wait for.
+  int number_of_navigations_;
+
+  // The MessageLoopRunner used to spin the message loop.
+  scoped_refptr<MessageLoopRunner> message_loop_runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestFrameNavigationObserver);
+};
+
+}  // namespace content
+
+#endif  // CONTENT_TEST_TEST_FRAME_NAVIGATION_OBSERVER_H_