1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CONTENT_TEST_TASK_RUNNER_DEFERRING_THROTTLE_H_
#define CONTENT_TEST_TASK_RUNNER_DEFERRING_THROTTLE_H_
#include "base/test/test_simple_task_runner.h"
#include "content/public/browser/navigation_throttle.h"
namespace content {
// This class defers a navigation via a no-op async task on the provided task
// runner.
class TaskRunnerDeferringThrottle : public NavigationThrottle {
public:
TaskRunnerDeferringThrottle(scoped_refptr<base::TaskRunner> task_runner,
bool defer_start,
bool defer_redirect,
bool defer_response,
NavigationThrottleRegistry& registry);
TaskRunnerDeferringThrottle(const TaskRunnerDeferringThrottle&) = delete;
TaskRunnerDeferringThrottle& operator=(const TaskRunnerDeferringThrottle&) =
delete;
~TaskRunnerDeferringThrottle() override;
static void Create(
scoped_refptr<base::TaskRunner> task_runner,
bool defer_start,
bool defer_redirect,
bool defer_response,
NavigationThrottleRegistry& registry);
// NavigationThrottle:
ThrottleCheckResult WillStartRequest() override;
ThrottleCheckResult WillRedirectRequest() override;
ThrottleCheckResult WillProcessResponse() override;
const char* GetNameForLogging() override;
protected:
// Tests may override this to change how/when Resume() is called.
virtual void AsyncResume();
private:
ThrottleCheckResult DeferToPostTask();
bool defer_start_ = true;
bool defer_redirect_ = true;
bool defer_response_ = true;
scoped_refptr<base::TaskRunner> task_runner_;
base::WeakPtrFactory<TaskRunnerDeferringThrottle> weak_factory_{this};
};
} // namespace content
#endif // CONTENT_TEST_TASK_RUNNER_DEFERRING_THROTTLE_H_
|
