fix the tests

xgfone · Aug 17, 2022 · 96298a0 · 96298a0
1 parent 76f8501
commit 96298a0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/atexit_test.go b/atexit_test.go
@@ -50,12 +50,12 @@ func TestRegisterAndExecute(t *testing.T) {
 		t.Errorf("expect unexecuted, but got executed")
 	}
 
-	RegisterWithPriority(1, func() { buf.WriteString("1") })
-	RegisterWithPriority(2, func() { buf.WriteString("2") })
-	RegisterWithPriority(3, func() { buf.WriteString("3") })
-	RegisterWithPriority(3, func() { buf.WriteString("4") })
-	RegisterWithPriority(2, func() { buf.WriteString("5") })
-	RegisterWithPriority(1, func() { buf.WriteString("6") })
+	OnExitWithPriority(1, func() { buf.WriteString("1") })
+	OnExitWithPriority(2, func() { buf.WriteString("2") })
+	OnExitWithPriority(3, func() { buf.WriteString("3") })
+	OnExitWithPriority(3, func() { buf.WriteString("4") })
+	OnExitWithPriority(2, func() { buf.WriteString("5") })
+	OnExitWithPriority(1, func() { buf.WriteString("6") })
 	go func() { time.Sleep(time.Second); Execute() }()
 
 	start := time.Now()

diff --git a/init_test.go b/init_test.go
@@ -18,13 +18,13 @@ import "fmt"
 
 func ExampleInit() {
 	// Register the init functions.
-	RegisterInit(func() { fmt.Println("init1") })
-	RegisterInit(func() { fmt.Println("init2") })
-	RegisterInitWithPriority(20, func() { fmt.Println("init3") })
-	RegisterInitWithPriority(10, func() { fmt.Println("init4") })
-	RegisterInitWithPriority(10, func() { fmt.Println("init5") })
-	RegisterInitWithPriority(30, func() { fmt.Println("init6") })
-	RegisterInit(func() { fmt.Println("init7") })
+	OnInit(func() { fmt.Println("init1") })
+	OnInit(func() { fmt.Println("init2") })
+	OnInitWithPriority(20, func() { fmt.Println("init3") })
+	OnInitWithPriority(10, func() { fmt.Println("init4") })
+	OnInitWithPriority(10, func() { fmt.Println("init5") })
+	OnInitWithPriority(30, func() { fmt.Println("init6") })
+	OnInit(func() { fmt.Println("init7") })
 
 	// Call the registered init functions.
 	Init()