Просмотр исходного кода

Merge pull request #9188 from somaopensource/8777-fix

Fix for #8777 (continuing PR #9061)
Jessie Frazelle 10 лет назад
Родитель
Сommit
d7626e97b6
3 измененных файлов с 72 добавлено и 1 удалено
  1. 10 0
      api/client/commands.go
  2. 59 0
      integration-cli/docker_cli_images_test.go
  3. 3 1
      pkg/parsers/filters/parse.go

+ 10 - 0
api/client/commands.go

@@ -47,6 +47,10 @@ const (
 	tarHeaderSize = 512
 	tarHeaderSize = 512
 )
 )
 
 
+var (
+	acceptedImageFilterTags = map[string]struct{}{"dangling": {}}
+)
+
 func (cli *DockerCli) CmdHelp(args ...string) error {
 func (cli *DockerCli) CmdHelp(args ...string) error {
 	if len(args) > 1 {
 	if len(args) > 1 {
 		method, exists := cli.getMethod(args[:2]...)
 		method, exists := cli.getMethod(args[:2]...)
@@ -1349,6 +1353,12 @@ func (cli *DockerCli) CmdImages(args ...string) error {
 		}
 		}
 	}
 	}
 
 
+	for name := range imageFilterArgs {
+		if _, ok := acceptedImageFilterTags[name]; !ok {
+			return fmt.Errorf("Invalid filter '%s'", name)
+		}
+	}
+
 	matchName := cmd.Arg(0)
 	matchName := cmd.Arg(0)
 	// FIXME: --viz and --tree are deprecated. Remove them in a future version.
 	// FIXME: --viz and --tree are deprecated. Remove them in a future version.
 	if *flViz || *flTree {
 	if *flViz || *flTree {

+ 59 - 0
integration-cli/docker_cli_images_test.go

@@ -1,7 +1,10 @@
 package main
 package main
 
 
 import (
 import (
+	"fmt"
 	"os/exec"
 	"os/exec"
+	"reflect"
+	"sort"
 	"strings"
 	"strings"
 	"testing"
 	"testing"
 	"time"
 	"time"
@@ -63,3 +66,59 @@ func TestImagesOrderedByCreationDate(t *testing.T) {
 
 
 	logDone("images - ordering by creation date")
 	logDone("images - ordering by creation date")
 }
 }
+
+func TestImagesErrorWithInvalidFilterNameTest(t *testing.T) {
+	imagesCmd := exec.Command(dockerBinary, "images", "-f", "FOO=123")
+	out, _, err := runCommandWithOutput(imagesCmd)
+	if !strings.Contains(out, "Invalid filter") {
+		t.Fatalf("error should occur when listing images with invalid filter name FOO, %s, %v", out, err)
+	}
+
+	logDone("images - invalid filter name check working")
+}
+
+func TestImagesFilterWhiteSpaceTrimmingAndLowerCasingWorking(t *testing.T) {
+	imageName := "images_filter_test"
+	defer deleteAllContainers()
+	defer deleteImages(imageName)
+	buildImage(imageName,
+		`FROM scratch
+		 RUN touch /test/foo
+		 RUN touch /test/bar
+		 RUN touch /test/baz`, true)
+
+	filters := []string{
+		"dangling=true",
+		"Dangling=true",
+		" dangling=true",
+		"dangling=true ",
+		"dangling = true",
+	}
+
+	imageListings := make([][]string, 5, 5)
+	for idx, filter := range filters {
+		cmd := exec.Command(dockerBinary, "images", "-f", filter)
+		out, _, err := runCommandWithOutput(cmd)
+		if err != nil {
+			t.Fatal(err)
+		}
+		listing := strings.Split(out, "\n")
+		sort.Strings(listing)
+		imageListings[idx] = listing
+	}
+
+	for idx, listing := range imageListings {
+		if idx < 4 && !reflect.DeepEqual(listing, imageListings[idx+1]) {
+			for idx, errListing := range imageListings {
+				fmt.Printf("out %d", idx)
+				for _, image := range errListing {
+					fmt.Print(image)
+				}
+				fmt.Print("")
+			}
+			t.Fatalf("All output must be the same")
+		}
+	}
+
+	logDone("images - white space trimming and lower casing")
+}

+ 3 - 1
pkg/parsers/filters/parse.go

@@ -29,7 +29,9 @@ func ParseFlag(arg string, prev Args) (Args, error) {
 	}
 	}
 
 
 	f := strings.SplitN(arg, "=", 2)
 	f := strings.SplitN(arg, "=", 2)
-	filters[f[0]] = append(filters[f[0]], f[1])
+	name := strings.ToLower(strings.TrimSpace(f[0]))
+	value := strings.TrimSpace(f[1])
+	filters[name] = append(filters[name], value)
 
 
 	return filters, nil
 	return filters, nil
 }
 }