Add Test Shutdown After CleanUp

QuangTung97 · Oct 17, 2023 · 3a3f12c · 3a3f12c
1 parent cd1e256
commit 3a3f12c
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/svloc.go b/svloc.go
@@ -346,6 +346,7 @@ func (u *universeData) cloneShutdownFuncList() []func() {
 	funcList := make([]func(), len(u.shutdownFuncs))
 	copy(funcList, u.shutdownFuncs)
 
+	u.shutdownFuncs = nil
 	u.alreadyShutdown = true
 	return funcList
 }

diff --git a/svloc_test.go b/svloc_test.go
@@ -813,4 +813,24 @@ func TestUniverse_CleanUp(t *testing.T) {
 		})
 		assert.Equal(t, errors.New("svloc: can NOT call 'Wrap' after 'CleanUp'"), err)
 	})
+
+	t.Run("success shutdown after clean up", func(t *testing.T) {
+		var shutdowns []string
+		repoLoc := Register[Repo](func(unv *Universe) Repo {
+			unv.OnShutdown(func() {
+				shutdowns = append(shutdowns, "repo")
+			})
+			return &UserRepo{}
+		})
+
+		unv := NewUniverse()
+
+		repoLoc.Get(unv)
+
+		unv.CleanUp()
+
+		unv.Shutdown()
+
+		assert.Equal(t, []string{"repo"}, shutdowns)
+	})
 }