forked from sensu/sensu-go
/
assets.go
32 lines (26 loc) · 841 Bytes
/
assets.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package mockstore
import (
"context"
"github.com/sensu/sensu-go/backend/store"
"github.com/sensu/sensu-go/types"
)
// DeleteAssetByName ...
func (s *MockStore) DeleteAssetByName(ctx context.Context, name string) error {
args := s.Called(ctx, name)
return args.Error(0)
}
// GetAssets ...
func (s *MockStore) GetAssets(ctx context.Context, pred *store.SelectionPredicate) ([]*types.Asset, error) {
args := s.Called(ctx, pred)
return args.Get(0).([]*types.Asset), args.Error(1)
}
// GetAssetByName ...
func (s *MockStore) GetAssetByName(ctx context.Context, name string) (*types.Asset, error) {
args := s.Called(ctx, name)
return args.Get(0).(*types.Asset), args.Error(1)
}
// UpdateAsset ...
func (s *MockStore) UpdateAsset(ctx context.Context, asset *types.Asset) error {
args := s.Called(ctx, asset)
return args.Error(0)
}