@@ -45,6 +45,11 @@ func (m *MockSyftPackagesExtractor) AnalyzeImages(images []types.ImageModel) ([]
45
45
return args .Get (0 ).([]* syftPackagesExtractor.ContainerResolution ), args .Error (1 )
46
46
}
47
47
48
+ func (m * MockSyftPackagesExtractor ) AnalyzeImagesWithPlatform (images []types.ImageModel , platform string ) ([]* syftPackagesExtractor.ContainerResolution , error ) {
49
+ args := m .Called (images , platform )
50
+ return args .Get (0 ).([]* syftPackagesExtractor.ContainerResolution ), args .Error (1 )
51
+ }
52
+
48
53
func createTestFolder (dir string ) {
49
54
// Create the directory if it doesn't exist
50
55
if _ , err := os .Stat (dir ); os .IsNotExist (err ) {
@@ -130,15 +135,15 @@ func TestResolve(t *testing.T) {
130
135
types .ToImageModels (images ),
131
136
map [string ]map [string ]string {"settings.json" : {"key" : "value" }}).
132
137
Return ([]types.ImageModel {{Name : "image1" }}, nil )
133
- mockSyftPackagesExtractor .On ("AnalyzeImages" , mock .Anything ).Return (expectedResolution , nil )
138
+ mockSyftPackagesExtractor .On ("AnalyzeImagesWithPlatform" , mock . Anything , mock .Anything ).Return (expectedResolution , nil )
134
139
mockImagesExtractor .On ("SaveObjectToFile" , checkmarxPath , expectedResolution ).Return (nil )
135
140
136
141
err := resolver .Resolve (scanPath , resolutionFolderPath , images , true )
137
142
assert .NoError (t , err )
138
143
139
144
mockImagesExtractor .AssertCalled (t , "ExtractFiles" , scanPath )
140
145
mockImagesExtractor .AssertCalled (t , "ExtractAndMergeImagesFromFiles" , sampleFileImages , mock .Anything , mock .Anything )
141
- mockSyftPackagesExtractor .AssertCalled (t , "AnalyzeImages " , mock .Anything )
146
+ mockSyftPackagesExtractor .AssertCalled (t , "AnalyzeImagesWithPlatform " , mock .Anything , "linux/amd64" )
142
147
mockImagesExtractor .AssertCalled (t , "SaveObjectToFile" , checkmarxPath , expectedResolution )
143
148
})
144
149
@@ -184,7 +189,7 @@ func TestResolve(t *testing.T) {
184
189
map [string ]map [string ]string {"settings.json" : {"key" : "value" }}).
185
190
Return ([]types.ImageModel {{Name : "image1" }}, nil )
186
191
187
- mockSyftPackagesExtractor .On ("AnalyzeImages " , mock .Anything ).Return (expectedResolution , errors .New ("error analyzing images" ))
192
+ mockSyftPackagesExtractor .On ("AnalyzeImagesWithPlatform " , mock .Anything , "linux/amd64" ).Return (expectedResolution , errors .New ("error analyzing images" ))
188
193
189
194
err := resolver .Resolve (scanPath , resolutionFolderPath , images , false )
190
195
assert .Error (t , err )
0 commit comments