New Criteria API

model/criteria/criteria_test.go

@@ -0,0 +1,84 @@
+package criteria
+
+import (
+	"bytes"
+	"encoding/json"
+
+	. "github.com/onsi/ginkgo"
+	"github.com/onsi/gomega"
+)
+
+var _ = Describe("Criteria", func() {
+	var goObj Criteria
+	var jsonObj string
+	BeforeEach(func() {
+		goObj = Criteria{
+			Expression: All{
+				Contains{"title": "love"},
+				NotContains{"title": "hate"},
+				Any{
+					IsNot{"artist": "u2"},
+					Is{"album": "best of"},
+				},
+				All{
+					StartsWith{"comment": "this"},
+					InTheRange{"year": []int{1980, 1990}},
+				},
+			},
+			Sort:   "title",
+			Order:  "asc",
+			Max:    20,
+			Offset: 10,
+		}
+		var b bytes.Buffer
+		err := json.Compact(&b, []byte(`
+{
+	"all": [
+		{ "contains": {"title": "love"} },
+		{ "notContains": {"title": "hate"} },
+		{ "any": [
+				{ "isNot": {"artist": "u2"} },
+				{ "is": {"album": "best of"} }
+			]
+		},
+		{ "all": [
+				{ "startsWith": {"comment": "this"} },
+				{ "inTheRange": {"year":[1980,1990]} }
+			]
+		}
+	],
+	"sort": "title",
+	"order": "asc",
+	"max": 20,
+	"offset": 10
+}
+`))
+		if err != nil {
+			panic(err)
+		}
+		jsonObj = b.String()
+	})
+
+	It("generates valid SQL", func() {
+		sql, args, err := goObj.ToSql()
+		gomega.Expect(err).ToNot(gomega.HaveOccurred())
+		gomega.Expect(sql).To(gomega.Equal("(media_file.title ILIKE ? AND media_file.title NOT ILIKE ? AND (media_file.artist <> ? OR media_file.album = ?) AND (media_file.comment ILIKE ? AND (media_file.year >= ? AND media_file.year <= ?)))"))
+		gomega.Expect(args).To(gomega.ConsistOf("%love%", "%hate%", "u2", "best of", "this%", 1980, 1990))
+	})
+
+	It("marshals to JSON", func() {
+		j, err := json.Marshal(goObj)
+		gomega.Expect(err).ToNot(gomega.HaveOccurred())
+		gomega.Expect(string(j)).To(gomega.Equal(jsonObj))
+	})
+
+	It("is reversible to/from JSON", func() {
+		var newObj Criteria
+		err := json.Unmarshal([]byte(jsonObj), &newObj)
+		gomega.Expect(err).ToNot(gomega.HaveOccurred())
+		j, err := json.Marshal(newObj)
+		gomega.Expect(err).ToNot(gomega.HaveOccurred())
+		gomega.Expect(string(j)).To(gomega.Equal(jsonObj))
+	})
+
+})