8567ec92b0
* make timestamp field time type * Add test for type on handleLogEvents data frame field response --------- Co-authored-by: Shirley Leu <4163034+fridgepoet@users.noreply.github.com>
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package mocks
|
|
|
|
import (
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/request"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatchlogs"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatchlogs/cloudwatchlogsiface"
|
|
"github.com/grafana/grafana/pkg/tsdb/cloudwatch/models/resources"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type LogsAPI struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (l *LogsAPI) DescribeLogGroups(input *cloudwatchlogs.DescribeLogGroupsInput) (*cloudwatchlogs.DescribeLogGroupsOutput, error) {
|
|
args := l.Called(input)
|
|
|
|
return args.Get(0).(*cloudwatchlogs.DescribeLogGroupsOutput), args.Error(1)
|
|
}
|
|
|
|
func (l *LogsAPI) GetLogGroupFields(input *cloudwatchlogs.GetLogGroupFieldsInput) (*cloudwatchlogs.GetLogGroupFieldsOutput, error) {
|
|
args := l.Called(input)
|
|
|
|
return args.Get(0).(*cloudwatchlogs.GetLogGroupFieldsOutput), args.Error(1)
|
|
}
|
|
|
|
type LogsService struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (l *LogsService) GetLogGroups(request resources.LogGroupsRequest) ([]resources.ResourceResponse[resources.LogGroup], error) {
|
|
args := l.Called(request)
|
|
|
|
return args.Get(0).([]resources.ResourceResponse[resources.LogGroup]), args.Error(1)
|
|
}
|
|
|
|
func (l *LogsService) GetLogGroupFields(request resources.LogGroupFieldsRequest) ([]resources.ResourceResponse[resources.LogGroupField], error) {
|
|
args := l.Called(request)
|
|
|
|
return args.Get(0).([]resources.ResourceResponse[resources.LogGroupField]), args.Error(1)
|
|
}
|
|
|
|
type MockFeatures struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (f *MockFeatures) IsEnabled(feature string) bool {
|
|
args := f.Called(feature)
|
|
|
|
return args.Bool(0)
|
|
}
|
|
|
|
type MockLogEvents struct {
|
|
cloudwatchlogsiface.CloudWatchLogsAPI
|
|
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *MockLogEvents) GetLogEventsWithContext(ctx aws.Context, input *cloudwatchlogs.GetLogEventsInput, option ...request.Option) (*cloudwatchlogs.GetLogEventsOutput, error) {
|
|
args := m.Called(ctx, input, option)
|
|
|
|
return args.Get(0).(*cloudwatchlogs.GetLogEventsOutput), args.Error(1)
|
|
}
|