@@ -21,7 +21,7 @@ func TestNewPruneCommandErrors(t *testing.T) {
2121 name string
2222 args []string
2323 expectedError string
24- imagesPruneFunc func (pruneFilter client.Filters ) (image. PruneReport , error )
24+ imagesPruneFunc func (client.ImagePruneOptions ) (client. ImagePruneResult , error )
2525 }{
2626 {
2727 name : "wrong-args" ,
@@ -32,8 +32,8 @@ func TestNewPruneCommandErrors(t *testing.T) {
3232 name : "prune-error" ,
3333 args : []string {"--force" },
3434 expectedError : "something went wrong" ,
35- imagesPruneFunc : func (pruneFilter client.Filters ) (image. PruneReport , error ) {
36- return image. PruneReport {}, errors .New ("something went wrong" )
35+ imagesPruneFunc : func (client.ImagePruneOptions ) (client. ImagePruneResult , error ) {
36+ return client. ImagePruneResult {}, errors .New ("something went wrong" )
3737 },
3838 },
3939 }
@@ -54,43 +54,47 @@ func TestNewPruneCommandSuccess(t *testing.T) {
5454 testCases := []struct {
5555 name string
5656 args []string
57- imagesPruneFunc func (pruneFilter client.Filters ) (image. PruneReport , error )
57+ imagesPruneFunc func (client.ImagePruneOptions ) (client. ImagePruneResult , error )
5858 }{
5959 {
6060 name : "all" ,
6161 args : []string {"--all" },
62- imagesPruneFunc : func (pruneFilter client.Filters ) (image. PruneReport , error ) {
63- assert .Check (t , pruneFilter ["dangling" ]["false" ])
64- return image. PruneReport {}, nil
62+ imagesPruneFunc : func (opts client.ImagePruneOptions ) (client. ImagePruneResult , error ) {
63+ assert .Check (t , opts . Filters ["dangling" ]["false" ])
64+ return client. ImagePruneResult {}, nil
6565 },
6666 },
6767 {
6868 name : "force-deleted" ,
6969 args : []string {"--force" },
70- imagesPruneFunc : func (pruneFilter client.Filters ) (image.PruneReport , error ) {
71- assert .Check (t , pruneFilter ["dangling" ]["true" ])
72- return image.PruneReport {
73- ImagesDeleted : []image.DeleteResponse {{Deleted : "image1" }},
74- SpaceReclaimed : 1 ,
70+ imagesPruneFunc : func (opts client.ImagePruneOptions ) (client.ImagePruneResult , error ) {
71+ assert .Check (t , opts .Filters ["dangling" ]["true" ])
72+ return client.ImagePruneResult {
73+ Report : image.PruneReport {
74+ ImagesDeleted : []image.DeleteResponse {{Deleted : "image1" }},
75+ SpaceReclaimed : 1 ,
76+ },
7577 }, nil
7678 },
7779 },
7880 {
7981 name : "label-filter" ,
8082 args : []string {"--force" , "--filter" , "label=foobar" },
81- imagesPruneFunc : func (pruneFilter client.Filters ) (image. PruneReport , error ) {
82- assert .Check (t , pruneFilter ["label" ]["foobar" ])
83- return image. PruneReport {}, nil
83+ imagesPruneFunc : func (opts client.ImagePruneOptions ) (client. ImagePruneResult , error ) {
84+ assert .Check (t , opts . Filters ["label" ]["foobar" ])
85+ return client. ImagePruneResult {}, nil
8486 },
8587 },
8688 {
8789 name : "force-untagged" ,
8890 args : []string {"--force" },
89- imagesPruneFunc : func (pruneFilter client.Filters ) (image.PruneReport , error ) {
90- assert .Check (t , pruneFilter ["dangling" ]["true" ])
91- return image.PruneReport {
92- ImagesDeleted : []image.DeleteResponse {{Untagged : "image1" }},
93- SpaceReclaimed : 2 ,
91+ imagesPruneFunc : func (opts client.ImagePruneOptions ) (client.ImagePruneResult , error ) {
92+ assert .Check (t , opts .Filters ["dangling" ]["true" ])
93+ return client.ImagePruneResult {
94+ Report : image.PruneReport {
95+ ImagesDeleted : []image.DeleteResponse {{Untagged : "image1" }},
96+ SpaceReclaimed : 2 ,
97+ },
9498 }, nil
9599 },
96100 },
@@ -116,8 +120,8 @@ func TestPrunePromptTermination(t *testing.T) {
116120 t .Cleanup (cancel )
117121
118122 cli := test .NewFakeCli (& fakeClient {
119- imagesPruneFunc : func (pruneFilter client.Filters ) (image. PruneReport , error ) {
120- return image. PruneReport {}, errors .New ("fakeClient imagesPruneFunc should not be called" )
123+ imagesPruneFunc : func (client.ImagePruneOptions ) (client. ImagePruneResult , error ) {
124+ return client. ImagePruneResult {}, errors .New ("fakeClient imagesPruneFunc should not be called" )
121125 },
122126 })
123127 cmd := newPruneCommand (cli )
0 commit comments