mirror of
https://github.com/ko-build/ko.git
synced 2025-03-20 20:55:00 +02:00
152 lines
3.1 KiB
Go
152 lines
3.1 KiB
Go
// Copyright 2018 Google LLC All Rights Reserved.
|
|
//
|
|
// 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 resolve
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
webPod = `apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
labels:
|
|
# comments should be preserved
|
|
app: web
|
|
name: rss-site
|
|
`
|
|
dbPod = `apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
labels:
|
|
# comments should be preserved
|
|
app: db
|
|
name: rss-db
|
|
`
|
|
podList = `apiVersion: v1
|
|
kind: List
|
|
metadata:
|
|
resourceVersion: ""
|
|
selfLink: ""
|
|
items:
|
|
- apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
labels:
|
|
app: web
|
|
name: rss-site
|
|
- apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
labels:
|
|
app: db
|
|
name: rss-db
|
|
`
|
|
webSelector = `app=web`
|
|
notWebSelector = `app!=web`
|
|
nopSelector = `foo!=bark`
|
|
|
|
webPodList = `apiVersion: v1
|
|
items:
|
|
- apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
labels:
|
|
app: web
|
|
name: rss-site
|
|
kind: List
|
|
metadata:
|
|
resourceVersion: ""
|
|
selfLink: ""
|
|
`
|
|
dbPodList = `apiVersion: v1
|
|
items:
|
|
- apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
labels:
|
|
app: db
|
|
name: rss-db
|
|
kind: List
|
|
metadata:
|
|
resourceVersion: ""
|
|
selfLink: ""
|
|
`
|
|
)
|
|
|
|
var bothPods = strings.Join([]string{webPod, dbPod}, "\n---\n")
|
|
|
|
func TestSelector(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
input string
|
|
selector string
|
|
expected string
|
|
}{{
|
|
desc: "single object with matching selector",
|
|
input: webPod,
|
|
selector: webSelector,
|
|
expected: webPod,
|
|
}, {
|
|
desc: "single object with non-matching selector",
|
|
input: webPod,
|
|
selector: notWebSelector,
|
|
expected: ``,
|
|
}, {
|
|
desc: "selector matching 1 of two objects",
|
|
input: bothPods,
|
|
selector: webSelector,
|
|
expected: webPod,
|
|
}, {
|
|
desc: "selector matching 1 of two objects",
|
|
input: bothPods,
|
|
selector: notWebSelector,
|
|
expected: dbPod,
|
|
}, {
|
|
desc: "selector matching both objects",
|
|
input: bothPods,
|
|
selector: nopSelector,
|
|
expected: bothPods,
|
|
}, {
|
|
desc: "selector matching elements of list object",
|
|
input: podList,
|
|
selector: webSelector,
|
|
expected: webPodList,
|
|
}, {
|
|
desc: "selector matching elements of list object",
|
|
input: podList,
|
|
selector: notWebSelector,
|
|
expected: dbPodList,
|
|
}, {
|
|
desc: "selector matching all elements of list object",
|
|
input: podList,
|
|
selector: ``,
|
|
expected: podList,
|
|
}}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
filtered, err := FilterBySelector([]byte(test.input), test.selector)
|
|
if err != nil {
|
|
t.Errorf("unexpected error: %v", err)
|
|
}
|
|
if strings.TrimSpace(string(filtered)) != strings.TrimSpace(test.expected) {
|
|
t.Errorf("expected \n%v\n to equal \n%v\n ", string(filtered), test.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|