lib: refine library to text

aliyun · Jan 4, 2024 · 516e575 · 516e575
1 parent a3d545d
commit 516e575
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 60 deletions.
diff --git a/i18n/library.go b/i18n/library.go
diff --git a/i18n/library_test.go b/i18n/library_test.go
diff --git a/i18n/text.go b/i18n/text.go
@@ -13,6 +13,17 @@
 // limitations under the License.
 package i18n
 
+func T(en string, zh string) *Text {
+	t := &Text{
+		dic: make(map[string]string),
+	}
+	t.dic["en"] = en
+	if zh != "" {
+		t.dic["zh"] = zh
+	}
+	return t
+}
+
 type Text struct {
 	dic map[string]string
 }

diff --git a/i18n/text_test.go b/i18n/text_test.go
@@ -33,3 +33,18 @@ func TestText(t *testing.T) {
 	language = "zh"
 	assert.Equal(t, "你好", tx.Text())
 }
+
+func TestLibrary(t *testing.T) {
+	//Test T(en string, zh string)*Text
+	text := T("hello", "你好")
+	assert.Equal(t, "hello", text.dic["en"])
+	assert.Equal(t, "你好", text.dic["zh"])
+
+	text = T("", "你好")
+	assert.Equal(t, "", text.dic["en"])
+	assert.Equal(t, "你好", text.dic["zh"])
+
+	text = T("hello", "")
+	assert.Equal(t, "hello", text.dic["en"])
+	assert.Equal(t, "", text.dic["zh"])
+}