78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
package countdown
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
write = "write"
|
|
sleep = "sleep"
|
|
)
|
|
|
|
type SpyCountdownOperations struct {
|
|
Calls []string
|
|
}
|
|
|
|
func (s *SpyCountdownOperations) Sleep() {
|
|
s.Calls = append(s.Calls, sleep)
|
|
}
|
|
|
|
func (s *SpyCountdownOperations) Write(p []byte) (n int, err error) {
|
|
s.Calls = append(s.Calls, write)
|
|
return 0, nil
|
|
}
|
|
|
|
type SpyTime struct {
|
|
durationSlept time.Duration
|
|
}
|
|
|
|
func (s *SpyTime) Sleep(duration time.Duration) {
|
|
s.durationSlept = duration
|
|
}
|
|
|
|
func TestCountdown(t *testing.T) {
|
|
t.Run("print 3 to Go!", func(t *testing.T) {
|
|
buffer := &bytes.Buffer{}
|
|
Countdown(buffer, &SpyCountdownOperations{})
|
|
|
|
got := buffer.String()
|
|
want := `3
|
|
2
|
|
1
|
|
Go!`
|
|
|
|
if got != want {
|
|
t.Errorf("got %q want %q", got, want)
|
|
}
|
|
})
|
|
|
|
t.Run("sleep before every print", func(t *testing.T) {
|
|
spySleepPrinter := &SpyCountdownOperations{}
|
|
|
|
Countdown(spySleepPrinter, spySleepPrinter)
|
|
|
|
want := []string{
|
|
write, sleep, write, sleep, write, sleep, write,
|
|
}
|
|
|
|
if !reflect.DeepEqual(want, spySleepPrinter.Calls) {
|
|
t.Errorf("wanted %v got %v", want, spySleepPrinter.Calls)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestConfigurableSleeper(t *testing.T) {
|
|
sleepTime := 5 * time.Second
|
|
|
|
spyTime := &SpyTime{}
|
|
sleeper := NewConfigurableSleeper(sleepTime, spyTime.Sleep)
|
|
sleeper.Sleep()
|
|
|
|
if spyTime.durationSlept != sleepTime {
|
|
t.Errorf("should have slept for %v but slept for %v", sleepTime, spyTime.durationSlept)
|
|
}
|
|
}
|