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
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79


content / browser / service_worker / test_service_worker_observer.cc [blame]

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

#include "content/browser/service_worker/test_service_worker_observer.h"

#include <utility>

#include "base/run_loop.h"
#include "base/test/test_simple_task_runner.h"
#include "content/browser/service_worker/service_worker_context_wrapper.h"

namespace content {

TestServiceWorkerObserver::TestServiceWorkerObserver(
    scoped_refptr<ServiceWorkerContextWrapper> wrapper)
    : wrapper_(std::move(wrapper)) {
  wrapper_->AddObserver(this);
}

TestServiceWorkerObserver::~TestServiceWorkerObserver() {
  wrapper_->RemoveObserver(this);
}

void TestServiceWorkerObserver::RunUntilStatusChange(
    ServiceWorkerVersion* version,
    ServiceWorkerVersion::Status status) {
  if (version->status() == status)
    return;

  base::RunLoop loop;
  version_id_for_status_change_ = version->version_id();
  status_for_status_change_ = status;
  DCHECK(!quit_closure_for_status_change_);
  quit_closure_for_status_change_ = loop.QuitClosure();
  loop.Run();
}

void TestServiceWorkerObserver::RunUntilActivated(
    ServiceWorkerVersion* version,
    scoped_refptr<base::TestSimpleTaskRunner> runner) {
  if (version->status() == ServiceWorkerVersion::ACTIVATED)
    return;

  // Call runner->RunUntilIdle() to skip the delay for the activate event in
  // ServiceWorkerRegistration.
  RunUntilStatusChange(version, ServiceWorkerVersion::ACTIVATING);
  runner->RunUntilIdle();
  RunUntilStatusChange(version, ServiceWorkerVersion::ACTIVATED);
}

void TestServiceWorkerObserver::RunUntilLiveVersion() {
  if (!wrapper_->GetAllLiveVersionInfo().empty())
    return;

  base::RunLoop loop;
  DCHECK(!quit_closure_for_live_version_);
  quit_closure_for_live_version_ = loop.QuitClosure();
  loop.Run();
}

void TestServiceWorkerObserver::OnVersionStateChanged(
    int64_t version_id,
    const GURL& scope,
    const blink::StorageKey& key,
    ServiceWorkerVersion::Status status) {
  if (version_id == version_id_for_status_change_ &&
      status == status_for_status_change_ && quit_closure_for_status_change_) {
    std::move(quit_closure_for_status_change_).Run();
  }
}

void TestServiceWorkerObserver::OnNewLiveVersion(
    const ServiceWorkerVersionInfo& version_info) {
  if (quit_closure_for_live_version_)
    std::move(quit_closure_for_live_version_).Run();
}

}  // namespace content