mirror of
https://github.com/fleetdm/fleet
synced 2026-05-22 16:39:01 +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.AppConfigStore = (*AppConfigStore)(nil)
|
|
|
|
type NewAppConfigFunc func(info *fleet.AppConfig) (*fleet.AppConfig, error)
|
|
|
|
type AppConfigFunc func() (*fleet.AppConfig, error)
|
|
|
|
type SaveAppConfigFunc func(info *fleet.AppConfig) error
|
|
|
|
type VerifyEnrollSecretFunc func(secret string) (*fleet.EnrollSecret, error)
|
|
|
|
type GetEnrollSecretsFunc func(teamID *uint) ([]*fleet.EnrollSecret, error)
|
|
|
|
type ApplyEnrollSecretsFunc func(teamID *uint, secrets []*fleet.EnrollSecret) error
|
|
|
|
type AppConfigStore struct {
|
|
NewAppConfigFunc NewAppConfigFunc
|
|
NewAppConfigFuncInvoked bool
|
|
|
|
AppConfigFunc AppConfigFunc
|
|
AppConfigFuncInvoked bool
|
|
|
|
SaveAppConfigFunc SaveAppConfigFunc
|
|
SaveAppConfigFuncInvoked bool
|
|
|
|
VerifyEnrollSecretFunc VerifyEnrollSecretFunc
|
|
VerifyEnrollSecretFuncInvoked bool
|
|
|
|
GetEnrollSecretsFunc GetEnrollSecretsFunc
|
|
GetEnrollSecretsFuncInvoked bool
|
|
|
|
ApplyEnrollSecretsFunc ApplyEnrollSecretsFunc
|
|
ApplyEnrollSecretsFuncInvoked bool
|
|
}
|
|
|
|
func (s *AppConfigStore) NewAppConfig(info *fleet.AppConfig) (*fleet.AppConfig, error) {
|
|
s.NewAppConfigFuncInvoked = true
|
|
return s.NewAppConfigFunc(info)
|
|
}
|
|
|
|
func (s *AppConfigStore) AppConfig() (*fleet.AppConfig, error) {
|
|
s.AppConfigFuncInvoked = true
|
|
return s.AppConfigFunc()
|
|
}
|
|
|
|
func (s *AppConfigStore) SaveAppConfig(info *fleet.AppConfig) error {
|
|
s.SaveAppConfigFuncInvoked = true
|
|
return s.SaveAppConfigFunc(info)
|
|
}
|
|
|
|
func (s *AppConfigStore) VerifyEnrollSecret(secret string) (*fleet.EnrollSecret, error) {
|
|
s.VerifyEnrollSecretFuncInvoked = true
|
|
return s.VerifyEnrollSecretFunc(secret)
|
|
}
|
|
|
|
func (s *AppConfigStore) GetEnrollSecrets(teamID *uint) ([]*fleet.EnrollSecret, error) {
|
|
s.GetEnrollSecretsFuncInvoked = true
|
|
return s.GetEnrollSecretsFunc(teamID)
|
|
}
|
|
|
|
func (s *AppConfigStore) ApplyEnrollSecrets(teamID *uint, secrets []*fleet.EnrollSecret) error {
|
|
s.ApplyEnrollSecretsFuncInvoked = true
|
|
return s.ApplyEnrollSecretsFunc(teamID, secrets)
|
|
}
|