package search

import (
	"testing"

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

func TestTranslit(t *testing.T) {
	assert.Equal(t, "kino", Translit("Кино"))
	assert.Equal(t, "shchedrin", Translit("Щедрин"))
	assert.Equal(t, "кино", Translit("Kino"))
	assert.Equal(t, "щедрин", Translit("shchedrin"))
	assert.Equal(t, "", Translit("1984"))
}

func TestSwitchLayout(t *testing.T) {
	assert.Equal(t, "привет", SwitchLayout("ghbdtn"))
	assert.Equal(t, "hello", SwitchLayout("руддщ"))
	assert.Equal(t, "кино", SwitchLayout("rbyj"))
	assert.Equal(t, "", SwitchLayout("123"))
}