@@ -10,10 +10,9 @@ import (
1010
1111 "github.com/docker/cli/cli/streams"
1212 "github.com/docker/cli/internal/test"
13- "github.com/moby/moby/api/types/filters"
1413 "github.com/moby/moby/api/types/image"
14+ "github.com/moby/moby/client"
1515 "gotest.tools/v3/assert"
16- is "gotest.tools/v3/assert/cmp"
1716 "gotest.tools/v3/golden"
1817)
1918
@@ -22,7 +21,7 @@ func TestNewPruneCommandErrors(t *testing.T) {
2221 name string
2322 args []string
2423 expectedError string
25- imagesPruneFunc func (pruneFilter filters. Args ) (image.PruneReport , error )
24+ imagesPruneFunc func (pruneFilter client. Filters ) (image.PruneReport , error )
2625 }{
2726 {
2827 name : "wrong-args" ,
@@ -33,7 +32,7 @@ func TestNewPruneCommandErrors(t *testing.T) {
3332 name : "prune-error" ,
3433 args : []string {"--force" },
3534 expectedError : "something went wrong" ,
36- imagesPruneFunc : func (pruneFilter filters. Args ) (image.PruneReport , error ) {
35+ imagesPruneFunc : func (pruneFilter client. Filters ) (image.PruneReport , error ) {
3736 return image.PruneReport {}, errors .New ("something went wrong" )
3837 },
3938 },
@@ -55,21 +54,21 @@ func TestNewPruneCommandSuccess(t *testing.T) {
5554 testCases := []struct {
5655 name string
5756 args []string
58- imagesPruneFunc func (pruneFilter filters. Args ) (image.PruneReport , error )
57+ imagesPruneFunc func (pruneFilter client. Filters ) (image.PruneReport , error )
5958 }{
6059 {
6160 name : "all" ,
6261 args : []string {"--all" },
63- imagesPruneFunc : func (pruneFilter filters. Args ) (image.PruneReport , error ) {
64- assert .Check (t , is . Equal ( "false" , pruneFilter . Get ( "dangling" )[ 0 ]) )
62+ imagesPruneFunc : func (pruneFilter client. Filters ) (image.PruneReport , error ) {
63+ assert .Check (t , pruneFilter [ "dangling" ][ "false" ] )
6564 return image.PruneReport {}, nil
6665 },
6766 },
6867 {
6968 name : "force-deleted" ,
7069 args : []string {"--force" },
71- imagesPruneFunc : func (pruneFilter filters. Args ) (image.PruneReport , error ) {
72- assert .Check (t , is . Equal ( "true" , pruneFilter . Get ( "dangling" )[ 0 ]) )
70+ imagesPruneFunc : func (pruneFilter client. Filters ) (image.PruneReport , error ) {
71+ assert .Check (t , pruneFilter [ "dangling" ][ "true" ] )
7372 return image.PruneReport {
7473 ImagesDeleted : []image.DeleteResponse {{Deleted : "image1" }},
7574 SpaceReclaimed : 1 ,
@@ -79,16 +78,16 @@ func TestNewPruneCommandSuccess(t *testing.T) {
7978 {
8079 name : "label-filter" ,
8180 args : []string {"--force" , "--filter" , "label=foobar" },
82- imagesPruneFunc : func (pruneFilter filters. Args ) (image.PruneReport , error ) {
83- assert .Check (t , is . Equal ( "foobar" , pruneFilter . Get ( "label" )[ 0 ]) )
81+ imagesPruneFunc : func (pruneFilter client. Filters ) (image.PruneReport , error ) {
82+ assert .Check (t , pruneFilter [ "label" ][ "foobar" ] )
8483 return image.PruneReport {}, nil
8584 },
8685 },
8786 {
8887 name : "force-untagged" ,
8988 args : []string {"--force" },
90- imagesPruneFunc : func (pruneFilter filters. Args ) (image.PruneReport , error ) {
91- assert .Check (t , is . Equal ( "true" , pruneFilter . Get ( "dangling" )[ 0 ]) )
89+ imagesPruneFunc : func (pruneFilter client. Filters ) (image.PruneReport , error ) {
90+ assert .Check (t , pruneFilter [ "dangling" ][ "true" ] )
9291 return image.PruneReport {
9392 ImagesDeleted : []image.DeleteResponse {{Untagged : "image1" }},
9493 SpaceReclaimed : 2 ,
@@ -117,7 +116,7 @@ func TestPrunePromptTermination(t *testing.T) {
117116 t .Cleanup (cancel )
118117
119118 cli := test .NewFakeCli (& fakeClient {
120- imagesPruneFunc : func (pruneFilter filters. Args ) (image.PruneReport , error ) {
119+ imagesPruneFunc : func (pruneFilter client. Filters ) (image.PruneReport , error ) {
121120 return image.PruneReport {}, errors .New ("fakeClient imagesPruneFunc should not be called" )
122121 },
123122 })
0 commit comments