347 lines
		
	
	
		
			7.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			347 lines
		
	
	
		
			7.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
/*
 | 
						|
   Copyright The containerd Authors.
 | 
						|
 | 
						|
   Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
   you may not use this file except in compliance with the License.
 | 
						|
   You may obtain a copy of the License at
 | 
						|
 | 
						|
       http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 | 
						|
   Unless required by applicable law or agreed to in writing, software
 | 
						|
   distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
   See the License for the specific language governing permissions and
 | 
						|
   limitations under the License.
 | 
						|
*/
 | 
						|
 | 
						|
package filters
 | 
						|
 | 
						|
import (
 | 
						|
	"reflect"
 | 
						|
	"strings"
 | 
						|
	"testing"
 | 
						|
)
 | 
						|
 | 
						|
func TestFilters(t *testing.T) {
 | 
						|
	type cEntry struct {
 | 
						|
		Name   string
 | 
						|
		Other  string
 | 
						|
		Labels map[string]string
 | 
						|
	}
 | 
						|
 | 
						|
	corpusS := []cEntry{
 | 
						|
		{
 | 
						|
			Name: "foo",
 | 
						|
			Labels: map[string]string{
 | 
						|
				"foo": "true",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "bar",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "foo",
 | 
						|
			Labels: map[string]string{
 | 
						|
				"foo":                "present",
 | 
						|
				"more complex label": "present",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "bar",
 | 
						|
			Labels: map[string]string{
 | 
						|
				"bar": "true",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "fooer",
 | 
						|
			Labels: map[string]string{
 | 
						|
				"more complex label with \\ and \"": "present",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "fooer",
 | 
						|
			Labels: map[string]string{
 | 
						|
				"more complex label with \\ and \".post": "present",
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name:  "baz",
 | 
						|
			Other: "too complex, yo",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name:  "bazo",
 | 
						|
			Other: "abc",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			Name: "compound",
 | 
						|
			Labels: map[string]string{
 | 
						|
				"foo": "omg_asdf.asdf-qwer",
 | 
						|
			},
 | 
						|
		},
 | 
						|
	}
 | 
						|
 | 
						|
	var corpus []interface{}
 | 
						|
	for _, entry := range corpusS {
 | 
						|
		corpus = append(corpus, entry)
 | 
						|
	}
 | 
						|
 | 
						|
	// adapt shows an example of how to build an adaptor function for a type.
 | 
						|
	adapt := func(o interface{}) Adaptor {
 | 
						|
		obj := o.(cEntry)
 | 
						|
		return AdapterFunc(func(fieldpath []string) (string, bool) {
 | 
						|
			switch fieldpath[0] {
 | 
						|
			case "name":
 | 
						|
				return obj.Name, len(obj.Name) > 0
 | 
						|
			case "other":
 | 
						|
				return obj.Other, len(obj.Other) > 0
 | 
						|
			case "labels":
 | 
						|
				value, ok := obj.Labels[strings.Join(fieldpath[1:], ".")]
 | 
						|
				return value, ok
 | 
						|
			}
 | 
						|
 | 
						|
			return "", false
 | 
						|
		})
 | 
						|
	}
 | 
						|
 | 
						|
	for _, testcase := range []struct {
 | 
						|
		name      string
 | 
						|
		input     string
 | 
						|
		expected  []interface{}
 | 
						|
		errString string
 | 
						|
	}{
 | 
						|
		{
 | 
						|
			name:     "Empty",
 | 
						|
			input:    "",
 | 
						|
			expected: corpus,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:     "Present",
 | 
						|
			input:    "name",
 | 
						|
			expected: corpus,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "LabelPresent",
 | 
						|
			input: "labels.foo",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[0],
 | 
						|
				corpus[2],
 | 
						|
				corpus[8],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "NameAndLabelPresent",
 | 
						|
			input: "labels.foo,name",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[0],
 | 
						|
				corpus[2],
 | 
						|
				corpus[8],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "LabelValue",
 | 
						|
			input: "labels.foo==true",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[0],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "LabelValuePunctuated",
 | 
						|
			input: "labels.foo==omg_asdf.asdf-qwer",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[8],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "LabelValueNoAltQuoting",
 | 
						|
			input:     "labels.|foo|==omg_asdf.asdf-qwer",
 | 
						|
			errString: "filters: parse error: [labels. >|||< foo|==omg_asdf.asdf-qwer]: invalid quote encountered",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "Name",
 | 
						|
			input: "name==bar",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[1],
 | 
						|
				corpus[3],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "NameNotEqual",
 | 
						|
			input: "name!=bar",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[0],
 | 
						|
				corpus[2],
 | 
						|
				corpus[4],
 | 
						|
				corpus[5],
 | 
						|
				corpus[6],
 | 
						|
				corpus[7],
 | 
						|
				corpus[8],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "NameAndLabelPresent",
 | 
						|
			input: "name==bar,labels.bar",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[3],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "QuotedValue",
 | 
						|
			input: "other==\"too complex, yo\"",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[6],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "RegexpValue",
 | 
						|
			input: "other~=[abc]+,name!=foo",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[6],
 | 
						|
				corpus[7],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "RegexpQuotedValue",
 | 
						|
			input: "other~=/[abc]+/,name!=foo",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[6],
 | 
						|
				corpus[7],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "RegexpQuotedValue",
 | 
						|
			input: "other~=/[abc]{1,2}/,name!=foo",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[6],
 | 
						|
				corpus[7],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "RegexpQuotedValueGarbage",
 | 
						|
			input: "other~=/[abc]{0,1}\"\\//,name!=foo",
 | 
						|
			// valid syntax, but doesn't match anything
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "NameAndLabelValue",
 | 
						|
			input: "name==bar,labels.bar==true",
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[3],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "NameAndLabelValueNoMatch",
 | 
						|
			input: "name==bar,labels.bar==wrong",
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "LabelQuotedFieldPathPresent",
 | 
						|
			input: `name==foo,labels."more complex label"`,
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[2],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "LabelQuotedFieldPathPresentWithQuoted",
 | 
						|
			input: `labels."more complex label with \\ and \""==present`,
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[4],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:  "LabelQuotedFieldPathPresentWithQuotedEmbed",
 | 
						|
			input: `labels."more complex label with \\ and \"".post==present`,
 | 
						|
			expected: []interface{}{
 | 
						|
				corpus[5],
 | 
						|
			},
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "LabelQuotedFieldPathPresentWithQuotedEmbedInvalid",
 | 
						|
			input:     `labels.?"more complex label with \\ and \"".post==present`,
 | 
						|
			errString: `filters: parse error: [labels. >|?|< "more complex label with \\ and \"".post==present]: expected field or quoted`,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "TrailingComma",
 | 
						|
			input:     "name==foo,",
 | 
						|
			errString: `filters: parse error: [name==foo,]: expected field or quoted`,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "TrailingFieldSeparator",
 | 
						|
			input:     "labels.",
 | 
						|
			errString: `filters: parse error: [labels.]: expected field or quoted`,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "MissingValue",
 | 
						|
			input:     "image~=,id?=?fbaq",
 | 
						|
			errString: `filters: parse error: [image~= >|,|< id?=?fbaq]: expected value or quoted`,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "FieldQuotedLiteralNotTerminated",
 | 
						|
			input:     "labels.ns/key==value",
 | 
						|
			errString: `filters: parse error: [labels.ns >|/|< key==value]: quoted literal not terminated`,
 | 
						|
		},
 | 
						|
		{
 | 
						|
			name:      "ValueQuotedLiteralNotTerminated",
 | 
						|
			input:     "labels.key==/value",
 | 
						|
			errString: `filters: parse error: [labels.key== >|/|< value]: quoted literal not terminated`,
 | 
						|
		},
 | 
						|
	} {
 | 
						|
		t.Run(testcase.name, func(t *testing.T) {
 | 
						|
			filter, err := Parse(testcase.input)
 | 
						|
			if testcase.errString != "" {
 | 
						|
				if err == nil {
 | 
						|
					t.Fatalf("expected an error, but received nil")
 | 
						|
				}
 | 
						|
				if err.Error() != testcase.errString {
 | 
						|
					t.Fatalf("error %v != %v", err, testcase.errString)
 | 
						|
				}
 | 
						|
 | 
						|
				return
 | 
						|
			}
 | 
						|
			if err != nil {
 | 
						|
				t.Fatal(err)
 | 
						|
			}
 | 
						|
 | 
						|
			if filter == nil {
 | 
						|
				t.Fatal("filter should not be nil")
 | 
						|
			}
 | 
						|
 | 
						|
			var results []interface{}
 | 
						|
			for _, item := range corpus {
 | 
						|
				adaptor := adapt(item)
 | 
						|
				if filter.Match(adaptor) {
 | 
						|
					results = append(results, item)
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			if !reflect.DeepEqual(results, testcase.expected) {
 | 
						|
				t.Fatalf("%q: %#v != %#v", testcase.input, results, testcase.expected)
 | 
						|
			}
 | 
						|
		})
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func TestOperatorStrings(t *testing.T) {
 | 
						|
	for _, testcase := range []struct {
 | 
						|
		op       operator
 | 
						|
		expected string
 | 
						|
	}{
 | 
						|
		{operatorPresent, "?"},
 | 
						|
		{operatorEqual, "=="},
 | 
						|
		{operatorNotEqual, "!="},
 | 
						|
		{operatorMatches, "~="},
 | 
						|
		{10, "unknown"},
 | 
						|
	} {
 | 
						|
		if !reflect.DeepEqual(testcase.op.String(), testcase.expected) {
 | 
						|
			t.Fatalf("return value unexpected: %v != %v", testcase.op.String(), testcase.expected)
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func FuzzFiltersParse(f *testing.F) {
 | 
						|
	f.Add("foo=bar")
 | 
						|
	f.Fuzz(func(t *testing.T, expr string) {
 | 
						|
		filter, err := Parse(expr)
 | 
						|
		if filter != nil && err != nil {
 | 
						|
			t.Fatal("either filter or err must be non-nil")
 | 
						|
		}
 | 
						|
	})
 | 
						|
}
 |