x/i18n/i18n_test.go

package i18n_test

import (
	"fmt"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/qor5/x/i18n"
	"github.com/theplant/testingutils"
	"golang.org/x/text/language"
)

type Messages struct {
	Update            string
	WelcomeToQOR5name string
}

var Messages_zh_CN = &Messages{
	Update:            "更新",
	WelcomeToQOR5name: "欢迎来到QOR5, {name}",
}

var Messages_en_US = &Messages{
	Update: "Update",
}

func TestLanguage(t *testing.T) {
	var mediaLibraryKey i18n.ModuleKey = "mediaLibraryKey"

	b := i18n.New().
		SupportLanguages(language.English, language.Japanese, language.SimplifiedChinese).
		RegisterForModule(language.SimplifiedChinese, mediaLibraryKey, Messages_zh_CN).
		RegisterForModule(language.English, mediaLibraryKey, Messages_en_US)

	h := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		msg := i18n.MustGetModuleMessages(r, mediaLibraryKey, Messages_en_US).(*Messages)
		_, _ = fmt.Fprintln(w, "")
		_, _ = fmt.Fprintln(w, msg.Update)
		_, _ = fmt.Fprintln(w, i18n.T(r, mediaLibraryKey, "Welcome Home &!@*#&^*!@^#*(!@ Felix"))
		_, _ = fmt.Fprintln(w, i18n.T(r, mediaLibraryKey, "Welcome to QOR5, {name}", "{name}", "Felix"))
		_, _ = fmt.Fprintln(w, i18n.PT(r, mediaLibraryKey, "Customer", "Name"))
	})

	recorder := httptest.NewRecorder()
	req, _ := http.NewRequest("GET", "/?lang=zh", nil)
	b.EnsureLanguage(h).ServeHTTP(recorder, req)

	diff := testingutils.PrettyJsonDiff(`
更新
Welcome Home &!@*#&^*!@^#*(!@ Felix
欢迎来到QOR5, Felix
Name
`, recorder.Body.String())
	if len(diff) > 0 {
		t.Error(diff)
	}

	if recorder.Header().Get("Set-Cookie") == "" {
		t.Error("cookie not set")
	}

	recorder = httptest.NewRecorder()
	req, _ = http.NewRequest("GET", "/", nil)
	req.AddCookie(&http.Cookie{Name: "lang", Value: "zh-Hans"})
	b.EnsureLanguage(h).ServeHTTP(recorder, req)

	if !strings.Contains(recorder.Body.String(), "更新") {
		t.Errorf("response is wrong, %s", recorder.Body.String())
	}

	recorder = httptest.NewRecorder()
	req, _ = http.NewRequest("GET", "/", nil)
	req.Header.Add("Accept-Language", "zh")
	b.EnsureLanguage(h).ServeHTTP(recorder, req)

	if !strings.Contains(recorder.Body.String(), "更新") {
		t.Errorf("response is wrong, %s", recorder.Body.String())
	}

	recorder = httptest.NewRecorder()
	req, _ = http.NewRequest("GET", "/", nil)
	b.EnsureLanguage(h).ServeHTTP(recorder, req)

	if !strings.Contains(recorder.Body.String(), "Update") {
		t.Errorf("response is wrong, %s", recorder.Body.String())
	}

}