44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
package mocks
|
|
|
|
import (
|
|
"github.com/fiskerinc/cloud-services/pkg/common"
|
|
"github.com/fiskerinc/cloud-services/pkg/db/queries"
|
|
"github.com/go-pg/pg/v10/orm"
|
|
)
|
|
|
|
// EccKey query methods
|
|
type MockSupplierOrganization struct {
|
|
DBMockHelper
|
|
MockListResponse []common.SupplierOrganization
|
|
MockSupplierOrganization *common.SupplierOrganization
|
|
}
|
|
|
|
func (so *MockSupplierOrganization) Count(supplierOrganization *common.SupplierOrganization) (int, error) {
|
|
return len(so.MockListResponse), so.Error
|
|
}
|
|
|
|
func (so *MockSupplierOrganization) Insert(supplierOrg *common.SupplierOrganization) (orm.Result, error) {
|
|
return so.ORMResponse, so.Error
|
|
}
|
|
|
|
func (so *MockSupplierOrganization) Update(supplierOrg *common.SupplierOrganization) (orm.Result, error) {
|
|
return so.ORMResponse, so.Error
|
|
}
|
|
|
|
func (so *MockSupplierOrganization) Delete(supplierOrg *common.SupplierOrganization) (orm.Result, error) {
|
|
return so.ORMResponse, so.Error
|
|
}
|
|
|
|
func (so *MockSupplierOrganization) Select(supplierOrg *common.SupplierOrganization, paging *queries.PageQueryOptions) ([]common.SupplierOrganization, error) {
|
|
so.LastFilter = supplierOrg
|
|
so.LastPaging = paging
|
|
|
|
return so.MockListResponse, so.Error
|
|
}
|
|
|
|
func (c *MockSupplierOrganization) SetListResp(list interface{}) {
|
|
if list != nil {
|
|
c.MockListResponse = list.([]common.SupplierOrganization)
|
|
}
|
|
}
|