chromeos/lacros/lacros_test_helper.cc

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chromeos/lacros/lacros_test_helper.h"

#include "base/check.h"
#include "chromeos/crosapi/mojom/test_controller.mojom-test-utils.h"
#include "chromeos/startup/browser_init_params.h"

namespace chromeos {
namespace {
base::Version GetAshVersion() {
  constexpr int min_mojo_version =
      crosapi::mojom::TestController::kGetAshVersionMinVersion;
  if (chromeos::LacrosService::Get()->GetInterfaceVersion(
          crosapi::mojom::TestController::Uuid_) < min_mojo_version) {
    return base::Version({0, 0, 0, 0});
  }

  std::string ash_version_str;
  crosapi::mojom::TestControllerAsyncWaiter async_waiter(
      chromeos::LacrosService::Get()
          ->GetRemote<crosapi::mojom::TestController>()
          .get());
  async_waiter.GetAshVersion(&ash_version_str);
  return base::Version(ash_version_str);
}
}  // namespace

ScopedDisableCrosapiForTesting::ScopedDisableCrosapiForTesting()
    : disable_crosapi_resetter_(
          &BrowserInitParams::disable_crosapi_for_testing_,
          true) {
  // Ensure that no instance exist, to prevent interference.
  CHECK(!LacrosService::Get());
}

// TODO(crbug.com/1196314): Ensure that no instance exist on destruction, too.
// Currently, browser_tests' shutdown is an exception.
ScopedDisableCrosapiForTesting::~ScopedDisableCrosapiForTesting() = default;

ScopedLacrosServiceTestHelper::ScopedLacrosServiceTestHelper() = default;

ScopedLacrosServiceTestHelper::~ScopedLacrosServiceTestHelper() = default;

bool IsAshVersionAtLeastForTesting(base::Version required_version) {
  DCHECK(required_version.IsValid());
  DCHECK(LacrosService::Get());
  static base::Version cached_ash_version = GetAshVersion();
  DCHECK(cached_ash_version.IsValid());
  return (cached_ash_version >= required_version);
}

}  // namespace chromeos