95 lines
1.5 KiB
Go
95 lines
1.5 KiB
Go
package trans_test
|
|
|
|
import (
|
|
"testing"
|
|
"."
|
|
"reflect"
|
|
"os"
|
|
)
|
|
|
|
|
|
type testData struct {
|
|
word string
|
|
fromLang string
|
|
toLang string
|
|
transOutput string
|
|
expectedResults []string
|
|
}
|
|
|
|
func TestTranslation(t *testing.T) {
|
|
cases := []testData{
|
|
testData{
|
|
"actuator", "en", "hu",
|
|
`működtető
|
|
|
|
noun
|
|
indítókar
|
|
actuator, starting lever
|
|
|
|
actuator
|
|
működtető, beavatkozó, hajtómű, aktuátor, hajtás
|
|
`,
|
|
[]string{"működtető", "indítókar", "beavatkozó", "hajtómű", "aktuátor", "hajtás"},
|
|
},
|
|
testData{
|
|
"szaxofon", "hu", "en",
|
|
`saxophone
|
|
|
|
szaxofon
|
|
saxophone
|
|
`,
|
|
[]string{"saxophone"},
|
|
},
|
|
testData{
|
|
"cat", "en", "hu",
|
|
`macska
|
|
|
|
noun
|
|
macska
|
|
cat, pussy
|
|
pletykás nő
|
|
cat
|
|
|
|
verb
|
|
felvon
|
|
hoist, cat
|
|
|
|
cat
|
|
macska, cica
|
|
`,
|
|
[]string{"macska", "pletykás nő", "felvon", "cica"},
|
|
},
|
|
testData{
|
|
"méltányosság", "hu", "en",
|
|
`equity
|
|
|
|
noun
|
|
equity
|
|
méltányosság, jogosság
|
|
fairness
|
|
méltányosság, szőkeség
|
|
justness
|
|
jogosság, méltányosság
|
|
|
|
méltányosság
|
|
equity, fairness
|
|
`,
|
|
[]string{"equity", "fairness", "justness"},
|
|
},
|
|
}
|
|
|
|
for _, data := range cases {
|
|
tr := trans.New(data.word)
|
|
tr.Language = data.fromLang
|
|
if _, ok := os.LookupEnv("NOMOCK"); !ok {
|
|
tr.Translator = func(string, string, string) (string, error) {
|
|
return data.transOutput, nil
|
|
}
|
|
}
|
|
r := tr.Translate(data.toLang)
|
|
if (!reflect.DeepEqual(r, data.expectedResults)) {
|
|
t.Errorf("\nExpected: '%v'\nActual: '%v'", data.expectedResults, r)
|
|
}
|
|
}
|
|
}
|