fix(gettext): Fix gettext to use with templv0.3.1

1 files changed, 15 insertions, 14 deletions

diff --git a/cmd/gettext/gettext_test.go b/cmd/gettext/gettext_test.go
index f1947be..8952c73 100644
--- a/cmd/gettext/gettext_test.go
+++ b/cmd/gettext/gettext_test.go
@@ -7,7 +7,7 @@ import (
 )
 
 func TestExtractKeys(t *testing.T) {
-	content := `{{ call .T "Hello" }}`
+	content := `{ login.Tr("Hello") }`
 
 	keys := extractKeys(content)
 
@@ -16,16 +16,7 @@ func TestExtractKeys(t *testing.T) {
 }
 
 func TestExtractKeysWithoutSpaces(t *testing.T) {
-	content := `{{call .T "Hello"}}`
-
-	keys := extractKeys(content)
-
-	assert.Len(t, keys, 1)
-	assert.Equal(t, 0, keys["Hello"])
-}
-
-func TestExtractKeysWithDollar(t *testing.T) {
-	content := `{{ call $.T "Hello" }}`
+	content := `{login.Tr("Hello")}`
 
 	keys := extractKeys(content)
 
@@ -34,7 +25,7 @@ func TestExtractKeysWithDollar(t *testing.T) {
 }
 
 func TestExtractKeysWithOneArgs(t *testing.T) {
-	content := `{{ call .T "approveButton" .Username }}`
+	content := `{ login.Tr("approveButton", a.Username) }`
 
 	keys := extractKeys(content)
 
@@ -43,7 +34,7 @@ func TestExtractKeysWithOneArgs(t *testing.T) {
 }
 
 func TestExtractKeysWithSeveralArgs(t *testing.T) {
-	content := `{{ call .T "approveButton" .Username .Age .Email }}`
+	content := `{ login.Tr("approveButton", a.Username, a.Age, a.Email) }`
 
 	keys := extractKeys(content)
 
@@ -52,10 +43,20 @@ func TestExtractKeysWithSeveralArgs(t *testing.T) {
 }
 
 func TestExtractKeysWithComplexArgs(t *testing.T) {
-	content := `{{ call $.T "raceStart_chosen" (.StartAt.Format "Monday, January 2, 2006 at 15:04") }}`
+	content := `{ login.Tr("raceStart_chosen", a.StartAt.Format("Monday, January 2, 2006 at 15:04")) }`
 
 	keys := extractKeys(content)
 
 	assert.Len(t, keys, 1)
 	assert.Equal(t, 1, keys["raceStart_chosen"])
 }
+
+func TestExtractMultipleKeys(t *testing.T) {
+	content := `{ login.Tr("test_1", a.StartAt.Format("Monday, January 2, 2006 at 15:04")) }{ login.Tr("test_2", a.Test) }`
+
+	keys := extractKeys(content)
+
+	assert.Len(t, keys, 2)
+	assert.Equal(t, 1, keys["test_1"])
+	assert.Equal(t, 1, keys["test_2"])
+}