49 lines
955 B
Go
49 lines
955 B
Go
package mocks
|
|
|
|
import (
|
|
"fiskerinc.com/modules/common"
|
|
"fiskerinc.com/modules/validator"
|
|
|
|
"github.com/go-pg/pg/v10/orm"
|
|
)
|
|
|
|
type MockFileKeys struct {
|
|
GetResponse *common.FileKey
|
|
GetMultiResponse []common.FileKey
|
|
DBMockHelper
|
|
}
|
|
|
|
func (fk *MockFileKeys) Delete(fileID string) (orm.Result, error) {
|
|
if fileID == "" {
|
|
return nil, &validator.FieldError{
|
|
ErrorMsg: "FileID required",
|
|
}
|
|
}
|
|
|
|
return fk.ORMResponse, fk.Error
|
|
}
|
|
|
|
func (fk *MockFileKeys) Insert(filekey common.FileKey) (orm.Result, error) {
|
|
err := validator.ValidateStruct(filekey)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return fk.ORMResponse, fk.Error
|
|
}
|
|
|
|
func (fk *MockFileKeys) Get(fileID string) (*common.FileKey, error) {
|
|
if fileID == "" {
|
|
return nil, &validator.FieldError{
|
|
ErrorMsg: "FileID required",
|
|
}
|
|
}
|
|
|
|
return fk.GetResponse, fk.Error
|
|
}
|
|
|
|
func (fk *MockFileKeys) GetMulti(fileIDs []string) ([]common.FileKey, error) {
|
|
return fk.GetMultiResponse, fk.Error
|
|
}
|