@@ -19,6 +19,7 @@ package worker
19
19
import (
20
20
"encoding/json"
21
21
"os"
22
+ "path/filepath"
22
23
"testing"
23
24
"time"
24
25
@@ -67,106 +68,84 @@ func TestSetImageSources(t *testing.T) {
67
68
require .Empty (t , image .Labels )
68
69
69
70
// test setting one source
70
- fp , err := os .CreateTemp ("" , "test" )
71
- require .NoError (t , err )
72
- defer func () {
73
- _ = fp .Close ()
74
- _ = os .Remove (fp .Name ())
75
- }()
76
- info , err := fp .Stat ()
71
+ test := filepath .Join (t .TempDir (), "test" )
72
+ require .NoError (t , os .WriteFile (test , nil , 0644 ))
73
+ info , err := os .Stat (test )
77
74
require .NoError (t , err )
78
75
image = images.Image {}
79
- expected := ImageSources {fp . Name () : info .ModTime ()}
76
+ expected := ImageSources {test : info .ModTime ()}
80
77
err = SetImageSources (& image , expected )
81
78
require .NoError (t , err )
82
79
got , err := GetImageSources (image )
83
80
require .NoError (t , err )
84
- require .True (t , expected [fp . Name () ].Equal (got [fp . Name () ]), "dates mismatch" )
81
+ require .True (t , expected [test ].Equal (got [test ]), "dates mismatch" )
85
82
86
83
// test sources replacement
87
- img0 , err := os .CreateTemp ("" , "test" )
88
- require .NoError (t , err )
89
- defer func () {
90
- _ = img0 .Close ()
91
- _ = os .Remove (img0 .Name ())
92
- }()
93
- info0 , err := img0 .Stat ()
84
+ img0 := filepath .Join (t .TempDir (), "test" )
85
+ require .NoError (t , os .WriteFile (img0 , nil , 0644 ))
86
+ info0 , err := os .Stat (img0 )
94
87
require .NoError (t , err )
95
88
96
- data , err := json .Marshal (map [string ]time.Time {img0 . Name () : info0 .ModTime ()})
89
+ data , err := json .Marshal (map [string ]time.Time {img0 : info0 .ModTime ()})
97
90
require .NoError (t , err )
98
91
image = images.Image {
99
92
Labels : map [string ]string {ImageSourcePathsLabel : string (data )},
100
93
}
101
94
102
- img1 , err := os .CreateTemp ("" , "test" )
103
- require .NoError (t , err )
104
- defer func () {
105
- _ = img1 .Close ()
106
- _ = os .Remove (img1 .Name ())
107
- }()
108
- info1 , err := img1 .Stat ()
95
+ img1 := filepath .Join (t .TempDir (), "test" )
96
+ require .NoError (t , os .WriteFile (img1 , nil , 0644 ))
97
+ info1 , err := os .Stat (img1 )
109
98
require .NoError (t , err )
110
99
111
- newsrc := ImageSources {img1 . Name () : info1 .ModTime ()}
100
+ newsrc := ImageSources {img1 : info1 .ModTime ()}
112
101
err = SetImageSources (& image , newsrc )
113
102
require .NoError (t , err )
114
103
115
- expected = ImageSources {img1 . Name () : info1 .ModTime ()}
104
+ expected = ImageSources {img1 : info1 .ModTime ()}
116
105
got , err = GetImageSources (image )
117
106
require .NoError (t , err )
118
- require .True (t , expected [img1 . Name () ].Equal (got [img1 . Name () ]), "dates mismatch" )
107
+ require .True (t , expected [img1 ].Equal (got [img1 ]), "dates mismatch" )
119
108
}
120
109
121
110
func TestAddToImageSources (t * testing.T ) {
122
111
// test replacing sources
123
- img0 , err := os .CreateTemp ("" , "test" )
124
- require .NoError (t , err )
125
- defer func () {
126
- _ = img0 .Close ()
127
- _ = os .Remove (img0 .Name ())
128
- }()
129
- info0 , err := img0 .Stat ()
112
+ img0 := filepath .Join (t .TempDir (), "test" )
113
+ require .NoError (t , os .WriteFile (img0 , nil , 0644 ))
114
+ info0 , err := os .Stat (img0 )
130
115
require .NoError (t , err )
131
116
132
- data , err := json .Marshal (map [string ]time.Time {img0 . Name () : info0 .ModTime ()})
117
+ data , err := json .Marshal (map [string ]time.Time {img0 : info0 .ModTime ()})
133
118
require .NoError (t , err )
134
119
image := images.Image {
135
120
Labels : map [string ]string {ImageSourcePathsLabel : string (data )},
136
121
}
137
122
138
- img1 , err := os .CreateTemp ("" , "test" )
139
- require .NoError (t , err )
140
- defer func () {
141
- _ = img1 .Close ()
142
- _ = os .Remove (img1 .Name ())
143
- }()
144
- info1 , err := img1 .Stat ()
123
+ img1 := filepath .Join (t .TempDir (), "test" )
124
+ require .NoError (t , os .WriteFile (img1 , nil , 0644 ))
125
+ info1 , err := os .Stat (img0 )
145
126
require .NoError (t , err )
146
127
147
- err = AddToImageSources (& image , img1 . Name () , info1 .ModTime ())
128
+ err = AddToImageSources (& image , img1 , info1 .ModTime ())
148
129
require .NoError (t , err )
149
130
150
131
expected := ImageSources {
151
- img0 . Name () : info0 .ModTime (),
152
- img1 . Name () : info1 .ModTime (),
132
+ img0 : info0 .ModTime (),
133
+ img1 : info1 .ModTime (),
153
134
}
154
135
got , err := GetImageSources (image )
155
136
require .NoError (t , err )
156
- require .True (t , expected [img0 . Name () ].Equal (got [img0 . Name () ]), "dates mismatch" )
157
- require .True (t , expected [img1 . Name () ].Equal (got [img1 . Name () ]), "dates mismatch" )
137
+ require .True (t , expected [img0 ].Equal (got [img0 ]), "dates mismatch" )
138
+ require .True (t , expected [img1 ].Equal (got [img1 ]), "dates mismatch" )
158
139
159
140
// test if it trims the sources
160
- err = img0 .Close ()
161
- require .NoError (t , err )
162
- err = os .Remove (img0 .Name ())
141
+ err = os .Remove (img0 )
163
142
require .NoError (t , err )
164
143
165
- err = AddToImageSources (& image , img1 . Name () , info1 .ModTime ())
144
+ err = AddToImageSources (& image , img1 , info1 .ModTime ())
166
145
require .NoError (t , err )
167
146
168
- expected = ImageSources {img1 . Name () : info1 .ModTime ()}
147
+ expected = ImageSources {img1 : info1 .ModTime ()}
169
148
got , err = GetImageSources (image )
170
149
require .NoError (t , err )
171
- require .True (t , expected [img1 . Name () ].Equal (got [img1 . Name () ]), "dates mismatch" )
150
+ require .True (t , expected [img1 ].Equal (got [img1 ]), "dates mismatch" )
172
151
}
0 commit comments