DB service (#132)

2026-01-12 03:05:14 +00:00 · 2021-06-02 15:29:52 +01:00
parent 4d12230338
commit 7157db9aaa
15 changed files with 1389 additions and 4 deletions
--- a/db/handler/parse_test.go
+++ b/db/handler/parse_test.go
@@ -0,0 +1,141 @@
+package handler
+
+import (
+	"fmt"
+	"reflect"
+	"testing"
+
+	"github.com/crufter/lexer"
+)
+
+func TestLexing(t *testing.T) {
+	tokens, err := lexer.Lex("a == 12", expressions)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(tokens) != 3 {
+		t.Fatal(tokens)
+	}
+	if tokens[0].Typ != itemFieldName || tokens[1].Typ != itemEquals || tokens[2].Typ != itemInt {
+		t.Fatal(tokens)
+	}
+
+	tokens, err = lexer.Lex(`a == 12 and name != "nandos"`, expressions)
+	if tokens[0].Typ != itemFieldName ||
+		tokens[1].Typ != itemEquals ||
+		tokens[2].Typ != itemInt ||
+		tokens[3].Typ != itemAnd ||
+		tokens[4].Typ != itemFieldName ||
+		tokens[5].Typ != itemNotEquals ||
+		tokens[6].Typ != itemString {
+		t.Fatal(tokens)
+	}
+}
+
+type tCase struct {
+	Q   string
+	E   []Query
+	Err error
+}
+
+func TestParsing(t *testing.T) {
+	tCases := []tCase{
+		tCase{
+			Q: `a == 12 and name != "nandos"`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: int64(12),
+					Op:    itemEquals,
+				},
+				Query{
+					Field: "name",
+					Value: "nandos",
+					Op:    itemNotEquals,
+				},
+			},
+		},
+		// test escaping quotes
+		tCase{
+			Q: `a == 12 and name != "He said ""yes""!"`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: int64(12),
+					Op:    itemEquals,
+				},
+				Query{
+					Field: "name",
+					Value: `He said "yes"!`,
+					Op:    itemNotEquals,
+				},
+			},
+		},
+		tCase{
+			Q: `a == false and b == true`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: false,
+					Op:    itemEquals,
+				},
+				Query{
+					Field: "b",
+					Value: true,
+					Op:    itemEquals,
+				},
+			},
+		},
+		// a < 20
+		tCase{
+			Q: `a < 20`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: int64(20),
+					Op:    itemLessThan,
+				},
+			},
+		},
+		tCase{
+			Q: `a <= 20`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: int64(20),
+					Op:    itemLessThanEquals,
+				},
+			},
+		},
+		tCase{
+			Q: `a > 20`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: int64(20),
+					Op:    itemGreaterThan,
+				},
+			},
+		},
+		tCase{
+			Q: `a >= 20`,
+			E: []Query{
+				Query{
+					Field: "a",
+					Value: int64(20),
+					Op:    itemGreaterThanEquals,
+				},
+			},
+		},
+	}
+	for _, tCase := range tCases {
+		fmt.Println("Parsing", tCase.Q)
+		qs, err := Parse(tCase.Q)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if !reflect.DeepEqual(qs, tCase.E) {
+			t.Fatal("Expected", tCase.E, "got", qs)
+		}
+	}
+}