content/test/test_navigation_url_loader.h

   1 // Copyright (c) 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_TEST_TEST_NAVIGATION_URL_LOADER_H_
   6 #define CONTENT_TEST_TEST_NAVIGATION_URL_LOADER_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "content/browser/frame_host/navigation_request_info.h"
  11 #include "content/browser/loader/navigation_url_loader.h"
  12 #include "content/common/navigation_params.h"
  13
  14 namespace net {
  15 struct RedirectInfo;
  16 }
  17
  18 namespace content {
  19
  20 class NavigationURLLoaderDelegate;
  21 class StreamHandle;
  22 struct ResourceResponse;
  23
  24 // PlzNavigate
  25 // Test implementation of NavigationURLLoader to simulate the network stack
  26 // response.
  27 class TestNavigationURLLoader
  28     : public NavigationURLLoader,
  29       public base::SupportsWeakPtr<TestNavigationURLLoader> {
  30  public:
  31   TestNavigationURLLoader(scoped_ptr<NavigationRequestInfo> request_info,
  32                           NavigationURLLoaderDelegate* delegate);
  33
  34   // NavigationURLLoader implementation.
  35   void FollowRedirect() override;
  36
  37   NavigationRequestInfo* request_info() const { return request_info_.get(); }
  38
  39   void SimulateServerRedirect(const GURL& redirect_url);
  40
  41   void CallOnRequestRedirected(const net::RedirectInfo& redirect_info,
  42                                const scoped_refptr<ResourceResponse>& response);
  43   void CallOnResponseStarted(const scoped_refptr<ResourceResponse>& response,
  44                              scoped_ptr<StreamHandle> body);
  45
  46   int redirect_count() { return redirect_count_; }
  47
  48  private:
  49   ~TestNavigationURLLoader() override;
  50
  51   scoped_ptr<NavigationRequestInfo> request_info_;
  52   NavigationURLLoaderDelegate* delegate_;
  53   int redirect_count_;
  54 };
  55
  56 }  // namespace content
  57
  58 #endif  // CONTENT_TEST_TEST_NAVIGATION_URL_LOADER_H_