missive/analytics/resource_collector_test.cc

// Copyright 2022 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/analytics/resource_collector.h"
#include "missive/analytics/resource_collector_mock.h"

#include <base/test/task_environment.h>
#include <base/time/time.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <metrics/metrics_library_mock.h>

namespace reporting::analytics {

class ResourceCollectorTest : public ::testing::TestWithParam<base::TimeDelta> {
 protected:
  void SetUp() override {
    // Replace the metrics library instance with a mock one
    resource_collector_.metrics_ = std::make_unique<MetricsLibraryMock>();
  }

  base::test::TaskEnvironment task_environment_{
      base::test::TaskEnvironment::TimeSource::MOCK_TIME};
  // The time interval that resource collector is expected to collect resources
  const base::TimeDelta interval_{GetParam()};
  ResourceCollectorMock resource_collector_{interval_};
};

TEST_P(ResourceCollectorTest, CallOnceInAWhile) {
  // Collect() should get called 3 times
  EXPECT_CALL(resource_collector_, Collect()).Times(3);
  // Time moving forward 3 * interval_
  task_environment_.FastForwardBy(3 * interval_);
  task_environment_.RunUntilIdle();
}

TEST_P(ResourceCollectorTest, DontCallIfTimeNotUp) {
  // Collect() should not be called
  EXPECT_CALL(resource_collector_, Collect()).Times(0);
  // Time moving forward half of interval_
  task_environment_.FastForwardBy(interval_ / 2);
  task_environment_.RunUntilIdle();
}

INSTANTIATE_TEST_SUITE_P(VaryingTimeInterval,
                         ResourceCollectorTest,
                         testing::Values(base::Minutes(10),
                                         base::Seconds(20),
                                         base::Hours(1)));
}  // namespace reporting::analytics