File: test_shared_storage_observer.h

package info (click to toggle)
chromium 138.0.7204.183-1~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-proposed-updates
  • size: 6,080,960 kB
  • sloc: cpp: 34,937,079; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,954; asm: 946,768; xml: 739,971; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,811; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (116 lines) | stat: -rw-r--r-- 4,179 bytes parent folder | download | duplicates (2)
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Copyright 2025 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_BROWSER_SHARED_STORAGE_TEST_SHARED_STORAGE_OBSERVER_H_
#define CONTENT_BROWSER_SHARED_STORAGE_TEST_SHARED_STORAGE_OBSERVER_H_

#include <optional>
#include <ostream>
#include <string>
#include <tuple>
#include <vector>

#include "base/time/time.h"
#include "content/browser/shared_storage/shared_storage_event_params.h"
#include "content/browser/shared_storage/shared_storage_runtime_manager.h"
#include "content/public/browser/global_routing_id.h"
#include "third_party/blink/public/common/shared_storage/shared_storage_utils.h"
#include "url/gurl.h"

namespace content {

class TestSharedStorageObserver
    : public SharedStorageRuntimeManager::SharedStorageObserverInterface {
 public:
  using AccessScope = blink::SharedStorageAccessScope;

  struct Access {
    AccessScope scope;
    AccessMethod method;
    GlobalRenderFrameHostId main_frame_id;
    std::string owner_origin;
    SharedStorageEventParams params;
    friend bool operator==(const Access& lhs, const Access& rhs);
  };

  struct OperationFinishedInfo {
    base::TimeDelta execution_time;
    AccessMethod method;
    int operation_id;
    int worklet_ordinal_id;
    base::UnguessableToken worklet_devtools_token;
    GlobalRenderFrameHostId main_frame_id;
    std::string owner_origin;
    OperationFinishedInfo();
    OperationFinishedInfo(base::TimeDelta execution_time,
                          AccessMethod method,
                          int operation_id,
                          int worklet_ordinal_id,
                          const base::UnguessableToken& worklet_devtools_token,
                          GlobalRenderFrameHostId main_frame_id,
                          std::string owner_origin);
    OperationFinishedInfo(const OperationFinishedInfo&);
    OperationFinishedInfo(OperationFinishedInfo&&);
    ~OperationFinishedInfo();
    OperationFinishedInfo& operator=(const OperationFinishedInfo&);
    OperationFinishedInfo& operator=(OperationFinishedInfo&&);
  };

  TestSharedStorageObserver();
  ~TestSharedStorageObserver() override;

  GlobalRenderFrameHostId AssociatedFrameHostId() const override;

  bool ShouldReceiveAllSharedStorageReports() const override;

  void OnSharedStorageAccessed(base::Time access_time,
                               AccessScope scope,
                               AccessMethod method,
                               GlobalRenderFrameHostId main_frame_id,
                               const std::string& owner_origin,
                               const SharedStorageEventParams& params) override;

  void OnSharedStorageSelectUrlUrnUuidGenerated(const GURL& urn_uuid) override;

  void OnSharedStorageSelectUrlConfigPopulated(
      const std::optional<FencedFrameConfig>& config) override;

  void OnSharedStorageWorkletOperationExecutionFinished(
      base::Time finished_time,
      base::TimeDelta execution_time,
      AccessMethod method,
      int operation_id,
      int worklet_ordinal_id,
      const base::UnguessableToken& worklet_devtools_token,
      GlobalRenderFrameHostId main_frame_id,
      const std::string& owner_origin) override;

  void ExpectAccessObserved(const std::vector<Access>& expected_accesses);

  void ExpectOperationFinishedInfosObserved(
      const std::vector<OperationFinishedInfo>& expected_infos);

  const std::vector<GURL>& urn_uuids_observed() const {
    return urn_uuids_observed_;
  }

 private:
  std::vector<Access> accesses_;
  std::vector<GURL> urn_uuids_observed_;
  std::vector<OperationFinishedInfo> operation_finished_infos_;
};

std::ostream& operator<<(std::ostream& os,
                         const TestSharedStorageObserver::Access& access);

bool operator==(const TestSharedStorageObserver::OperationFinishedInfo& lhs,
                const TestSharedStorageObserver::OperationFinishedInfo& rhs);

std::ostream& operator<<(
    std::ostream& os,
    const TestSharedStorageObserver::OperationFinishedInfo& info);

}  // namespace content

#endif  // CONTENT_BROWSER_SHARED_STORAGE_TEST_SHARED_STORAGE_OBSERVER_H_