Add unit test

Signed-off-by: Hongxin Liang <[email protected]>
flyteorg · May 16, 2023 · 271647f · 271647f
1 parent db387cd
commit 271647f
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pkg/util/util_test.go b/pkg/util/util_test.go
@@ -27,12 +27,18 @@ func TestSetupFlyteDir(t *testing.T) {
 	assert.Nil(t, SetupFlyteDir())
 }
 
-func TestPrintSandboxMessage(t *testing.T) {
+func TestPrintSandboxStartMessage(t *testing.T) {
 	t.Run("Print Sandbox Message", func(t *testing.T) {
 		PrintSandboxStartMessage(SandBoxConsolePort, docker.SandboxKubeconfig, false)
 	})
 }
 
+func TestPrintSandboxTeardownMessage(t *testing.T) {
+	t.Run("Print Sandbox Message", func(t *testing.T) {
+		PrintSandboxTeardownMessage(SandBoxConsolePort, docker.SandboxKubeconfig)
+	})
+}
+
 func TestSendRequest(t *testing.T) {
 	t.Run("Successful get request", func(t *testing.T) {
 		response, err := SendRequest("GET", "https://github.com", nil)