package client

import (
	"context"
	"testing"
	"time"
)

func TestBackoffExp(t *testing.T) {
	results := []time.Duration{
		0 * time.Second,
		100 * time.Millisecond,
		600 * time.Millisecond,
		1900 * time.Millisecond,
		4300 * time.Millisecond,
		7900 * time.Millisecond,
	}

	r := &testRequest{
		service: "test",
		method:  "test",
	}

	for i := 0; i < 5; i++ {
		d, err := BackoffExp(context.TODO(), r, i)
		if err != nil {
			t.Fatal(err)
		}

		if d != results[i] {
			t.Fatalf("Expected equal than %v, got %v", results[i], d)
		}
	}
}

func TestBackoffInterval(t *testing.T) {
	minTime := 100 * time.Millisecond
	maxTime := 300 * time.Millisecond

	r := &testRequest{
		service: "test",
		method:  "test",
	}

	fn := BackoffInterval(minTime, maxTime)
	for i := 0; i < 5; i++ {
		d, err := fn(context.TODO(), r, i)
		if err != nil {
			t.Fatal(err)
		}

		if d < minTime || d > maxTime {
			t.Fatalf("Expected %v < %v < %v", minTime, d, maxTime)
		}
	}
}