forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathresource_load_observer.h
87 lines (69 loc) · 2.84 KB
/
resource_load_observer.h
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// Copyright 2019 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_RESOURCE_LOAD_OBSERVER_H_
#define CONTENT_TEST_RESOURCE_LOAD_OBSERVER_H_
#include <string>
#include <vector>
#include "base/callback_forward.h"
#include "base/files/file_util.h"
#include "base/time/time.h"
#include "content/public/browser/web_contents_observer.h"
#include "content/shell/browser/shell.h"
#include "third_party/blink/public/mojom/loader/resource_load_info.mojom.h"
#include "url/gurl.h"
namespace content {
// Observer class to track resource loads.
class ResourceLoadObserver : public WebContentsObserver {
public:
explicit ResourceLoadObserver(Shell* shell);
ResourceLoadObserver(const ResourceLoadObserver&) = delete;
ResourceLoadObserver& operator=(const ResourceLoadObserver&) = delete;
~ResourceLoadObserver() override;
const std::vector<blink::mojom::ResourceLoadInfoPtr>& resource_load_infos()
const {
return resource_load_infos_;
}
const std::vector<bool>& resource_is_associated_with_main_frame() const {
return resource_is_associated_with_main_frame_;
}
const std::vector<GURL>& memory_cached_loaded_urls() const {
return memory_cached_loaded_urls_;
}
// Use this method with the SCOPED_TRACE macro, so it shows the caller context
// if it fails.
void CheckResourceLoaded(
const GURL& original_url,
const GURL& referrer,
const std::string& load_method,
network::mojom::RequestDestination request_destination,
const base::FilePath::StringPieceType& served_file_name,
const std::string& mime_type,
const std::string& ip_address,
bool was_cached,
bool first_network_request,
const base::TimeTicks& before_request,
const base::TimeTicks& after_request);
// Returns the resource with the given url if found, otherwise nullptr.
blink::mojom::ResourceLoadInfoPtr* GetResource(const GURL& original_url);
void Reset();
void WaitForResourceCompletion(const GURL& original_url);
private:
// WebContentsObserver implementation:
void ResourceLoadComplete(
content::RenderFrameHost* render_frame_host,
const GlobalRequestID& request_id,
const blink::mojom::ResourceLoadInfo& resource_load_info) override;
void DidLoadResourceFromMemoryCache(
content::RenderFrameHost* render_frame_host,
const GURL& url,
const std::string& mime_type,
network::mojom::RequestDestination request_destination) override;
std::vector<GURL> memory_cached_loaded_urls_;
std::vector<blink::mojom::ResourceLoadInfoPtr> resource_load_infos_;
std::vector<bool> resource_is_associated_with_main_frame_;
GURL waiting_original_url_;
base::OnceClosure waiting_callback_;
};
} // namespace content
#endif // CONTENT_TEST_RESOURCE_LOAD_OBSERVER_H_