// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/testkit/testapp.h>
#include <vespa/slobrok/backoff.h>

#include <vespa/log/log.h>
LOG_SETUP("backoff_test");

using slobrok::api::BackOff;

TEST_SETUP(Test);

//-----------------------------------------------------------------------------

int
Test::Main()
{
    TEST_INIT("backoff_test");

    BackOff one;
    EXPECT_TRUE(one.shouldWarn());
    EXPECT_EQUAL(0.500, one.get());
    for (int i = 2; i < 41; i++) {
        EXPECT_EQUAL(0.5 * i, one.get());
    }
    for (int i = 1; i < 1000; i++) {
        EXPECT_EQUAL(20.0, one.get());
    }
    TEST_FLUSH();

    BackOff two;
    for (int i = 1; i < 50; i++) {
        double expect = 0.5 * i;
        if (expect > 20.0) expect = 20.0;
        EXPECT_EQUAL(expect, two.get());
        if (i == 1 || i == 7 || i == 18) {
            EXPECT_TRUE(two.shouldWarn());
        } else {
            EXPECT_FALSE(two.shouldWarn());
        }
    }
    two.reset();
    for (int i = 1; i < 50; i++) {
        double expect = 0.5 * i;
        if (expect > 20.0) expect = 20.0;
        EXPECT_EQUAL(expect, two.get());
        if (i == 1 || i == 7 || i == 18) {
            EXPECT_TRUE(two.shouldWarn());
        } else {
            EXPECT_FALSE(two.shouldWarn());
        }
    }
    for (int i = 0; i < 50000; i++) {
        EXPECT_EQUAL(20.0, two.get());
        if ((i % 180) == 5) {
            EXPECT_TRUE(two.shouldWarn());
        } else {
            EXPECT_FALSE(two.shouldWarn());
        }
    }
    TEST_DONE();
}