Use Cleanup to remove old entries in ts map

2020-05-12 18:00:07 +02:00 · 2020-05-12 18:00:07 +02:00 · d20ac9444a
commit d20ac9444a
parent fe8a28286e
2 changed files with 5 additions and 0 deletions
--- a/assert.go
+++ b/assert.go
@ -24,6 +24,10 @@ func New(t T) Assert {
 	f := runtime.FuncForPC(reflect.ValueOf(a).Pointer())
 	ts[f] = t

+	t.Cleanup(func() {
+		delete(ts, f)
+	})
+
 	return a
 }

--- a/type.go
+++ b/type.go
@ -4,4 +4,5 @@ package assert
 type T interface {
 	Error(...interface{})
 	Helper()
+	Cleanup(func())
 }