// Copyright 2016 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 "platform/Histogram.h"

#include "base/metrics/histogram_samples.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace blink {

class ScopedUsHistogramTimerTest : public ::testing::Test {
public:
    static void advanceClock(double microseconds)
    {
        s_timeElapsed += microseconds;
    }

protected:
    static double returnMockTime()
    {
        return s_timeElapsed;
    }

    virtual void SetUp()
    {
        s_timeElapsed = 0.0;
        m_originalTimeFunction = setTimeFunctionsForTesting(returnMockTime);
    }

    virtual void TearDown()
    {
        setTimeFunctionsForTesting(m_originalTimeFunction);
    }

private:
    static double s_timeElapsed;
    TimeFunction m_originalTimeFunction;
};

double ScopedUsHistogramTimerTest::s_timeElapsed;

class TestCustomCountHistogram : public CustomCountHistogram {
public:
    TestCustomCountHistogram(const char* name, base::HistogramBase::Sample min, base::HistogramBase::Sample max, int32_t bucketCount)
    : CustomCountHistogram(name, min, max, bucketCount) {}

    base::HistogramBase* histogram() { return m_histogram; }
};

TEST_F(ScopedUsHistogramTimerTest, Basic)
{
    TestCustomCountHistogram scopedUsCounter("test", 0, 10000000, 50);
    {
        ScopedUsHistogramTimer timer(scopedUsCounter);
        advanceClock(0.5);
    }
    // 0.5s == 500000us
    EXPECT_EQ(500000, scopedUsCounter.histogram()->SnapshotSamples()->sum());
}

} // namespace blink