mapping_test.go

package rql

import (
	"fmt"
	"testing"

	"github.com/jinzhu/gorm"
)

func TestMapping(t *testing.T) {
	tests := []struct {
		name    string
		conf    Config
		input   []byte
		wantErr bool
		wantOut *Params
	}{
		{
			name: "test simple mapping of column names",
			conf: Config{
				ColumnFn: func(columnName string) string {
					return fmt.Sprintf("person.%s", gorm.ToDBName(columnName))
				},
				Model: new(struct {
					Age     int    `rql:"filter, sort"`
					Name    string `rql:"filter"`
					Address string `rql:"filter"`
				}),
				DefaultSort:  []string{"+age"},
				DefaultLimit: 25,
			},
			input: []byte(`{
				"filter": {
					"name": "foo",
					"age": 12,
					"$or": [
						{ "name": { "$in": ["foo", "bar"] } },
						{ "address": "DC" },
						{ "address": "Marvel" }
					],
					"$and": [
						{ "age": { "$neq": 10} },
						{ "age": { "$neq": 20} },
						{ "$or": [{ "age": 11 }, {"age": 10}] }
					]
				}
			}`),
			wantOut: &Params{
				Limit:      25,
				FilterExp:  "person.name = ? AND person.age = ? AND (person.name IN (?,?) OR person.address = ? OR person.address = ?) AND (person.age <> ? AND person.age <> ? AND (person.age = ? OR person.age = ?))",
				FilterArgs: []interface{}{"foo", 12, "foo", "bar", "DC", "Marvel", 10, 20, 11, 10},
				Sort:       "person.age asc",
			},
		},
		{
			name: "test simple mapping of column values",
			conf: Config{
				ValueFn: func(columnName string) func(interface{}) interface{} {
					return func(val interface{}) interface{} {
						if columnName == "age" {
							return val.(float64) + 20
						}
						return val
					}
				},
				Model: new(struct {
					Age     int    `rql:"filter, sort"`
					Name    string `rql:"filter"`
					Address string `rql:"filter"`
				}),
				DefaultSort:  []string{"+age"},
				DefaultLimit: 25,
			},
			input: []byte(`{
				"filter": {
					"name": "foo",
					"age": 12,
					"$or": [
						{ "age": 22 },
						{ "age": 32 }
					]
				}
			}`),
			wantOut: &Params{
				Limit:      25,
				FilterExp:  "name = ? AND age = ? AND (age = ? OR age = ?)",
				FilterArgs: []interface{}{"foo", 32, 42, 52},
				Sort:       "age asc",
			},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.conf.Log = t.Logf
			p, err := NewParser(tt.conf)
			if err != nil {
				t.Fatalf("failed to build parser: %v", err)
			}
			out, err := p.Parse(tt.input)
			if tt.wantErr != (err != nil) {
				t.Fatalf("want: %v\ngot:%v\nerr: %v", tt.wantErr, err != nil, err)
			}
			assertParams(t, out, tt.wantOut)
		})
	}
}