Release catalyst

caql/blevebuilder_test.go

@@ -0,0 +1,50 @@
+package caql
+
+import (
+	"testing"
+)
+
+func TestBleveBuilder(t *testing.T) {
+	tests := []struct {
+		name           string
+		saql           string
+		wantBleve      string
+		wantParseErr   bool
+		wantRebuildErr bool
+	}{
+		{name: "Search 1", saql: `"Bob"`, wantBleve: `"Bob"`},
+		{name: "Search 2", saql: `"Bob" AND title == 'Name'`, wantBleve: `"Bob" title:"Name"`},
+		{name: "Search 3", saql: `"Bob" OR title == 'Name'`, wantRebuildErr: true},
+		{name: "Search 4", saql: `title == 'malware' AND 'wannacry'`, wantBleve: `title:"malware" "wannacry"`},
+	}
+	for _, tt := range tests {
+		parser := &Parser{}
+
+		t.Run(tt.name, func(t *testing.T) {
+			expr, err := parser.Parse(tt.saql)
+			if (err != nil) != tt.wantParseErr {
+				t.Errorf("Parse() error = %v, wantErr %v", err, tt.wantParseErr)
+				if expr != nil {
+					t.Error(expr.String())
+				}
+				return
+			}
+			if err != nil {
+				return
+			}
+
+			got, err := expr.BleveString()
+			if (err != nil) != tt.wantRebuildErr {
+				t.Error(expr.String())
+				t.Errorf("String() error = %v, wantErr %v", err, tt.wantParseErr)
+				return
+			}
+			if err != nil {
+				return
+			}
+			if got != tt.wantBleve {
+				t.Errorf("String() got = %v, want %v", got, tt.wantBleve)
+			}
+		})
+	}
+}