chrome/browser/ash/crosapi/field_trial_service_ash_unittest.cc

// Copyright 2021 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.

#include "chrome/browser/ash/crosapi/field_trial_service_ash.h"

#include "base/metrics/field_trial.h"
#include "base/test/bind.h"
#include "base/test/task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace crosapi {

namespace {
constexpr char kTrialName1[] = "trial1";
constexpr char kTrialName2[] = "trial2";
constexpr char kTrialName3[] = "trial3";
constexpr char kGroupName[] = "Default";

void CreateAndActivateFieldTrial(const std::string& trial_name,
                                 const std::string& group_name) {
  base::FieldTrialList::CreateFieldTrial(trial_name, group_name)->group();
}

void VerifyFieldTrial(const mojom::FieldTrialGroupInfoPtr& info,
                      const std::string& trial_name,
                      const std::string& group_name) {
  EXPECT_EQ(trial_name, info->trial_name);
  EXPECT_EQ(group_name, info->group_name);
}

class TestFieldTrialObserver : public mojom::FieldTrialObserver {
 public:
  using OnActivateCallback = base::RepeatingCallback<void(
      const std::vector<mojom::FieldTrialGroupInfoPtr>&)>;

  mojo::PendingRemote<mojom::FieldTrialObserver> BindAndGetRemote() {
    DCHECK(!receiver_.is_bound());
    return receiver_.BindNewPipeAndPassRemote();
  }

  void set_on_activate(OnActivateCallback callback) { callback_ = callback; }

  mojo::Receiver<mojom::FieldTrialObserver> receiver_{this};

 private:
  // mojom::FieldTrialObserver:
  void OnFieldTrialGroupActivated(
      std::vector<mojom::FieldTrialGroupInfoPtr> infos) override {
    if (callback_)
      callback_.Run(infos);
  }

  OnActivateCallback callback_;
};
}  // namespace

class FieldTrialServiceAshTest : public testing::Test {
 public:
  FieldTrialServiceAshTest() = default;
  FieldTrialServiceAshTest(const FieldTrialServiceAshTest&) = delete;
  FieldTrialServiceAshTest& operator=(const FieldTrialServiceAshTest&) = delete;
  ~FieldTrialServiceAshTest() override = default;

 protected:
  TestFieldTrialObserver observer_;

 private:
  base::test::SingleThreadTaskEnvironment task_environment;
};

TEST_F(FieldTrialServiceAshTest, SendInitialFieldTrialsAndUpdate) {
  CreateAndActivateFieldTrial(kTrialName1, kGroupName);
  CreateAndActivateFieldTrial(kTrialName2, kGroupName);

  // Send initial field trials.
  base::RunLoop run_loop1;
  observer_.set_on_activate(base::BindLambdaForTesting(
      [&](const std::vector<mojom::FieldTrialGroupInfoPtr>& infos) {
        EXPECT_EQ(2, infos.size());
        VerifyFieldTrial(infos[0], kTrialName1, kGroupName);
        VerifyFieldTrial(infos[1], kTrialName2, kGroupName);
        run_loop1.Quit();
        // Test won't proceed until this is called.
      }));
  FieldTrialServiceAsh service;
  service.AddFieldTrialObserver(observer_.BindAndGetRemote());
  run_loop1.Run();

  // Send field trial updates.
  CreateAndActivateFieldTrial(kTrialName3, kGroupName);
  base::RunLoop run_loop2;
  observer_.set_on_activate(base::BindLambdaForTesting(
      [&](const std::vector<mojom::FieldTrialGroupInfoPtr>& infos) {
        EXPECT_EQ(1, infos.size());
        VerifyFieldTrial(infos[0], kTrialName3, kGroupName);
        run_loop2.Quit();
        // Test won't exit until this is called.
      }));
  run_loop2.Run();
}

TEST_F(FieldTrialServiceAshTest, SendEmptyInitialFieldTrial) {
  FieldTrialServiceAsh service;
  service.AddFieldTrialObserver(observer_.BindAndGetRemote());

  // Send empty payload even if no field trials are active.
  // This is to make it more convenient for browser tests.
  base::RunLoop run_loop;
  observer_.set_on_activate(base::BindLambdaForTesting(
      [&](const std::vector<mojom::FieldTrialGroupInfoPtr>& infos) {
        EXPECT_EQ(0, infos.size());
        run_loop.Quit();
        // Test won't exit until this is called.
      }));
  run_loop.Run();
}

}  // namespace crosapi