package countdown

import (
	"fmt"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestCountdownWillCallback(t *testing.T) {

	called := make(chan int)
	OnTimeoutFn := func(time time.Time) error {
		called <- 1
		return nil
	}

	countdown := NewCountDown(1000 * time.Millisecond)
	countdown.OnTimeoutFn = OnTimeoutFn
	countdown.Reset()
	<-called
	fmt.Println("Times up, successfully called OnTimeoutFn")
}

func TestCountdownShouldReset(t *testing.T) {
	called := make(chan int)
	OnTimeoutFn := func(time time.Time) error {
		called <- 1
		return nil
	}

	countdown := NewCountDown(5000 * time.Millisecond)
	countdown.OnTimeoutFn = OnTimeoutFn
	// Check countdown did not start
	assert.False(t, countdown.getInitilisedValue())
	countdown.Reset()
	// Now the countdown should already started
	assert.True(t, countdown.getInitilisedValue())
	expectedCalledTime := time.Now().Add(9000 * time.Millisecond)
	resetTimer := time.NewTimer(4000 * time.Millisecond)

firstReset:
	for {
		select {
		case <-called:
			if time.Now().After(expectedCalledTime) {
				// Make sure the countdown runs forever
				assert.True(t, countdown.getInitilisedValue())
				fmt.Println("Correctly reset the countdown once")
			} else {
				t.Fatalf("Countdown did not reset correctly first time")
			}
			break firstReset
		case <-resetTimer.C:
			countdown.Reset()
		}
	}

	// Now the countdown is paused after calling the callback function, let's reset it again
	assert.True(t, countdown.getInitilisedValue())
	expectedTimeAfterReset := time.Now().Add(5000 * time.Millisecond)
	countdown.Reset()
	<-called
	// Always initilised
	assert.True(t, countdown.getInitilisedValue())
	if time.Now().After(expectedTimeAfterReset) {
		fmt.Println("Correctly reset the countdown second time")
	} else {
		t.Fatalf("Countdown did not reset correctly second time")
	}
}

func TestCountdownShouldBeAbleToStop(t *testing.T) {
	called := make(chan int)
	OnTimeoutFn := func(time time.Time) error {
		called <- 1
		return nil
	}

	countdown := NewCountDown(5000 * time.Millisecond)
	countdown.OnTimeoutFn = OnTimeoutFn
	// Check countdown did not start
	assert.False(t, countdown.getInitilisedValue())
	countdown.Reset()
	// Now the countdown should already started
	assert.True(t, countdown.getInitilisedValue())
	// Try manually stop the timer before it triggers the callback
	stopTimer := time.NewTimer(4000 * time.Millisecond)
	<-stopTimer.C
	countdown.StopTimer()
	assert.False(t, countdown.getInitilisedValue())
}