timsolov / rest-query-parser Goto Github PK

func Test_RemoveOrEntries(t *testing.T) {
	type testCase struct {
		name string
		urlQuery string
		filterToRemove string
		wantWhere             string
	}
	tests := []testCase{
		{
			name: "should fix OR statements after removing EndOR filter with 2 items",
			urlQuery: "?test1[eq]=test10|test2[eq]=test10",
			filterToRemove: "test2",
			wantWhere: "test1 = ?",
		},
		{
			name: "should fix OR statements after removing StartOR filter with 2 items",
			urlQuery: "?test1[eq]=test10|test2[eq]=test10",
			filterToRemove: "test1",
			wantWhere: "test2 = ?",
		},
		{
			name: "should fix OR statements after removing StartOR filter with 3 items",
			urlQuery: "?test1[eq]=test10|test2[eq]=test10|test3[eq]=test10",
			filterToRemove: "test1",
			wantWhere: "(test2 = ? OR test3 = ?)",
		},
		{
			name: "should fix OR statements after removing EndOR filter with 3 items",
			urlQuery: "?test1[eq]=test10|test2[eq]=test10|test3[eq]=test10",
			filterToRemove: "test3",
			wantWhere: "(test1 = ? OR test2 = ?)",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			// Arrange
			URL, _ := url.Parse(tt.urlQuery)
			q := rqp.NewQV(nil, rqp.Validations{
				"test1": nil,
				"test2": nil,
				"test3": nil,
			})
			_ = q.SetUrlQuery(URL.Query()).Parse()

			// Act
			_ = q.RemoveFilter(tt.filterToRemove)

			// Assert
			assert.Equal(t, tt.wantWhere, q.WHERE())
		})
	}
}

q2, _ := rqp.NewParse(r.URL.Query(), rqp.Validations{
		"limit:required": rqp.MinMax(10, 100),  // limit must present in the Query part and must be between 10 and 100 (default: Min(1))
		"sort":           rqp.In("id", "name"), // sort could be or not in the query but if it is present it must be equal to "in" or "name"
		"s":              rqp.In("one", "two"), // filter: s - string and equal
		"id:int":         nil,                  // filter: id is integer without additional validation
		"i:int": func(value interface{}) error { // filter: custom func for validating
			if value.(int) > 1 && value.(int) < 10 {
				return nil
			}
			return errors.New("i: must be greater then 1 and lower then 10")
		},
		"email": nil,
		"name":  nil,
	})