allow -count to be passed to go test

superfly · Jan 23, 2024 · e761a26 · e761a26
1 parent b0ac63c
commit e761a26
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/caveat.go b/caveat.go
@@ -136,6 +136,10 @@ func RegisterCaveatJSONAlias(typ CaveatType, alias string) {
 	s2t[alias] = typ
 }
 
+func unegisterCaveatJSONAlias(alias string) {
+	delete(s2t, alias)
+}
+
 func typeToCaveat(t CaveatType) Caveat {
 	cav, ok := t2c[t]
 	if !ok {

diff --git a/caveat_test.go b/caveat_test.go
@@ -20,6 +20,7 @@ func TestCaveatRegistry(t *testing.T) {
 	assert.Equal(t, c, cs.Caveats[0])
 
 	RegisterCaveatJSONAlias(cavTestParentResource, "Foobar")
+	t.Cleanup(func() { unegisterCaveatJSONAlias("Foobar") })
 
 	assert.NoError(t, json.Unmarshal(j1, cs))
 	assert.Equal(t, 1, len(cs.Caveats))