chrome/browser/web_applications/test/web_app_registration_waiter.cc

// Copyright 2019 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/web_applications/test/web_app_registration_waiter.h"

#include "base/test/bind.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace web_app {

WebAppRegistrationWaiter::WebAppRegistrationWaiter(
    ExternallyManagedAppManager* manager)
    : manager_(manager) {
  manager_->SetRegistrationCallbackForTesting(base::BindLambdaForTesting(
      [this](const GURL& install_url, RegistrationResultCode code) {
        ASSERT_EQ(install_url_, install_url);
        if (code_)
          ASSERT_EQ(code_, code);
        else
          ASSERT_NE(code, RegistrationResultCode::kTimeout);
        run_loop_.Quit();
      }));
  manager_->SetRegistrationsCompleteCallbackForTesting(
      complete_run_loop_.QuitClosure());
}

WebAppRegistrationWaiter::~WebAppRegistrationWaiter() {
  manager_->ClearRegistrationCallbackForTesting();
}

void WebAppRegistrationWaiter::AwaitNextRegistration(
    const GURL& install_url,
    RegistrationResultCode code) {
  install_url_ = install_url;
  code_ = code;
  run_loop_.Run();
}

void WebAppRegistrationWaiter::AwaitNextNonFailedRegistration(
    const GURL& install_url) {
  install_url_ = install_url;
  code_ = absl::nullopt;
  run_loop_.Run();
}

void WebAppRegistrationWaiter::AwaitRegistrationsComplete() {
  complete_run_loop_.Run();
}

}  // namespace web_app