mirror of
https://github.com/aquasecurity/trivy.git
synced 2025-12-22 07:10:41 -08:00
44 lines
900 B
Go
44 lines
900 B
Go
package db
|
|
|
|
import (
|
|
bolt "github.com/etcd-io/bbolt"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type MockDBConfig struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (_m *MockDBConfig) SetVersion(version int) error {
|
|
ret := _m.Called(version)
|
|
return ret.Error(0)
|
|
}
|
|
|
|
func (_m *MockDBConfig) Update(a, b, c string, d interface{}) error {
|
|
ret := _m.Called(a, b, c, d)
|
|
return ret.Error(0)
|
|
}
|
|
|
|
func (_m *MockDBConfig) BatchUpdate(f func(*bolt.Tx) error) error {
|
|
ret := _m.Called(f)
|
|
return ret.Error(0)
|
|
}
|
|
|
|
func (_m *MockDBConfig) PutNestedBucket(a *bolt.Tx, b, c, d string, e interface{}) error {
|
|
ret := _m.Called(a, b, c, d, e)
|
|
return ret.Error(0)
|
|
}
|
|
|
|
func (_m *MockDBConfig) ForEach(a string, b string) (map[string][]byte, error) {
|
|
ret := _m.Called(a, b)
|
|
ret0 := ret.Get(0)
|
|
if ret0 == nil {
|
|
return nil, ret.Error(1)
|
|
}
|
|
r, ok := ret0.(map[string][]byte)
|
|
if !ok {
|
|
return nil, ret.Error(1)
|
|
}
|
|
return r, ret.Error(1)
|
|
}
|