From 49600a6f46d99ff35d883381670548de1f01846e Mon Sep 17 00:00:00 2001 From: Ophestra Date: Sun, 31 Aug 2025 23:11:25 +0900 Subject: container/stub: export stub helpers These are very useful in many packages containing relatively large amount of code making calls to difficult or impossible to stub functions. Signed-off-by: Ophestra --- container/stub/exit_test.go | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 container/stub/exit_test.go (limited to 'container/stub/exit_test.go') diff --git a/container/stub/exit_test.go b/container/stub/exit_test.go new file mode 100644 index 00000000..50219712 --- /dev/null +++ b/container/stub/exit_test.go @@ -0,0 +1,30 @@ +package stub_test + +import ( + "testing" + + "hakurei.app/container/stub" +) + +func TestHandleExit(t *testing.T) { + t.Run("exit", func(t *testing.T) { + defer stub.HandleExit() + panic(stub.PanicExit) + }) + + t.Run("nil", func(t *testing.T) { + defer stub.HandleExit() + }) + + t.Run("passthrough", func(t *testing.T) { + defer func() { + want := 0xcafebabe + if r := recover(); r != want { + t.Errorf("recover: %v, want %v", r, want) + } + + }() + defer stub.HandleExit() + panic(0xcafebabe) + }) +} -- cgit v1.3.1