Add Reset() (#9)

Author: 6543

i18n.go

@@ -174,6 +174,11 @@ func SetMessageWithDesc(lang, langDesc string, localeFile interface{}, otherLoca
 	return err
 }
 
+// Reset resets locale store.
+func Reset() {
+	locales = &localeStore{store: make(map[string]*locale)}
+}
+
 // SetMessage sets the message file for localization.
 func SetMessage(lang string, localeFile interface{}, otherLocaleFiles ...interface{}) error {
 	return SetMessageWithDesc(lang, lang, localeFile, otherLocaleFiles...)

i18n_test.go

@@ -44,6 +44,20 @@ func Test_Tr(t *testing.T) {
 	if result != `test value <span style="color: ; background: ">more text</span>` {
 		t.Errorf(`expect 'test value <span style="color: ; background: ">more text</span>', got '%s'`, result)
 	}
+
+	langs := ListLangs()
+	if len(langs) != 1 {
+		t.Errorf("expect '1', got '%d'", len(langs))
+	} else if langs[0] != "en-US" {
+		t.Errorf("expect 'en-US', got '%s'", langs[0])
+	}
+
+	Reset()
+
+	langs = ListLangs()
+	if len(langs) != 0 {
+		t.Errorf("expect '0', got '%d'", len(langs))
+	}
 }
 
 func Benchmark_Tr(b *testing.B) {