File: state_store_test_observer.cc

package info (click to toggle)
chromium 138.0.7204.183-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 6,071,908 kB
  • sloc: cpp: 34,937,088; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; 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,806; 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 (49 lines) | stat: -rw-r--r-- 1,648 bytes parent folder | download | duplicates (7)
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
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "extensions/browser/state_store_test_observer.h"

#include "base/containers/contains.h"
#include "content/public/browser/browser_thread.h"
#include "extensions/browser/extension_system.h"
#include "extensions/common/extension_id.h"

namespace extensions {

StateStoreTestObserver::StateStoreTestObserver(content::BrowserContext* context)
    : state_store_(ExtensionSystem::Get(context)->state_store()) {
  observed_.Observe(state_store_.get());
}

StateStoreTestObserver::~StateStoreTestObserver() = default;

void StateStoreTestObserver::WaitForExtensionAndKey(
    const ExtensionId& extension_id,
    const std::string& key) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);

  waiting_for_map_key_ = std::make_pair(extension_id, key);
  // If the key has already been reported, flush the state store.
  if (base::Contains(ids_and_keys_with_writes_, waiting_for_map_key_)) {
    state_store_->FlushForTesting(run_loop_.QuitWhenIdleClosure());
  }
  run_loop_.Run();
}

void StateStoreTestObserver::WillSetExtensionValue(
    const ExtensionId& extension_id,
    const std::string& key) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);

  auto map_key = std::make_pair(extension_id, key);
  // If the key matches, flush the state store. Otherwise, add it to the
  // map.
  if (map_key == waiting_for_map_key_) {
    state_store_->FlushForTesting(run_loop_.QuitWhenIdleClosure());
  } else {
    ids_and_keys_with_writes_.emplace(std::move(map_key));
  }
}

}  // namespace extensions