srgs/token_test.go at master · robcapo/srgs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package srgs

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestToken_MatchPrefixMode(t *testing.T) {
	assert := assert.New(t)

	tok := Token{token: "my name is"}

	// Test covered utterance
	tok.Match("my name is rob", ModePrefix)
	str, err := tok.Next()

	assert.Nil(err)
	assert.Equal("rob", str)

	// Ensure next can only be called once
	str, err = tok.Next()
	assert.Equal(NoMatch, err)

	// Test prefix utterance
	tok.Match("my name", ModePrefix)
	str, err = tok.Next()

	assert.Nil(err)
	assert.Equal("", str)

	tok.Match("my name is", ModePrefix)
	str, err = tok.Next()

	assert.Nil(err)
	assert.Empty(str)
}

func TestToken_MatchExactMode(t *testing.T) {
	assert := assert.New(t)

	tok := Token{token: "my name is"}

	// Test covered utterance
	tok.Match("my name is rob", ModeExact)
	str, err := tok.Next()

	assert.Nil(err)
	assert.Equal("rob", str)

	// Ensure next can only be called once
	str, err = tok.Next()
	assert.Equal(NoMatch, err)

	// Test prefix utterance
	tok.Match("my name", ModeExact)
	str, err = tok.Next()

	assert.Equal(PrefixOnly, err)
	assert.Equal("", str)

	tok.Match("my name is", ModeExact)
	str, err = tok.Next()

	assert.Nil(err)
	assert.Empty(str)
}