Merge pull request #64 from anisjonischkeit/patch-1

Added *time.Time to Default arbitraries
leanovate · Jan 24, 2020 · f24d09d · f24d09d
2 parents aed0dd4 + 36affa8
commit f24d09d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/arbitrary/arbitraries.go b/arbitrary/arbitraries.go
@@ -20,7 +20,13 @@ type Arbitraries struct {
 func DefaultArbitraries() *Arbitraries {
 	return &Arbitraries{
 		generators: map[reflect.Type]gopter.Gen{
-			reflect.TypeOf(time.Now()): gen.Time(),
+			reflect.TypeOf(time.Time{}): gen.Time(),
+			reflect.TypeOf(&time.Time{}): gen.Time().
+				Map(func(time time.Time) *time.Time { return &time }).
+				WithShrinker(func(v interface{}) gopter.Shrink {
+					t := v.(*time.Time)
+					return gen.TimeShrinker(*t)
+				}),
 		},
 	}
 }