go-astikit/translator_test.go

package astikit

import (
	"net/http"
	"net/http/httptest"
	"reflect"
	"strings"
	"testing"
)

func TestTranslator(t *testing.T) {
	// Setup
	tl := NewTranslator(TranslatorOptions{DefaultLanguage: "fr"})

	// Parse dir
	err := tl.ParseDir("testdata/translator")
	if err != nil {
		t.Fatalf("expected no error, got %v", err)
	}
	if e := map[string]string{
		"en.1":       "1",
		"en.2.3":     "3",
		"en.d1.5":    "5",
		"en.d1.d2.6": "6",
		"en.f":       "f%sf",
		"fr.4":       "4",
	}; !reflect.DeepEqual(e, tl.p) {
		t.Fatalf("expected %+v, got %+v", e, tl.p)
	}

	// Middleware
	var o string
	s := httptest.NewServer(ChainHTTPMiddlewares(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
		var args []any
		if v := r.Header.Get("args"); v != "" {
			for _, s := range strings.Split(v, ",") {
				args = append(args, s)
			}
		}
		if len(args) > 0 {
			o = tl.TranslateCf(r.Context(), r.Header.Get("key"), args...)
		} else {
			o = tl.TranslateC(r.Context(), r.Header.Get("key"))
		}
	}), tl.HTTPMiddleware))
	defer s.Close()

	// Translate
	for _, v := range []struct {
		args     []string
		expected string
		key      string
		language string
	}{
		{
			expected: "4",
			key:      "4",
		},
		{
			expected: "fr.1",
			key:      "1",
		},
		{
			expected: "3",
			key:      "2.3",
			language: "en-US,en;q=0.8",
		},
		{
			expected: "4",
			key:      "4",
			language: "en",
		},
		{
			expected: "en.5",
			key:      "5",
			language: "en",
		},
		{
			expected: "6",
			key:      "d1.d2.6",
			language: "en",
		},
		{
			expected: "4",
			key:      "4",
			language: "it",
		},
		{
			args:     []string{"arg"},
			expected: "fargf",
			key:      "f",
			language: "en",
		},
	} {
		r, err := http.NewRequest(http.MethodGet, s.URL, nil)
		if err != nil {
			t.Fatalf("expected no error, got %+v", err)
		}
		if len(v.args) > 0 {
			r.Header.Set("args", strings.Join(v.args, ","))
		}
		r.Header.Set("key", v.key)
		if v.language != "" {
			r.Header.Set("Accept-Language", v.language)
		}
		_, err = http.DefaultClient.Do(r)
		if err != nil {
			t.Fatalf("expected no error, got %+v", err)
		}
		if !reflect.DeepEqual(v.expected, o) {
			t.Fatalf("expected %+v, got %+v", v.expected, o)
		}
	}
}

func TestTranslator_ParseAcceptLanguage(t *testing.T) {
	tl := NewTranslator(TranslatorOptions{ValidLanguages: []string{"en", "fr"}})
	if e, g := "", tl.parseAcceptLanguage(""); !reflect.DeepEqual(e, g) {
		t.Fatalf("expected %+v, got %+v", e, g)
	}
	if e, g := "fr", tl.parseAcceptLanguage(" fr-FR, fr ; q=0.9 ,en;q=0.7,en-US;q=0.8 "); !reflect.DeepEqual(e, g) {
		t.Fatalf("expected %+v, got %+v", e, g)
	}
}