/*
 *  Copyright 2023 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "test/jitter/logging_delay_variation_calculator.h"

#include <algorithm>
#include <iterator>
#include <map>
#include <string>
#include <utility>
#include <vector>

#include "api/test/metrics/metric.h"
#include "api/test/metrics/metrics_logger.h"
#include "api/units/data_size.h"
#include "api/units/timestamp.h"
#include "system_wrappers/include/clock.h"
#include "test/gtest.h"

namespace webrtc {
namespace test {

TEST(LoggingDelayVariationCalculator, TimeSeriesWithTwoFrames) {
  SimulatedClock clock(123456);
  DefaultMetricsLogger logger(&clock);
  {
    LoggingDelayVariationCalculator calc("TimeSeriesWithTwoFrames", &logger);
    calc.Insert(3000, Timestamp::Millis(33), DataSize::Bytes(100));
    calc.Insert(6000, Timestamp::Millis(66), DataSize::Bytes(100));
    // Metrics are logged when `calc` goes out of scope.
  }

  std::vector<Metric> metrics = logger.GetCollectedMetrics();
  std::map<std::string, double> last_sample_by_metric_name;
  std::transform(metrics.begin(), metrics.end(),
                 std::inserter(last_sample_by_metric_name,
                               last_sample_by_metric_name.end()),
                 [](const Metric& metric) {
                   EXPECT_FALSE(metric.time_series.samples.empty());
                   return std::make_pair(
                       metric.name, metric.time_series.samples.back().value);
                 });
  EXPECT_EQ(last_sample_by_metric_name["rtp_timestamp"], 6000.0);
  EXPECT_EQ(last_sample_by_metric_name["arrival_time"], 66.0);
  EXPECT_EQ(last_sample_by_metric_name["size_bytes"], 100.0);
  EXPECT_EQ(last_sample_by_metric_name["inter_departure_time"], 33.333);
  EXPECT_EQ(last_sample_by_metric_name["inter_arrival_time"], 33.0);
  EXPECT_EQ(last_sample_by_metric_name["inter_delay_variation"], -0.333);
  EXPECT_EQ(last_sample_by_metric_name["inter_size_variation"], 0.0);
}

}  // namespace test
}  // namespace webrtc