mirror of
https://github.com/project-zot/zot.git
synced 2024-12-16 21:56:37 -05:00
0a04b2a4ed
Signed-off-by: Laurentiu Niculae <niculae.laurentiu1@gmail.com>
131 lines
3.8 KiB
Go
131 lines
3.8 KiB
Go
package mocks
|
|
|
|
import (
|
|
"zotregistry.io/zot/pkg/common"
|
|
cvemodel "zotregistry.io/zot/pkg/extensions/search/cve/model"
|
|
)
|
|
|
|
type CveInfoMock struct {
|
|
GetImageListForCVEFn func(repo, cveID string) ([]cvemodel.TagInfo, error)
|
|
GetImageListWithCVEFixedFn func(repo, cveID string) ([]cvemodel.TagInfo, error)
|
|
GetCVEListForImageFn func(repo string, reference string, searchedCVE string, pageInput cvemodel.PageInput,
|
|
) ([]cvemodel.CVE, common.PageInfo, error)
|
|
GetCVESummaryForImageFn func(repo string, reference string,
|
|
) (cvemodel.ImageCVESummary, error)
|
|
GetCVESummaryForImageMediaFn func(repo string, digest, mediaType string,
|
|
) (cvemodel.ImageCVESummary, error)
|
|
CompareSeveritiesFn func(severity1, severity2 string) int
|
|
UpdateDBFn func() error
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) GetImageListForCVE(repo, cveID string) ([]cvemodel.TagInfo, error) {
|
|
if cveInfo.GetImageListForCVEFn != nil {
|
|
return cveInfo.GetImageListForCVEFn(repo, cveID)
|
|
}
|
|
|
|
return []cvemodel.TagInfo{}, nil
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) GetImageListWithCVEFixed(repo, cveID string) ([]cvemodel.TagInfo, error) {
|
|
if cveInfo.GetImageListWithCVEFixedFn != nil {
|
|
return cveInfo.GetImageListWithCVEFixedFn(repo, cveID)
|
|
}
|
|
|
|
return []cvemodel.TagInfo{}, nil
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) GetCVEListForImage(repo string, reference string,
|
|
searchedCVE string, pageInput cvemodel.PageInput,
|
|
) (
|
|
[]cvemodel.CVE,
|
|
common.PageInfo,
|
|
error,
|
|
) {
|
|
if cveInfo.GetCVEListForImageFn != nil {
|
|
return cveInfo.GetCVEListForImageFn(repo, reference, searchedCVE, pageInput)
|
|
}
|
|
|
|
return []cvemodel.CVE{}, common.PageInfo{}, nil
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) GetCVESummaryForImage(repo string, reference string,
|
|
) (cvemodel.ImageCVESummary, error) {
|
|
if cveInfo.GetCVESummaryForImageFn != nil {
|
|
return cveInfo.GetCVESummaryForImageFn(repo, reference)
|
|
}
|
|
|
|
return cvemodel.ImageCVESummary{}, nil
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) GetCVESummaryForImageMedia(repo, digest, mediaType string,
|
|
) (cvemodel.ImageCVESummary, error) {
|
|
if cveInfo.GetCVESummaryForImageMediaFn != nil {
|
|
return cveInfo.GetCVESummaryForImageMediaFn(repo, digest, mediaType)
|
|
}
|
|
|
|
return cvemodel.ImageCVESummary{}, nil
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) CompareSeverities(severity1, severity2 string) int {
|
|
if cveInfo.CompareSeveritiesFn != nil {
|
|
return cveInfo.CompareSeveritiesFn(severity1, severity2)
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
func (cveInfo CveInfoMock) UpdateDB() error {
|
|
if cveInfo.UpdateDBFn != nil {
|
|
return cveInfo.UpdateDBFn()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type CveScannerMock struct {
|
|
IsImageFormatScannableFn func(repo string, reference string) (bool, error)
|
|
IsImageMediaScannableFn func(repo string, digest, mediaType string) (bool, error)
|
|
ScanImageFn func(image string) (map[string]cvemodel.CVE, error)
|
|
CompareSeveritiesFn func(severity1, severity2 string) int
|
|
UpdateDBFn func() error
|
|
}
|
|
|
|
func (scanner CveScannerMock) IsImageFormatScannable(repo string, reference string) (bool, error) {
|
|
if scanner.IsImageFormatScannableFn != nil {
|
|
return scanner.IsImageFormatScannableFn(repo, reference)
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
func (scanner CveScannerMock) IsImageMediaScannable(repo string, digest, mediaType string) (bool, error) {
|
|
if scanner.IsImageMediaScannableFn != nil {
|
|
return scanner.IsImageMediaScannableFn(repo, digest, mediaType)
|
|
}
|
|
|
|
return true, nil
|
|
}
|
|
|
|
func (scanner CveScannerMock) ScanImage(image string) (map[string]cvemodel.CVE, error) {
|
|
if scanner.ScanImageFn != nil {
|
|
return scanner.ScanImageFn(image)
|
|
}
|
|
|
|
return map[string]cvemodel.CVE{}, nil
|
|
}
|
|
|
|
func (scanner CveScannerMock) CompareSeverities(severity1, severity2 string) int {
|
|
if scanner.CompareSeveritiesFn != nil {
|
|
return scanner.CompareSeveritiesFn(severity1, severity2)
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
func (scanner CveScannerMock) UpdateDB() error {
|
|
if scanner.UpdateDBFn != nil {
|
|
return scanner.UpdateDBFn()
|
|
}
|
|
|
|
return nil
|
|
}
|