diff --git a/caveat.go b/caveat.go index 9306584..e07a70e 100644 --- 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 index 3f73942..2eb9f68 100644 --- 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))