package main

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestExtractKeys(t *testing.T) {
	content := `{ login.Tr("fr", "Hello") }`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 0, keys["Hello"])
}

func TestExtractKeysWithoutSpaces(t *testing.T) {
	content := `{login.Tr("fr", "Hello")}`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 0, keys["Hello"])
}

func TestExtractKeysWithOneArgs(t *testing.T) {
	content := `{ login.Tr("fr", "approveButton", a.Username) }`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 1, keys["approveButton"])
}

func TestExtractKeysWithSeveralArgs(t *testing.T) {
	content := `{ login.Tr("fr", "approveButton", a.Username, a.Age, a.Email) }`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 3, keys["approveButton"])
}

func TestExtractKeysWithComplexArgs(t *testing.T) {
	content := `{ login.Tr("fr", "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("fr", "test_1", a.StartAt.Format("Monday, January 2, 2006 at 15:04")) }{ login.Tr("fr", "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"])
}

func TestExtractKeyFromSimpleTrFunc(t *testing.T) {
	content := `{ Tr("fr", "test") }`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 0, keys["test"])
}

func TestExtractKeyWithSpan(t *testing.T) {
	content := `{ Tr("fr", "Hello <span class=\"text-bold\">%s</span>", username) }`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 1, keys["Hello <span class=\"text-bold\">%s</span>"])
}

func TestComponent(t *testing.T) {
	content := `@ki18n.Tr(login.Lng, "Recharge yourself in nature.")`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 0, keys["Recharge yourself in nature."])
}

func TestStr(t *testing.T) {
	content := `{ ki18n.Str(login.Lng, "Username") }`

	keys := extractKeys(content)

	assert.Len(t, keys, 1)
	assert.Equal(t, 0, keys["Username"])
}