added some tests

shixzie · Sep 18, 2017 · fb7169b · fb7169b
1 parent 8c2ad72
commit fb7169b
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 11 deletions.
diff --git a/nlp_test.go b/nlp_test.go
@@ -34,6 +34,7 @@ func TestNL_P(t *testing.T) {
 		"dur {Dur}",
 		"string {String} int {Int}",
 		"string {String} time {Time}",
+		"need {String} since {Time}",
 	}
 
 	nl := New()
@@ -55,52 +56,60 @@ func TestNL_P(t *testing.T) {
 		expression string
 		want       *T
 	}{
-		{
+		0: {
 			"string",
 			"string Hello World",
 			&T{String: "Hello World"},
 		},
-		{
+		1: {
 			"int",
 			"int 42",
 			&T{Int: 42},
 		},
-		{
+		2: {
 			"uint",
 			"uint 43",
 			&T{Uint: 43},
 		},
-		{
+		3: {
 			"float",
 			"float 44",
 			&T{Float: 44},
 		},
-		{
+		4: {
 			"time",
 			"time 05-18-1999_6:42pm",
 			&T{Time: tim},
 		},
-		{
+		5: {
 			"duration",
 			"dur 4h2m",
 			&T{Dur: dur},
 		},
-		{
+		6: {
 			"string int",
 			"string Lmao int 42",
 			&T{
 				String: "Lmao",
 				Int:    42,
 			},
 		},
-		{
+		7: {
 			"string time",
 			"string What's Up Boy time 05-18-1999_6:42pm",
 			&T{
 				String: "What's Up Boy",
 				Time:   tim,
 			},
 		},
+		8: {
+			"word string time",
+			"Hi, I am Patrice, I need Issue#4 since 05-18-1999_6:42pm",
+			&T{
+				String: "Issue#4",
+				Time:   tim,
+			},
+		},
 	}
 	for i, tt := range cases {
 		t.Run(tt.name, func(t *testing.T) {

diff --git a/parser/parser_test.go b/parser/parser_test.go
@@ -16,13 +16,13 @@ func TestParseSample(t *testing.T) {
 		want    []Token
 		wantErr bool
 	}{
-		{
+		0: {
 			"err: empty sample",
 			args{0, nil},
 			nil,
 			true,
 		},
-		{
+		1: {
 			"normal sample",
 			args{1, []byte("play {Name} from {Artist}")},
 			[]Token{
@@ -33,7 +33,7 @@ func TestParseSample(t *testing.T) {
 			},
 			false,
 		},
-		{
+		2: {
 			"spacing inside keys",
 			args{1, []byte("play { 	Name} from {	Artist		}")},
 			[]Token{
@@ -44,6 +44,18 @@ func TestParseSample(t *testing.T) {
 			},
 			false,
 		},
+		3: {
+			"multi word",
+			args{1, []byte("I need {Name} since {Since}")},
+			[]Token{
+				{Val: []byte("I")},
+				{Val: []byte("need")},
+				{Kw: true, Val: []byte("Name")},
+				{Val: []byte("since")},
+				{Kw: true, Val: []byte("Since")},
+			},
+			false,
+		},
 	}
 	for i, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {