Fix Testable Examples

taman9333 · Jun 21, 2024 · 59138dd · 59138dd
1 parent 66675f4
commit 59138dd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/io_example_test.go b/io_example_test.go
@@ -5,9 +5,20 @@ import (
 	"time"
 )
 
+var mockTimeNowYear = func() int {
+	return time.Now().Year()
+}
+
 func ExampleIO() {
+	originalTimeNowYear := mockTimeNowYear
+	defer func() { mockTimeNowYear = originalTimeNowYear }() // Restore the original function after the test
+
+	mockTimeNowYear = func() int {
+		return 2023
+	}
+
 	io := NewIO(func() int {
-		return time.Now().Year()
+		return mockTimeNowYear()
 	})
 
 	result1 := io.Run()

diff --git a/task_example_test.go b/task_example_test.go
@@ -2,13 +2,19 @@ package mo
 
 import (
 	"fmt"
-	"time"
 )
 
 func ExampleTask() {
+	originalTimeNowYear := mockTimeNowYear
+	defer func() { mockTimeNowYear = originalTimeNowYear }() // Restore the original function after the test
+
+	mockTimeNowYear = func() int {
+		return 2023
+	}
+
 	task := NewTask(func() *Future[int] {
 		return NewFuture(func(resolve func(int), reject func(error)) {
-			resolve(time.Now().Year())
+			resolve(mockTimeNowYear())
 		})
 	})