/*
Copyright IBM Corp. All Rights Reserved.

SPDX-License-Identifier: Apache-2.0
*/

package onboarding

import (
	"sync"
	"testing"
	"time"

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

func TestExponentialDuration(t *testing.T) {
	t.Parallel()
	exp := exponentialDurationSeries(time.Millisecond*100, time.Second)
	prev := exp()
	for i := 0; i < 3; i++ {
		n := exp()
		assert.Equal(t, prev*2, n)
		prev = n
		assert.True(t, n < time.Second)
	}

	for i := 0; i < 10; i++ {
		assert.Equal(t, time.Second, exp())
	}
}

func TestTicker(t *testing.T) {
	t.Parallel()
	everyMillis := func() time.Duration {
		return time.Millisecond
	}

	t.Run("Stop ticker serially", func(t *testing.T) {
		ticker := newTicker(everyMillis)
		for i := 0; i < 10; i++ {
			<-ticker.C
		}

		ticker.stop()
		// Ensure the ticker channel is closed once stop() is called.
		gt := gomega.NewGomegaWithT(t)
		gt.Eventually(func() bool {
			_, ok := <-ticker.C
			return ok
		}).Should(gomega.BeFalse())
	})

	t.Run("Stop ticker concurrently", func(t *testing.T) {
		ticker := newTicker(func() time.Duration {
			return time.Millisecond
		})

		var tickerStopped sync.WaitGroup
		tickerStopped.Add(1)

		go func() {
			defer tickerStopped.Done()
			time.Sleep(time.Millisecond * 50)
			ticker.stop()
			<-ticker.C
		}()

		tickerStopped.Wait()
		_, ok := <-ticker.C
		assert.False(t, ok)

	})

}
