@@ -2,8 +2,6 @@ package command
22
33import (
44 "crypto/rand"
5- "io/ioutil"
6- "os"
75 "testing"
86
97 "github.com/docker/cli/cli/config/configfile"
@@ -39,21 +37,17 @@ func testDefaultMetadata() store.Metadata {
3937 }
4038}
4139
42- func testStore (t * testing.T , meta store.Metadata , tls store.ContextTLSData ) (store.Store , func ()) {
43- testDir , err := ioutil .TempDir ("" , t .Name ())
44- assert .NilError (t , err )
45- s := & ContextStoreWithDefault {
46- Store : store .New (testDir , testCfg ),
40+ func testStore (t * testing.T , meta store.Metadata , tls store.ContextTLSData ) store.Store {
41+ t .Helper ()
42+ return & ContextStoreWithDefault {
43+ Store : store .New (t .TempDir (), testCfg ),
4744 Resolver : func () (* DefaultContext , error ) {
4845 return & DefaultContext {
4946 Meta : meta ,
5047 TLS : tls ,
5148 }, nil
5249 },
5350 }
54- return s , func () {
55- _ = os .RemoveAll (testDir )
56- }
5751}
5852
5953func TestDefaultContextInitializer (t * testing.T ) {
@@ -78,7 +72,7 @@ func TestExportDefaultImport(t *testing.T) {
7872 rand .Read (file1 )
7973 file2 := make ([]byte , 3700 )
8074 rand .Read (file2 )
81- s , cleanup := testStore (t , testDefaultMetadata (), store.ContextTLSData {
75+ s := testStore (t , testDefaultMetadata (), store.ContextTLSData {
8276 Endpoints : map [string ]store.EndpointTLSData {
8377 "ep2" : {
8478 Files : map [string ][]byte {
@@ -88,7 +82,6 @@ func TestExportDefaultImport(t *testing.T) {
8882 },
8983 },
9084 })
91- defer cleanup ()
9285 r := store .Export ("default" , s )
9386 defer r .Close ()
9487 err := store .Import ("dest" , s , r )
@@ -127,26 +120,23 @@ func TestExportDefaultImport(t *testing.T) {
127120
128121func TestListDefaultContext (t * testing.T ) {
129122 meta := testDefaultMetadata ()
130- s , cleanup := testStore (t , meta , store.ContextTLSData {})
131- defer cleanup ()
123+ s := testStore (t , meta , store.ContextTLSData {})
132124 result , err := s .List ()
133125 assert .NilError (t , err )
134126 assert .Equal (t , 1 , len (result ))
135127 assert .DeepEqual (t , meta , result [0 ])
136128}
137129
138130func TestGetDefaultContextStorageInfo (t * testing.T ) {
139- s , cleanup := testStore (t , testDefaultMetadata (), store.ContextTLSData {})
140- defer cleanup ()
131+ s := testStore (t , testDefaultMetadata (), store.ContextTLSData {})
141132 result := s .GetStorageInfo (DefaultContextName )
142133 assert .Equal (t , "<IN MEMORY>" , result .MetadataPath )
143134 assert .Equal (t , "<IN MEMORY>" , result .TLSPath )
144135}
145136
146137func TestGetDefaultContextMetadata (t * testing.T ) {
147138 meta := testDefaultMetadata ()
148- s , cleanup := testStore (t , meta , store.ContextTLSData {})
149- defer cleanup ()
139+ s := testStore (t , meta , store.ContextTLSData {})
150140 result , err := s .GetMetadata (DefaultContextName )
151141 assert .NilError (t , err )
152142 assert .Equal (t , DefaultContextName , result .Name )
@@ -156,8 +146,7 @@ func TestGetDefaultContextMetadata(t *testing.T) {
156146
157147func TestErrCreateDefault (t * testing.T ) {
158148 meta := testDefaultMetadata ()
159- s , cleanup := testStore (t , meta , store.ContextTLSData {})
160- defer cleanup ()
149+ s := testStore (t , meta , store.ContextTLSData {})
161150 err := s .CreateOrUpdate (store.Metadata {
162151 Endpoints : map [string ]interface {}{
163152 "ep1" : endpoint {Foo : "bar" },
@@ -170,16 +159,14 @@ func TestErrCreateDefault(t *testing.T) {
170159
171160func TestErrRemoveDefault (t * testing.T ) {
172161 meta := testDefaultMetadata ()
173- s , cleanup := testStore (t , meta , store.ContextTLSData {})
174- defer cleanup ()
162+ s := testStore (t , meta , store.ContextTLSData {})
175163 err := s .Remove ("default" )
176164 assert .Error (t , err , "default context cannot be removed" )
177165}
178166
179167func TestErrTLSDataError (t * testing.T ) {
180168 meta := testDefaultMetadata ()
181- s , cleanup := testStore (t , meta , store.ContextTLSData {})
182- defer cleanup ()
169+ s := testStore (t , meta , store.ContextTLSData {})
183170 _ , err := s .GetTLSData ("default" , "noop" , "noop" )
184171 assert .Check (t , store .IsErrTLSDataDoesNotExist (err ))
185172}
0 commit comments