refactor test

samanhappy · Aug 23, 2024 · 752154a · 752154a
1 parent 2fe3fc5
commit 752154a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/monkey/monkey_test.go b/monkey/monkey_test.go
@@ -32,9 +32,9 @@ func (s *MyStruct) Method() string {
 
 func TestPatch(t *testing.T) {
 	original := func() string { return "original" }
-	replacement := func() string { return "replacement" }
+	assert.Equal(t, "original", original())
 
-	Patch(original, replacement)
+	Patch(original, func() string { return "replacement" })
 	assert.Equal(t, "replacement", original())
 
 	Unpatch(original)