Add test cases for internal utils

aliyun · Aug 16, 2024 · e181d48 · e181d48
1 parent b45804a
commit e181d48
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/sdk/internal/utils_test.go b/sdk/internal/utils_test.go
@@ -0,0 +1,23 @@
+package internal
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetDefaultString(t *testing.T) {
+	assert.Equal(t, "default", GetDefaultString("", "default"))
+	assert.Equal(t, "custom", GetDefaultString("custom", "default"))
+	assert.Equal(t, "", GetDefaultString("", "", ""))
+}
+
+func TestMemoryAndRollback(t *testing.T) {
+	os.Setenv("test", "old")
+	rollback := Memory("test")
+	os.Setenv("test", "new")
+	rollback()
+
+	assert.Equal(t, "old", os.Getenv("test"))
+}