28 lines
489 B
Go
28 lines
489 B
Go
package test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"allaboutapps.dev/aw/go-starter/internal/api"
|
|
"github.com/dropbox/godropbox/time2"
|
|
)
|
|
|
|
func GetMockClock(t *testing.T, clock time2.Clock) *time2.MockClock {
|
|
t.Helper()
|
|
mc, ok := clock.(*time2.MockClock)
|
|
if !ok {
|
|
t.Fatalf("invalid clock type, got %T, want *time2.MockClock", clock)
|
|
}
|
|
|
|
return mc
|
|
}
|
|
|
|
func SetMockClock(t *testing.T, s *api.Server, time time.Time) {
|
|
t.Helper()
|
|
|
|
mockClock := GetMockClock(t, s.Clock)
|
|
|
|
mockClock.Set(time)
|
|
}
|