fix defer timeout

samanhappy · Aug 22, 2024 · e0bc1ed · e0bc1ed
1 parent 88fa5c6
commit e0bc1ed
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/monkey/monkey_test.go b/monkey/monkey_test.go
@@ -34,8 +34,7 @@ func TestPatch(t *testing.T) {
 	originalFunc := func() string { return "original" }
 	replacementFunc := func() string { return "replacement" }
 
-	patch := Patch(originalFunc, replacementFunc)
-	defer patch.Reset()
+	Patch(originalFunc, replacementFunc)
 
 	assert.Equal(t, "replacement", originalFunc())
 }
@@ -52,7 +51,6 @@ func TestUnpatch(t *testing.T) {
 
 func TestPatchInstanceMethod(t *testing.T) {
 	PatchInstanceMethod(reflect.TypeOf(&MyStruct{}), "Method", func(*MyStruct) string { return "replacement" })
-	defer UnpatchInstanceMethod(reflect.TypeOf(&MyStruct{}), "Method")
 
 	assert.Equal(t, "replacement", (&MyStruct{}).Method())
 }