// Copyright 2021 The Chromium OS 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 "missive/compression/scoped_compression_feature.h"

#include <memory>
#include <utility>

#include <base/feature_list.h>

#include "missive/compression/compression_module.h"

namespace reporting {
namespace test {

ScopedCompressionFeature::ScopedCompressionFeature(bool enable) {
  std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);
  if (enable) {
    feature_list->InitializeFromCommandLine(
        {CompressionModule::kCompressReportingFeature}, {});
  } else {
    feature_list->InitializeFromCommandLine(
        {}, {CompressionModule::kCompressReportingFeature});
  }
  original_feature_list_ = base::FeatureList::ClearInstanceForTesting();
  base::FeatureList::SetInstance(std::move(feature_list));
}

ScopedCompressionFeature::~ScopedCompressionFeature() {
  base::FeatureList::ClearInstanceForTesting();
  base::FeatureList::RestoreInstanceForTesting(
      std::move(original_feature_list_));
}

}  // namespace test
}  // namespace reporting