summaryrefslogtreecommitdiff
path: root/chromium/components/offline_items_collection/core/test_support/scoped_mock_offline_content_provider.cc
blob: 6b3d93c9034a9df3d1fabb4085a97ce5e2e02c8c (plain)
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
// Copyright 2017 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 "components/offline_items_collection/core/test_support/scoped_mock_offline_content_provider.h"

#include "components/offline_items_collection/core/offline_content_aggregator.h"
#include "components/offline_items_collection/core/offline_content_provider.h"

namespace offline_items_collection {

ScopedMockOfflineContentProvider::ScopedMockObserver::ScopedMockObserver(
    OfflineContentProvider* provider)
    : provider_(provider) {
  provider_->AddObserver(this);
}

ScopedMockOfflineContentProvider::ScopedMockObserver::~ScopedMockObserver() {
  provider_->RemoveObserver(this);
}

ScopedMockOfflineContentProvider::ScopedMockOfflineContentProvider(
    const std::string& name_space,
    OfflineContentAggregator* aggregator)
    : name_space_(name_space), aggregator_(aggregator) {
  aggregator_->RegisterProvider(name_space_, this);
}

ScopedMockOfflineContentProvider::~ScopedMockOfflineContentProvider() {
  Unregister();
}

void ScopedMockOfflineContentProvider::Unregister() {
  if (!aggregator_)
    return;

  aggregator_->UnregisterProvider(name_space_);
  aggregator_ = nullptr;
}

}  // namespace offline_items_collection