forgejo/modules/translation/mock.go

// Copyright 2020 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package translation

import (
	"fmt"
	"html/template"

	"forgejo.org/modules/translation/i18n"
)

// MockLocale provides a mocked locale without any translations, other than those inserted into MockTranslations by a testcase
type MockLocale struct {
	Lang, LangName string // these fields are used directly in templates: ctx.Locale.Lang

	MockTranslations map[string]string
}

var _ Locale = (*MockLocale)(nil)

func (l MockLocale) Language() string {
	return "en"
}

func (l MockLocale) TrString(s string, _ ...any) string {
	if val, ok := l.MockTranslations[s]; ok {
		return val
	}
	return s
}

func (l MockLocale) Tr(s string, a ...any) template.HTML {
	return template.HTML(l.TrString(s))
}

func (l MockLocale) TrN(cnt any, key1, keyN string, args ...any) template.HTML {
	return template.HTML(key1)
}

func (l MockLocale) TrPluralString(count any, trKey string, trArgs ...any) template.HTML {
	return template.HTML(trKey)
}

// TrPluralStringAllForms implements Locale.
func (l MockLocale) TrPluralStringAllForms(trKey string) ([]string, []string) {
	return []string{l.TrString(trKey + i18n.PluralFormSeparator + "one"), l.TrString(trKey + i18n.PluralFormSeparator + "other")}, nil
}

func (l MockLocale) TrSize(s int64) ReadableSize {
	return ReadableSize{fmt.Sprint(s), ""}
}

func (l MockLocale) HasKey(key string) bool {
	return true
}

func (l MockLocale) PrettyNumber(v any) string {
	return fmt.Sprint(v)
}