// Copyright 2020 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 "third_party/blink/public/common/privacy_budget/scoped_switch_sample_collector.h"

#include "third_party/blink/common/privacy_budget/aggregating_sample_collector.h"
#include "third_party/blink/common/privacy_budget/identifiability_sample_collector_test_utils.h"
#include "third_party/blink/public/common/privacy_budget/identifiability_sample_collector.h"

namespace blink {
namespace test {

ScopedSwitchSampleCollector::ScopedSwitchSampleCollector(
    IdentifiabilitySampleCollector* new_aggregator) {
  SetCollectorInstanceForTesting(new_aggregator);
}

ScopedSwitchSampleCollector::~ScopedSwitchSampleCollector() {
  // No need to restore original collector since
  // `SetCollectorInstanceForTesting` doesn't allow nested scopes.
  SetCollectorInstanceForTesting(nullptr);
}

}  // namespace test
}  // namespace blink