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

#ifndef COMPONENTS_METRICS_TEST_TEST_METRICS_LOG_UPLOADER_H_
#define COMPONENTS_METRICS_TEST_TEST_METRICS_LOG_UPLOADER_H_

#include "base/memory/weak_ptr.h"
#include "components/metrics/metrics_log.h"
#include "components/metrics/metrics_log_uploader.h"
#include "third_party/metrics_proto/reporting_info.pb.h"

namespace metrics {

class TestMetricsLogUploader : public MetricsLogUploader {
 public:
  explicit TestMetricsLogUploader(
      const MetricsLogUploader::UploadCallback& on_upload_complete);

  TestMetricsLogUploader(const TestMetricsLogUploader&) = delete;
  TestMetricsLogUploader& operator=(const TestMetricsLogUploader&) = delete;

  ~TestMetricsLogUploader() override;

  // Mark the current upload complete with the given response code.
  void CompleteUpload(int response_code, bool force_discard = false);

  // Check if UploadLog has been called.
  bool is_uploading() const { return is_uploading_; }

  const ReportingInfo& reporting_info() const { return last_reporting_info_; }

  base::WeakPtr<TestMetricsLogUploader> AsWeakPtr() {
    return weak_ptr_factory_.GetWeakPtr();
  }

 private:
  // MetricsLogUploader:
  void UploadLog(const std::string& compressed_log_data,
                 const LogMetadata& log_metadata,
                 const std::string& log_hash,
                 const std::string& log_signature,
                 const ReportingInfo& reporting_info) override;

  const MetricsLogUploader::UploadCallback on_upload_complete_;
  ReportingInfo last_reporting_info_;
  bool is_uploading_;
  base::WeakPtrFactory<TestMetricsLogUploader> weak_ptr_factory_{this};
};

}  // namespace metrics

#endif  // COMPONENTS_METRICS_TEST_TEST_METRICS_LOG_UPLOADER_H_