mirror of
https://github.com/fleetdm/fleet
synced 2026-05-17 14:08:25 +00:00
f8b7a83cc6
* WIP * WIP * Make path optional and fix tests * Add first generate * Move to nvd package * remove replace * Re-add replace * It's path, not file name * Change how db path is set and use etag * Fix typos * Make db generation faster * Remove quotes * Doesn't like comments * Samitize etag and save to file * Refactor some things and improve writing of etagenv * Compress file and truncate amount of items for faster testing * Remove quotes * Try to improve performance * Ignore truncate error if not exists * Minor cleanup and make sqlite have cpe prefix * Simplify code and test sync * Add VCR for sync test * Check for nvdRelease nil * Add test for the actual translation * Address review comments * Rename generate command because we'll have a cve one too * Move to its own dir * Add first cve db generation * WIP but with final strategy, preparring to merge main * Fix merge conflicts * WIP * wip * Insert CVEs to the db * Remove unused code * Use wg instead of counting * Call cancelFunc to avoid ctx leak * Fix logs for better readability * Point code to fleetdm instead of my repo
69 lines
2 KiB
Go
69 lines
2 KiB
Go
// Automatically generated by mockimpl. DO NOT EDIT!
|
|
|
|
package mock
|
|
|
|
import "github.com/fleetdm/fleet/v4/server/fleet"
|
|
|
|
var _ fleet.SoftwareStore = (*SoftwareStore)(nil)
|
|
|
|
type SaveHostSoftwareFunc func(host *fleet.Host) error
|
|
|
|
type LoadHostSoftwareFunc func(host *fleet.Host) error
|
|
|
|
type AllSoftwareWithoutCPEIteratorFunc func() (fleet.SoftwareIterator, error)
|
|
|
|
type AddCPEForSoftwareFunc func(software fleet.Software, cpe string) error
|
|
|
|
type AllCPEsFunc func() ([]string, error)
|
|
|
|
type InsertCVEForCPEFunc func(cve string, cpes []string) error
|
|
|
|
type SoftwareStore struct {
|
|
SaveHostSoftwareFunc SaveHostSoftwareFunc
|
|
SaveHostSoftwareFuncInvoked bool
|
|
|
|
LoadHostSoftwareFunc LoadHostSoftwareFunc
|
|
LoadHostSoftwareFuncInvoked bool
|
|
|
|
AllSoftwareWithoutCPEIteratorFunc AllSoftwareWithoutCPEIteratorFunc
|
|
AllSoftwareWithoutCPEIteratorFuncInvoked bool
|
|
|
|
AddCPEForSoftwareFunc AddCPEForSoftwareFunc
|
|
AddCPEForSoftwareFuncInvoked bool
|
|
|
|
AllCPEsFunc AllCPEsFunc
|
|
AllCPEsFuncInvoked bool
|
|
|
|
InsertCVEForCPEFunc InsertCVEForCPEFunc
|
|
InsertCVEForCPEFuncInvoked bool
|
|
}
|
|
|
|
func (s *SoftwareStore) SaveHostSoftware(host *fleet.Host) error {
|
|
s.SaveHostSoftwareFuncInvoked = true
|
|
return s.SaveHostSoftwareFunc(host)
|
|
}
|
|
|
|
func (s *SoftwareStore) LoadHostSoftware(host *fleet.Host) error {
|
|
s.LoadHostSoftwareFuncInvoked = true
|
|
return s.LoadHostSoftwareFunc(host)
|
|
}
|
|
|
|
func (s *SoftwareStore) AllSoftwareWithoutCPEIterator() (fleet.SoftwareIterator, error) {
|
|
s.AllSoftwareWithoutCPEIteratorFuncInvoked = true
|
|
return s.AllSoftwareWithoutCPEIteratorFunc()
|
|
}
|
|
|
|
func (s *SoftwareStore) AddCPEForSoftware(software fleet.Software, cpe string) error {
|
|
s.AddCPEForSoftwareFuncInvoked = true
|
|
return s.AddCPEForSoftwareFunc(software, cpe)
|
|
}
|
|
|
|
func (s *SoftwareStore) AllCPEs() ([]string, error) {
|
|
s.AllCPEsFuncInvoked = true
|
|
return s.AllCPEsFunc()
|
|
}
|
|
|
|
func (s *SoftwareStore) InsertCVEForCPE(cve string, cpes []string) error {
|
|
s.InsertCVEForCPEFuncInvoked = true
|
|
return s.InsertCVEForCPEFunc(cve, cpes)
|
|
}
|