File: feature_promo_test_util.cc

package info (click to toggle)
chromium 120.0.6099.224-1~deb11u1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 6,112,112 kB
  • sloc: cpp: 32,907,025; ansic: 8,148,123; javascript: 3,679,536; python: 2,031,248; asm: 959,718; java: 804,675; xml: 617,256; sh: 111,417; objc: 100,835; perl: 88,443; cs: 53,032; makefile: 29,579; fortran: 24,137; php: 21,162; tcl: 21,147; sql: 20,809; ruby: 17,735; pascal: 12,864; yacc: 8,045; lisp: 3,388; lex: 1,323; ada: 727; awk: 329; jsp: 267; csh: 117; exp: 43; sed: 37
file content (65 lines) | stat: -rw-r--r-- 2,310 bytes parent folder | download
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
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/user_education/test/feature_promo_test_util.h"

#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/time/time.h"
#include "components/feature_engagement/public/tracker.h"
#include "components/user_education/common/feature_promo_controller.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace user_education::test {

bool WaitForFeatureEngagementReady(feature_engagement::Tracker* tracker) {
  if (!tracker)
    return false;

  base::RunLoop run_loop(base::RunLoop::Type::kNestableTasksAllowed);
  auto quit_closure = run_loop.QuitClosure();
  bool success = false;
  tracker->AddOnInitializedCallback(
      base::BindLambdaForTesting([&](bool initialized) {
        success = initialized;
        std::move(quit_closure).Run();
      }));
  run_loop.Run();
  DCHECK(!success || tracker->IsInitialized());
  return success;
}

bool WaitForFeatureEngagementReady(FeaturePromoControllerCommon* controller) {
  return controller && WaitForFeatureEngagementReady(
                           controller->feature_engagement_tracker());
}

bool WaitForStartupPromo(feature_engagement::Tracker* tracker,
                         const base::Feature& iph_feature) {
  const bool fe_init_succeeded = WaitForFeatureEngagementReady(tracker);
  EXPECT_TRUE(fe_init_succeeded);
  return fe_init_succeeded &&
         tracker->GetTriggerState(iph_feature) ==
             feature_engagement::Tracker::TriggerState::HAS_BEEN_DISPLAYED;
}

bool WaitForStartupPromo(FeaturePromoControllerCommon* controller,
                         const base::Feature& iph_feature) {
  return controller &&
         WaitForStartupPromo(controller->feature_engagement_tracker(),
                             iph_feature);
}

bool SetClock(FeaturePromoControllerCommon* controller,
              const base::Clock& clock,
              base::Time initial_time) {
  if (!controller || !controller->feature_engagement_tracker()) {
    return false;
  }
  controller->feature_engagement_tracker()->SetClockForTesting(clock,
                                                               initial_time);
  return true;
}

}  // namespace user_education::test