fix test

go-kiss · May 7, 2024 · 67d8a47 · 67d8a47
1 parent 6eb21be
commit 67d8a47
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/monkey.go b/monkey.go
@@ -64,7 +64,7 @@ func PatchInstanceMethod(target reflect.Type, methodName string, replacement int
 		panic(fmt.Sprintf("unknown method %s", methodName))
 	}
 
-	o := &opt{}
+	o := &opt{global: true}
 	r := reflect.ValueOf(replacement)
 	patchValue(m.Func, r, o)
 	return &PatchGuard{m.Func, r, o}

diff --git a/monkey_test.go b/monkey_test.go
@@ -90,9 +90,9 @@ func (f *f) No() bool { return false }
 func TestOnInstanceMethod(t *testing.T) {
 	i := &f{}
 	assert(t, !i.No())
-	monkey.PatchInstanceMethod(reflect.TypeOf(i), "No", func(_ *f) bool { return true })
+	pg := monkey.PatchInstanceMethod(reflect.TypeOf(i), "No", func(_ *f) bool { return true })
 	assert(t, i.No())
-	assert(t, monkey.UnpatchInstanceMethod(reflect.TypeOf(i), "No"))
+	pg.Unpatch()
 	assert(t, !i.No())
 }