Add query package with first set of conditions and tests

2026-04-28 20:00:33 +00:00 · 2018-08-29 20:02:02 +02:00 · 2018-08-29 20:02:02 +02:00 · 6ed50f34fb
commit 6ed50f34fb
parent 1bf7e42d8a
16 changed files with 933 additions and 0 deletions
--- a/database/query/query_test.go
+++ b/database/query/query_test.go
@ -0,0 +1,104 @@
+package query
+
+import (
+	"testing"
+)
+
+var (
+	// copied from https://github.com/tidwall/gjson/blob/master/gjson_test.go
+	testJson = `{"age":100, "name":{"here":"B\\\"R"},
+  "noop":{"what is a wren?":"a bird"},
+  "happy":true,"immortal":false,
+  "items":[1,2,3,{"tags":[1,2,3],"points":[[1,2],[3,4]]},4,5,6,7],
+  "arr":["1",2,"3",{"hello":"world"},"4",5],
+  "vals":[1,2,3,{"sadf":sdf"asdf"}],"name":{"first":"tom","last":null},
+  "created":"2014-05-16T08:28:06.989Z",
+  "loggy":{
+  	"programmers": [
+    	    {
+    	        "firstName": "Brett",
+    	        "lastName": "McLaughlin",
+    	        "email": "aaaa",
+  			"tag": "good"
+    	    },
+    	    {
+    	        "firstName": "Jason",
+    	        "lastName": "Hunter",
+    	        "email": "bbbb",
+  			"tag": "bad"
+    	    },
+    	    {
+    	        "firstName": "Elliotte",
+    	        "lastName": "Harold",
+    	        "email": "cccc",
+  			"tag":, "good"
+    	    },
+  		{
+  			"firstName": 1002.3,
+  			"age": 101
+  		}
+    	]
+  },
+  "lastly":{"yay":"final"},
+	"temperature": 120.413
+}`
+)
+
+func testQuery(t *testing.T, f Fetcher, shouldMatch bool, condition Condition) {
+	q := MustCompile("test:")
+	err := q.Append(condition)
+	if err != nil {
+		t.Errorf("append failed: %s", err)
+	}
+
+	matched := q.Matches(f)
+	switch {
+	case matched && !shouldMatch:
+		t.Errorf("query should match")
+	case !matched && shouldMatch:
+		t.Errorf("query should not match")
+	}
+}
+
+func TestQuery(t *testing.T) {
+
+	// if !gjson.Valid(testJson) {
+	// 	t.Fatal("test json is invalid")
+	// }
+	f := NewJSONFetcher(testJson)
+
+	testQuery(t, f, true, Where("age", Equals, 100))
+	testQuery(t, f, true, Where("age", GreaterThan, uint8(99)))
+	testQuery(t, f, true, Where("age", GreaterThanOrEqual, 99))
+	testQuery(t, f, true, Where("age", GreaterThanOrEqual, 100))
+	testQuery(t, f, true, Where("age", LessThan, 101))
+	testQuery(t, f, true, Where("age", LessThanOrEqual, "101"))
+	testQuery(t, f, true, Where("age", LessThanOrEqual, 100))
+
+	testQuery(t, f, true, Where("temperature", FloatEquals, 120.413))
+	testQuery(t, f, true, Where("temperature", FloatGreaterThan, 120))
+	testQuery(t, f, true, Where("temperature", FloatGreaterThanOrEqual, 120))
+	testQuery(t, f, true, Where("temperature", FloatGreaterThanOrEqual, 120.413))
+	testQuery(t, f, true, Where("temperature", FloatLessThan, 121))
+	testQuery(t, f, true, Where("temperature", FloatLessThanOrEqual, "121"))
+	testQuery(t, f, true, Where("temperature", FloatLessThanOrEqual, "120.413"))
+
+	testQuery(t, f, true, Where("lastly.yay", Matches, "final"))
+	testQuery(t, f, true, Where("lastly.yay", Contains, "ina"))
+	testQuery(t, f, true, Where("lastly.yay", StartsWith, "fin"))
+	testQuery(t, f, true, Where("lastly.yay", EndsWith, "nal"))
+	testQuery(t, f, true, Where("lastly.yay", In, "draft,final"))
+	testQuery(t, f, true, Where("lastly.yay", In, "final,draft"))
+
+	testQuery(t, f, true, Where("happy", Is, "true"))
+	testQuery(t, f, true, Where("happy", Is, "t"))
+	testQuery(t, f, true, Where("happy", Is, "1"))
+	testQuery(t, f, true, Not(Where("happy", Is, "false")))
+	testQuery(t, f, true, Not(Where("happy", Is, "f")))
+	testQuery(t, f, true, Not(Where("happy", Is, "0")))
+
+	testQuery(t, f, true, Where("happy", Exists, nil))
+
+	testQuery(t, f, true, Where("created", Matches, "^2014-[0-9]{2}-[0-9]{2}T"))
+
+}