diff --git a/.github/CONTRIBUTING.md b/.github/CONTRIBUTING.md index 053a4751e6..691b6ff62c 100644 --- a/.github/CONTRIBUTING.md +++ b/.github/CONTRIBUTING.md @@ -76,8 +76,6 @@ For coding style, you may refer to the [code style](CODE_STYLE.md) document whic Generate the mock code with `go generate -run "mockgen" ./...` from your working directory. This will also update existing mocks. You can update all mocks by running `go generate -run "mockgen" ./...` from the repository root. Note this does not log anything out. - ⚠️ We are still using `mockery` instead of `gomock` for older mocks, so you need to run `go generate -run "mockery|mockgen" ./...` to update all the existing mocks. We are slowly migrating our mocks and test code to use `gomock`. - > To execute `//go:generate` commands that are placed at files with `//go:build integration` remember to add `-tags integration` in the `go generate` command eg. `go generate -tags integration ...` 9. **Lint your changes.** diff --git a/.github/workflows/mocks.yml b/.github/workflows/mocks.yml index be8e92ce2a..3af2ed2a12 100644 --- a/.github/workflows/mocks.yml +++ b/.github/workflows/mocks.yml @@ -21,15 +21,12 @@ jobs: stable: true check-latest: true - - run: go install github.com/vektra/mockery/v2@v2.14 - - run: go install github.com/golang/mock/mockgen@v1.6 - name: Check devnet module run: | cd devnet && \ go mod download && \ - go generate -run "mockery" -tags integration ./... && \ go generate -run "mockgen" -tags integration ./... && \ git diff --exit-code && \ cd .. @@ -37,6 +34,5 @@ jobs: - name: Check Gossamer module run: | go mod download && \ - go generate -run "mockery" -tags integration ./... && \ go generate -run "mockgen" -tags integration ./... && \ git diff --exit-code diff --git a/devnet/cmd/scale-down-ecs-service/internal/interfaces.go b/devnet/cmd/scale-down-ecs-service/internal/interfaces.go new file mode 100644 index 0000000000..a91ddb46c0 --- /dev/null +++ b/devnet/cmd/scale-down-ecs-service/internal/interfaces.go @@ -0,0 +1,20 @@ +// Copyright 2023 ChainSafe Systems (ON) +// SPDX-License-Identifier: LGPL-3.0-only + +package internal + +import ( + "github.com/aws/aws-sdk-go/aws" + "github.com/aws/aws-sdk-go/aws/request" + "github.com/aws/aws-sdk-go/service/ecs" +) + +// ECSAPI is the interface for the ECS API. +type ECSAPI interface { + ListServicesWithContext(aws.Context, *ecs.ListServicesInput, ...request.Option) ( + *ecs.ListServicesOutput, error) + UpdateServiceWithContext(aws.Context, *ecs.UpdateServiceInput, ...request.Option) ( + *ecs.UpdateServiceOutput, error) + DescribeServicesWithContext(aws.Context, *ecs.DescribeServicesInput, ...request.Option) ( + *ecs.DescribeServicesOutput, error) +} diff --git a/devnet/cmd/scale-down-ecs-service/mocks/ecsapi.go b/devnet/cmd/scale-down-ecs-service/mocks/ecsapi.go deleted file mode 100644 index bab021dff0..0000000000 --- a/devnet/cmd/scale-down-ecs-service/mocks/ecsapi.go +++ /dev/null @@ -1,4509 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - context "context" - - ecs "github.com/aws/aws-sdk-go/service/ecs" - - mock "github.com/stretchr/testify/mock" - - request "github.com/aws/aws-sdk-go/aws/request" -) - -// ECSAPI is an autogenerated mock type for the ECSAPI type -type ECSAPI struct { - mock.Mock -} - -// CreateCapacityProvider provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateCapacityProvider(_a0 *ecs.CreateCapacityProviderInput) (*ecs.CreateCapacityProviderOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.CreateCapacityProviderOutput - if rf, ok := ret.Get(0).(func(*ecs.CreateCapacityProviderInput) *ecs.CreateCapacityProviderOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateCapacityProviderOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.CreateCapacityProviderInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateCapacityProviderRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateCapacityProviderRequest(_a0 *ecs.CreateCapacityProviderInput) (*request.Request, *ecs.CreateCapacityProviderOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.CreateCapacityProviderInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.CreateCapacityProviderOutput - if rf, ok := ret.Get(1).(func(*ecs.CreateCapacityProviderInput) *ecs.CreateCapacityProviderOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.CreateCapacityProviderOutput) - } - } - - return r0, r1 -} - -// CreateCapacityProviderWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) CreateCapacityProviderWithContext(_a0 context.Context, _a1 *ecs.CreateCapacityProviderInput, _a2 ...request.Option) (*ecs.CreateCapacityProviderOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.CreateCapacityProviderOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.CreateCapacityProviderInput, ...request.Option) *ecs.CreateCapacityProviderOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateCapacityProviderOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.CreateCapacityProviderInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateCluster provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateCluster(_a0 *ecs.CreateClusterInput) (*ecs.CreateClusterOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.CreateClusterOutput - if rf, ok := ret.Get(0).(func(*ecs.CreateClusterInput) *ecs.CreateClusterOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateClusterOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.CreateClusterInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateClusterRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateClusterRequest(_a0 *ecs.CreateClusterInput) (*request.Request, *ecs.CreateClusterOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.CreateClusterInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.CreateClusterOutput - if rf, ok := ret.Get(1).(func(*ecs.CreateClusterInput) *ecs.CreateClusterOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.CreateClusterOutput) - } - } - - return r0, r1 -} - -// CreateClusterWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) CreateClusterWithContext(_a0 context.Context, _a1 *ecs.CreateClusterInput, _a2 ...request.Option) (*ecs.CreateClusterOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.CreateClusterOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.CreateClusterInput, ...request.Option) *ecs.CreateClusterOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateClusterOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.CreateClusterInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateService provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateService(_a0 *ecs.CreateServiceInput) (*ecs.CreateServiceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.CreateServiceOutput - if rf, ok := ret.Get(0).(func(*ecs.CreateServiceInput) *ecs.CreateServiceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateServiceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.CreateServiceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateServiceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateServiceRequest(_a0 *ecs.CreateServiceInput) (*request.Request, *ecs.CreateServiceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.CreateServiceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.CreateServiceOutput - if rf, ok := ret.Get(1).(func(*ecs.CreateServiceInput) *ecs.CreateServiceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.CreateServiceOutput) - } - } - - return r0, r1 -} - -// CreateServiceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) CreateServiceWithContext(_a0 context.Context, _a1 *ecs.CreateServiceInput, _a2 ...request.Option) (*ecs.CreateServiceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.CreateServiceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.CreateServiceInput, ...request.Option) *ecs.CreateServiceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateServiceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.CreateServiceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateTaskSet provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateTaskSet(_a0 *ecs.CreateTaskSetInput) (*ecs.CreateTaskSetOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.CreateTaskSetOutput - if rf, ok := ret.Get(0).(func(*ecs.CreateTaskSetInput) *ecs.CreateTaskSetOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.CreateTaskSetInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CreateTaskSetRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) CreateTaskSetRequest(_a0 *ecs.CreateTaskSetInput) (*request.Request, *ecs.CreateTaskSetOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.CreateTaskSetInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.CreateTaskSetOutput - if rf, ok := ret.Get(1).(func(*ecs.CreateTaskSetInput) *ecs.CreateTaskSetOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.CreateTaskSetOutput) - } - } - - return r0, r1 -} - -// CreateTaskSetWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) CreateTaskSetWithContext(_a0 context.Context, _a1 *ecs.CreateTaskSetInput, _a2 ...request.Option) (*ecs.CreateTaskSetOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.CreateTaskSetOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.CreateTaskSetInput, ...request.Option) *ecs.CreateTaskSetOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.CreateTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.CreateTaskSetInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteAccountSetting provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteAccountSetting(_a0 *ecs.DeleteAccountSettingInput) (*ecs.DeleteAccountSettingOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeleteAccountSettingOutput - if rf, ok := ret.Get(0).(func(*ecs.DeleteAccountSettingInput) *ecs.DeleteAccountSettingOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteAccountSettingOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeleteAccountSettingInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteAccountSettingRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteAccountSettingRequest(_a0 *ecs.DeleteAccountSettingInput) (*request.Request, *ecs.DeleteAccountSettingOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeleteAccountSettingInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeleteAccountSettingOutput - if rf, ok := ret.Get(1).(func(*ecs.DeleteAccountSettingInput) *ecs.DeleteAccountSettingOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeleteAccountSettingOutput) - } - } - - return r0, r1 -} - -// DeleteAccountSettingWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeleteAccountSettingWithContext(_a0 context.Context, _a1 *ecs.DeleteAccountSettingInput, _a2 ...request.Option) (*ecs.DeleteAccountSettingOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeleteAccountSettingOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeleteAccountSettingInput, ...request.Option) *ecs.DeleteAccountSettingOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteAccountSettingOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeleteAccountSettingInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteAttributes provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteAttributes(_a0 *ecs.DeleteAttributesInput) (*ecs.DeleteAttributesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeleteAttributesOutput - if rf, ok := ret.Get(0).(func(*ecs.DeleteAttributesInput) *ecs.DeleteAttributesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteAttributesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeleteAttributesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteAttributesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteAttributesRequest(_a0 *ecs.DeleteAttributesInput) (*request.Request, *ecs.DeleteAttributesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeleteAttributesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeleteAttributesOutput - if rf, ok := ret.Get(1).(func(*ecs.DeleteAttributesInput) *ecs.DeleteAttributesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeleteAttributesOutput) - } - } - - return r0, r1 -} - -// DeleteAttributesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeleteAttributesWithContext(_a0 context.Context, _a1 *ecs.DeleteAttributesInput, _a2 ...request.Option) (*ecs.DeleteAttributesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeleteAttributesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeleteAttributesInput, ...request.Option) *ecs.DeleteAttributesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteAttributesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeleteAttributesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteCapacityProvider provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteCapacityProvider(_a0 *ecs.DeleteCapacityProviderInput) (*ecs.DeleteCapacityProviderOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeleteCapacityProviderOutput - if rf, ok := ret.Get(0).(func(*ecs.DeleteCapacityProviderInput) *ecs.DeleteCapacityProviderOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteCapacityProviderOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeleteCapacityProviderInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteCapacityProviderRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteCapacityProviderRequest(_a0 *ecs.DeleteCapacityProviderInput) (*request.Request, *ecs.DeleteCapacityProviderOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeleteCapacityProviderInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeleteCapacityProviderOutput - if rf, ok := ret.Get(1).(func(*ecs.DeleteCapacityProviderInput) *ecs.DeleteCapacityProviderOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeleteCapacityProviderOutput) - } - } - - return r0, r1 -} - -// DeleteCapacityProviderWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeleteCapacityProviderWithContext(_a0 context.Context, _a1 *ecs.DeleteCapacityProviderInput, _a2 ...request.Option) (*ecs.DeleteCapacityProviderOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeleteCapacityProviderOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeleteCapacityProviderInput, ...request.Option) *ecs.DeleteCapacityProviderOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteCapacityProviderOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeleteCapacityProviderInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteCluster provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteCluster(_a0 *ecs.DeleteClusterInput) (*ecs.DeleteClusterOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeleteClusterOutput - if rf, ok := ret.Get(0).(func(*ecs.DeleteClusterInput) *ecs.DeleteClusterOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteClusterOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeleteClusterInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteClusterRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteClusterRequest(_a0 *ecs.DeleteClusterInput) (*request.Request, *ecs.DeleteClusterOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeleteClusterInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeleteClusterOutput - if rf, ok := ret.Get(1).(func(*ecs.DeleteClusterInput) *ecs.DeleteClusterOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeleteClusterOutput) - } - } - - return r0, r1 -} - -// DeleteClusterWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeleteClusterWithContext(_a0 context.Context, _a1 *ecs.DeleteClusterInput, _a2 ...request.Option) (*ecs.DeleteClusterOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeleteClusterOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeleteClusterInput, ...request.Option) *ecs.DeleteClusterOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteClusterOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeleteClusterInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteService provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteService(_a0 *ecs.DeleteServiceInput) (*ecs.DeleteServiceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeleteServiceOutput - if rf, ok := ret.Get(0).(func(*ecs.DeleteServiceInput) *ecs.DeleteServiceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteServiceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeleteServiceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteServiceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteServiceRequest(_a0 *ecs.DeleteServiceInput) (*request.Request, *ecs.DeleteServiceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeleteServiceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeleteServiceOutput - if rf, ok := ret.Get(1).(func(*ecs.DeleteServiceInput) *ecs.DeleteServiceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeleteServiceOutput) - } - } - - return r0, r1 -} - -// DeleteServiceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeleteServiceWithContext(_a0 context.Context, _a1 *ecs.DeleteServiceInput, _a2 ...request.Option) (*ecs.DeleteServiceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeleteServiceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeleteServiceInput, ...request.Option) *ecs.DeleteServiceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteServiceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeleteServiceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteTaskSet provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteTaskSet(_a0 *ecs.DeleteTaskSetInput) (*ecs.DeleteTaskSetOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeleteTaskSetOutput - if rf, ok := ret.Get(0).(func(*ecs.DeleteTaskSetInput) *ecs.DeleteTaskSetOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeleteTaskSetInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeleteTaskSetRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeleteTaskSetRequest(_a0 *ecs.DeleteTaskSetInput) (*request.Request, *ecs.DeleteTaskSetOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeleteTaskSetInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeleteTaskSetOutput - if rf, ok := ret.Get(1).(func(*ecs.DeleteTaskSetInput) *ecs.DeleteTaskSetOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeleteTaskSetOutput) - } - } - - return r0, r1 -} - -// DeleteTaskSetWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeleteTaskSetWithContext(_a0 context.Context, _a1 *ecs.DeleteTaskSetInput, _a2 ...request.Option) (*ecs.DeleteTaskSetOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeleteTaskSetOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeleteTaskSetInput, ...request.Option) *ecs.DeleteTaskSetOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeleteTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeleteTaskSetInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeregisterContainerInstance provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeregisterContainerInstance(_a0 *ecs.DeregisterContainerInstanceInput) (*ecs.DeregisterContainerInstanceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeregisterContainerInstanceOutput - if rf, ok := ret.Get(0).(func(*ecs.DeregisterContainerInstanceInput) *ecs.DeregisterContainerInstanceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeregisterContainerInstanceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeregisterContainerInstanceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeregisterContainerInstanceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeregisterContainerInstanceRequest(_a0 *ecs.DeregisterContainerInstanceInput) (*request.Request, *ecs.DeregisterContainerInstanceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeregisterContainerInstanceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeregisterContainerInstanceOutput - if rf, ok := ret.Get(1).(func(*ecs.DeregisterContainerInstanceInput) *ecs.DeregisterContainerInstanceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeregisterContainerInstanceOutput) - } - } - - return r0, r1 -} - -// DeregisterContainerInstanceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeregisterContainerInstanceWithContext(_a0 context.Context, _a1 *ecs.DeregisterContainerInstanceInput, _a2 ...request.Option) (*ecs.DeregisterContainerInstanceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeregisterContainerInstanceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeregisterContainerInstanceInput, ...request.Option) *ecs.DeregisterContainerInstanceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeregisterContainerInstanceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeregisterContainerInstanceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeregisterTaskDefinition provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeregisterTaskDefinition(_a0 *ecs.DeregisterTaskDefinitionInput) (*ecs.DeregisterTaskDefinitionOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DeregisterTaskDefinitionOutput - if rf, ok := ret.Get(0).(func(*ecs.DeregisterTaskDefinitionInput) *ecs.DeregisterTaskDefinitionOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeregisterTaskDefinitionOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DeregisterTaskDefinitionInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DeregisterTaskDefinitionRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DeregisterTaskDefinitionRequest(_a0 *ecs.DeregisterTaskDefinitionInput) (*request.Request, *ecs.DeregisterTaskDefinitionOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DeregisterTaskDefinitionInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DeregisterTaskDefinitionOutput - if rf, ok := ret.Get(1).(func(*ecs.DeregisterTaskDefinitionInput) *ecs.DeregisterTaskDefinitionOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DeregisterTaskDefinitionOutput) - } - } - - return r0, r1 -} - -// DeregisterTaskDefinitionWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DeregisterTaskDefinitionWithContext(_a0 context.Context, _a1 *ecs.DeregisterTaskDefinitionInput, _a2 ...request.Option) (*ecs.DeregisterTaskDefinitionOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DeregisterTaskDefinitionOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DeregisterTaskDefinitionInput, ...request.Option) *ecs.DeregisterTaskDefinitionOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DeregisterTaskDefinitionOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DeregisterTaskDefinitionInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeCapacityProviders provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeCapacityProviders(_a0 *ecs.DescribeCapacityProvidersInput) (*ecs.DescribeCapacityProvidersOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeCapacityProvidersOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeCapacityProvidersInput) *ecs.DescribeCapacityProvidersOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeCapacityProvidersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeCapacityProvidersInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeCapacityProvidersRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeCapacityProvidersRequest(_a0 *ecs.DescribeCapacityProvidersInput) (*request.Request, *ecs.DescribeCapacityProvidersOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeCapacityProvidersInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeCapacityProvidersOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeCapacityProvidersInput) *ecs.DescribeCapacityProvidersOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeCapacityProvidersOutput) - } - } - - return r0, r1 -} - -// DescribeCapacityProvidersWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeCapacityProvidersWithContext(_a0 context.Context, _a1 *ecs.DescribeCapacityProvidersInput, _a2 ...request.Option) (*ecs.DescribeCapacityProvidersOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeCapacityProvidersOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeCapacityProvidersInput, ...request.Option) *ecs.DescribeCapacityProvidersOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeCapacityProvidersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeCapacityProvidersInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeClusters provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeClusters(_a0 *ecs.DescribeClustersInput) (*ecs.DescribeClustersOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeClustersOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeClustersInput) *ecs.DescribeClustersOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeClustersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeClustersInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeClustersRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeClustersRequest(_a0 *ecs.DescribeClustersInput) (*request.Request, *ecs.DescribeClustersOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeClustersInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeClustersOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeClustersInput) *ecs.DescribeClustersOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeClustersOutput) - } - } - - return r0, r1 -} - -// DescribeClustersWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeClustersWithContext(_a0 context.Context, _a1 *ecs.DescribeClustersInput, _a2 ...request.Option) (*ecs.DescribeClustersOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeClustersOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeClustersInput, ...request.Option) *ecs.DescribeClustersOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeClustersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeClustersInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeContainerInstances provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeContainerInstances(_a0 *ecs.DescribeContainerInstancesInput) (*ecs.DescribeContainerInstancesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeContainerInstancesOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeContainerInstancesInput) *ecs.DescribeContainerInstancesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeContainerInstancesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeContainerInstancesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeContainerInstancesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeContainerInstancesRequest(_a0 *ecs.DescribeContainerInstancesInput) (*request.Request, *ecs.DescribeContainerInstancesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeContainerInstancesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeContainerInstancesOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeContainerInstancesInput) *ecs.DescribeContainerInstancesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeContainerInstancesOutput) - } - } - - return r0, r1 -} - -// DescribeContainerInstancesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeContainerInstancesWithContext(_a0 context.Context, _a1 *ecs.DescribeContainerInstancesInput, _a2 ...request.Option) (*ecs.DescribeContainerInstancesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeContainerInstancesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeContainerInstancesInput, ...request.Option) *ecs.DescribeContainerInstancesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeContainerInstancesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeContainerInstancesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeServices provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeServices(_a0 *ecs.DescribeServicesInput) (*ecs.DescribeServicesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeServicesOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeServicesInput) *ecs.DescribeServicesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeServicesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeServicesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeServicesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeServicesRequest(_a0 *ecs.DescribeServicesInput) (*request.Request, *ecs.DescribeServicesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeServicesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeServicesOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeServicesInput) *ecs.DescribeServicesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeServicesOutput) - } - } - - return r0, r1 -} - -// DescribeServicesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeServicesWithContext(_a0 context.Context, _a1 *ecs.DescribeServicesInput, _a2 ...request.Option) (*ecs.DescribeServicesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeServicesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeServicesInput, ...request.Option) *ecs.DescribeServicesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeServicesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeServicesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeTaskDefinition provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeTaskDefinition(_a0 *ecs.DescribeTaskDefinitionInput) (*ecs.DescribeTaskDefinitionOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeTaskDefinitionOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeTaskDefinitionInput) *ecs.DescribeTaskDefinitionOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeTaskDefinitionOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeTaskDefinitionInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeTaskDefinitionRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeTaskDefinitionRequest(_a0 *ecs.DescribeTaskDefinitionInput) (*request.Request, *ecs.DescribeTaskDefinitionOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeTaskDefinitionInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeTaskDefinitionOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeTaskDefinitionInput) *ecs.DescribeTaskDefinitionOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeTaskDefinitionOutput) - } - } - - return r0, r1 -} - -// DescribeTaskDefinitionWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeTaskDefinitionWithContext(_a0 context.Context, _a1 *ecs.DescribeTaskDefinitionInput, _a2 ...request.Option) (*ecs.DescribeTaskDefinitionOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeTaskDefinitionOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeTaskDefinitionInput, ...request.Option) *ecs.DescribeTaskDefinitionOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeTaskDefinitionOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeTaskDefinitionInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeTaskSets provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeTaskSets(_a0 *ecs.DescribeTaskSetsInput) (*ecs.DescribeTaskSetsOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeTaskSetsOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeTaskSetsInput) *ecs.DescribeTaskSetsOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeTaskSetsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeTaskSetsInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeTaskSetsRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeTaskSetsRequest(_a0 *ecs.DescribeTaskSetsInput) (*request.Request, *ecs.DescribeTaskSetsOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeTaskSetsInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeTaskSetsOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeTaskSetsInput) *ecs.DescribeTaskSetsOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeTaskSetsOutput) - } - } - - return r0, r1 -} - -// DescribeTaskSetsWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeTaskSetsWithContext(_a0 context.Context, _a1 *ecs.DescribeTaskSetsInput, _a2 ...request.Option) (*ecs.DescribeTaskSetsOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeTaskSetsOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeTaskSetsInput, ...request.Option) *ecs.DescribeTaskSetsOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeTaskSetsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeTaskSetsInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeTasks provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeTasks(_a0 *ecs.DescribeTasksInput) (*ecs.DescribeTasksOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DescribeTasksOutput - if rf, ok := ret.Get(0).(func(*ecs.DescribeTasksInput) *ecs.DescribeTasksOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeTasksOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DescribeTasksInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DescribeTasksRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DescribeTasksRequest(_a0 *ecs.DescribeTasksInput) (*request.Request, *ecs.DescribeTasksOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DescribeTasksInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DescribeTasksOutput - if rf, ok := ret.Get(1).(func(*ecs.DescribeTasksInput) *ecs.DescribeTasksOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DescribeTasksOutput) - } - } - - return r0, r1 -} - -// DescribeTasksWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DescribeTasksWithContext(_a0 context.Context, _a1 *ecs.DescribeTasksInput, _a2 ...request.Option) (*ecs.DescribeTasksOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DescribeTasksOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeTasksInput, ...request.Option) *ecs.DescribeTasksOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DescribeTasksOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DescribeTasksInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DiscoverPollEndpoint provides a mock function with given fields: _a0 -func (_m *ECSAPI) DiscoverPollEndpoint(_a0 *ecs.DiscoverPollEndpointInput) (*ecs.DiscoverPollEndpointOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.DiscoverPollEndpointOutput - if rf, ok := ret.Get(0).(func(*ecs.DiscoverPollEndpointInput) *ecs.DiscoverPollEndpointOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DiscoverPollEndpointOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.DiscoverPollEndpointInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// DiscoverPollEndpointRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) DiscoverPollEndpointRequest(_a0 *ecs.DiscoverPollEndpointInput) (*request.Request, *ecs.DiscoverPollEndpointOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.DiscoverPollEndpointInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.DiscoverPollEndpointOutput - if rf, ok := ret.Get(1).(func(*ecs.DiscoverPollEndpointInput) *ecs.DiscoverPollEndpointOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.DiscoverPollEndpointOutput) - } - } - - return r0, r1 -} - -// DiscoverPollEndpointWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) DiscoverPollEndpointWithContext(_a0 context.Context, _a1 *ecs.DiscoverPollEndpointInput, _a2 ...request.Option) (*ecs.DiscoverPollEndpointOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.DiscoverPollEndpointOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DiscoverPollEndpointInput, ...request.Option) *ecs.DiscoverPollEndpointOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.DiscoverPollEndpointOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.DiscoverPollEndpointInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ExecuteCommand provides a mock function with given fields: _a0 -func (_m *ECSAPI) ExecuteCommand(_a0 *ecs.ExecuteCommandInput) (*ecs.ExecuteCommandOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ExecuteCommandOutput - if rf, ok := ret.Get(0).(func(*ecs.ExecuteCommandInput) *ecs.ExecuteCommandOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ExecuteCommandOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ExecuteCommandInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ExecuteCommandRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ExecuteCommandRequest(_a0 *ecs.ExecuteCommandInput) (*request.Request, *ecs.ExecuteCommandOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ExecuteCommandInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ExecuteCommandOutput - if rf, ok := ret.Get(1).(func(*ecs.ExecuteCommandInput) *ecs.ExecuteCommandOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ExecuteCommandOutput) - } - } - - return r0, r1 -} - -// ExecuteCommandWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ExecuteCommandWithContext(_a0 context.Context, _a1 *ecs.ExecuteCommandInput, _a2 ...request.Option) (*ecs.ExecuteCommandOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ExecuteCommandOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ExecuteCommandInput, ...request.Option) *ecs.ExecuteCommandOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ExecuteCommandOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ExecuteCommandInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListAccountSettings provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListAccountSettings(_a0 *ecs.ListAccountSettingsInput) (*ecs.ListAccountSettingsOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListAccountSettingsOutput - if rf, ok := ret.Get(0).(func(*ecs.ListAccountSettingsInput) *ecs.ListAccountSettingsOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListAccountSettingsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListAccountSettingsInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListAccountSettingsPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListAccountSettingsPages(_a0 *ecs.ListAccountSettingsInput, _a1 func(*ecs.ListAccountSettingsOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListAccountSettingsInput, func(*ecs.ListAccountSettingsOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListAccountSettingsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListAccountSettingsPagesWithContext(_a0 context.Context, _a1 *ecs.ListAccountSettingsInput, _a2 func(*ecs.ListAccountSettingsOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListAccountSettingsInput, func(*ecs.ListAccountSettingsOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListAccountSettingsRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListAccountSettingsRequest(_a0 *ecs.ListAccountSettingsInput) (*request.Request, *ecs.ListAccountSettingsOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListAccountSettingsInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListAccountSettingsOutput - if rf, ok := ret.Get(1).(func(*ecs.ListAccountSettingsInput) *ecs.ListAccountSettingsOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListAccountSettingsOutput) - } - } - - return r0, r1 -} - -// ListAccountSettingsWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListAccountSettingsWithContext(_a0 context.Context, _a1 *ecs.ListAccountSettingsInput, _a2 ...request.Option) (*ecs.ListAccountSettingsOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListAccountSettingsOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListAccountSettingsInput, ...request.Option) *ecs.ListAccountSettingsOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListAccountSettingsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListAccountSettingsInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListAttributes provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListAttributes(_a0 *ecs.ListAttributesInput) (*ecs.ListAttributesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListAttributesOutput - if rf, ok := ret.Get(0).(func(*ecs.ListAttributesInput) *ecs.ListAttributesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListAttributesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListAttributesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListAttributesPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListAttributesPages(_a0 *ecs.ListAttributesInput, _a1 func(*ecs.ListAttributesOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListAttributesInput, func(*ecs.ListAttributesOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListAttributesPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListAttributesPagesWithContext(_a0 context.Context, _a1 *ecs.ListAttributesInput, _a2 func(*ecs.ListAttributesOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListAttributesInput, func(*ecs.ListAttributesOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListAttributesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListAttributesRequest(_a0 *ecs.ListAttributesInput) (*request.Request, *ecs.ListAttributesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListAttributesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListAttributesOutput - if rf, ok := ret.Get(1).(func(*ecs.ListAttributesInput) *ecs.ListAttributesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListAttributesOutput) - } - } - - return r0, r1 -} - -// ListAttributesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListAttributesWithContext(_a0 context.Context, _a1 *ecs.ListAttributesInput, _a2 ...request.Option) (*ecs.ListAttributesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListAttributesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListAttributesInput, ...request.Option) *ecs.ListAttributesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListAttributesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListAttributesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListClusters provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListClusters(_a0 *ecs.ListClustersInput) (*ecs.ListClustersOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListClustersOutput - if rf, ok := ret.Get(0).(func(*ecs.ListClustersInput) *ecs.ListClustersOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListClustersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListClustersInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListClustersPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListClustersPages(_a0 *ecs.ListClustersInput, _a1 func(*ecs.ListClustersOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListClustersInput, func(*ecs.ListClustersOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListClustersPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListClustersPagesWithContext(_a0 context.Context, _a1 *ecs.ListClustersInput, _a2 func(*ecs.ListClustersOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListClustersInput, func(*ecs.ListClustersOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListClustersRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListClustersRequest(_a0 *ecs.ListClustersInput) (*request.Request, *ecs.ListClustersOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListClustersInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListClustersOutput - if rf, ok := ret.Get(1).(func(*ecs.ListClustersInput) *ecs.ListClustersOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListClustersOutput) - } - } - - return r0, r1 -} - -// ListClustersWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListClustersWithContext(_a0 context.Context, _a1 *ecs.ListClustersInput, _a2 ...request.Option) (*ecs.ListClustersOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListClustersOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListClustersInput, ...request.Option) *ecs.ListClustersOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListClustersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListClustersInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListContainerInstances provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListContainerInstances(_a0 *ecs.ListContainerInstancesInput) (*ecs.ListContainerInstancesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListContainerInstancesOutput - if rf, ok := ret.Get(0).(func(*ecs.ListContainerInstancesInput) *ecs.ListContainerInstancesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListContainerInstancesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListContainerInstancesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListContainerInstancesPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListContainerInstancesPages(_a0 *ecs.ListContainerInstancesInput, _a1 func(*ecs.ListContainerInstancesOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListContainerInstancesInput, func(*ecs.ListContainerInstancesOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListContainerInstancesPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListContainerInstancesPagesWithContext(_a0 context.Context, _a1 *ecs.ListContainerInstancesInput, _a2 func(*ecs.ListContainerInstancesOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListContainerInstancesInput, func(*ecs.ListContainerInstancesOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListContainerInstancesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListContainerInstancesRequest(_a0 *ecs.ListContainerInstancesInput) (*request.Request, *ecs.ListContainerInstancesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListContainerInstancesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListContainerInstancesOutput - if rf, ok := ret.Get(1).(func(*ecs.ListContainerInstancesInput) *ecs.ListContainerInstancesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListContainerInstancesOutput) - } - } - - return r0, r1 -} - -// ListContainerInstancesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListContainerInstancesWithContext(_a0 context.Context, _a1 *ecs.ListContainerInstancesInput, _a2 ...request.Option) (*ecs.ListContainerInstancesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListContainerInstancesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListContainerInstancesInput, ...request.Option) *ecs.ListContainerInstancesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListContainerInstancesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListContainerInstancesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListServices provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListServices(_a0 *ecs.ListServicesInput) (*ecs.ListServicesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListServicesOutput - if rf, ok := ret.Get(0).(func(*ecs.ListServicesInput) *ecs.ListServicesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListServicesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListServicesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListServicesPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListServicesPages(_a0 *ecs.ListServicesInput, _a1 func(*ecs.ListServicesOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListServicesInput, func(*ecs.ListServicesOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListServicesPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListServicesPagesWithContext(_a0 context.Context, _a1 *ecs.ListServicesInput, _a2 func(*ecs.ListServicesOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListServicesInput, func(*ecs.ListServicesOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListServicesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListServicesRequest(_a0 *ecs.ListServicesInput) (*request.Request, *ecs.ListServicesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListServicesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListServicesOutput - if rf, ok := ret.Get(1).(func(*ecs.ListServicesInput) *ecs.ListServicesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListServicesOutput) - } - } - - return r0, r1 -} - -// ListServicesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListServicesWithContext(_a0 context.Context, _a1 *ecs.ListServicesInput, _a2 ...request.Option) (*ecs.ListServicesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListServicesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListServicesInput, ...request.Option) *ecs.ListServicesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListServicesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListServicesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTagsForResource provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTagsForResource(_a0 *ecs.ListTagsForResourceInput) (*ecs.ListTagsForResourceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListTagsForResourceOutput - if rf, ok := ret.Get(0).(func(*ecs.ListTagsForResourceInput) *ecs.ListTagsForResourceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTagsForResourceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListTagsForResourceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTagsForResourceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTagsForResourceRequest(_a0 *ecs.ListTagsForResourceInput) (*request.Request, *ecs.ListTagsForResourceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListTagsForResourceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListTagsForResourceOutput - if rf, ok := ret.Get(1).(func(*ecs.ListTagsForResourceInput) *ecs.ListTagsForResourceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListTagsForResourceOutput) - } - } - - return r0, r1 -} - -// ListTagsForResourceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListTagsForResourceWithContext(_a0 context.Context, _a1 *ecs.ListTagsForResourceInput, _a2 ...request.Option) (*ecs.ListTagsForResourceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListTagsForResourceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTagsForResourceInput, ...request.Option) *ecs.ListTagsForResourceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTagsForResourceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListTagsForResourceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTaskDefinitionFamilies provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTaskDefinitionFamilies(_a0 *ecs.ListTaskDefinitionFamiliesInput) (*ecs.ListTaskDefinitionFamiliesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListTaskDefinitionFamiliesOutput - if rf, ok := ret.Get(0).(func(*ecs.ListTaskDefinitionFamiliesInput) *ecs.ListTaskDefinitionFamiliesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTaskDefinitionFamiliesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListTaskDefinitionFamiliesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTaskDefinitionFamiliesPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListTaskDefinitionFamiliesPages(_a0 *ecs.ListTaskDefinitionFamiliesInput, _a1 func(*ecs.ListTaskDefinitionFamiliesOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListTaskDefinitionFamiliesInput, func(*ecs.ListTaskDefinitionFamiliesOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListTaskDefinitionFamiliesPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListTaskDefinitionFamiliesPagesWithContext(_a0 context.Context, _a1 *ecs.ListTaskDefinitionFamiliesInput, _a2 func(*ecs.ListTaskDefinitionFamiliesOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTaskDefinitionFamiliesInput, func(*ecs.ListTaskDefinitionFamiliesOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListTaskDefinitionFamiliesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTaskDefinitionFamiliesRequest(_a0 *ecs.ListTaskDefinitionFamiliesInput) (*request.Request, *ecs.ListTaskDefinitionFamiliesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListTaskDefinitionFamiliesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListTaskDefinitionFamiliesOutput - if rf, ok := ret.Get(1).(func(*ecs.ListTaskDefinitionFamiliesInput) *ecs.ListTaskDefinitionFamiliesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListTaskDefinitionFamiliesOutput) - } - } - - return r0, r1 -} - -// ListTaskDefinitionFamiliesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListTaskDefinitionFamiliesWithContext(_a0 context.Context, _a1 *ecs.ListTaskDefinitionFamiliesInput, _a2 ...request.Option) (*ecs.ListTaskDefinitionFamiliesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListTaskDefinitionFamiliesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTaskDefinitionFamiliesInput, ...request.Option) *ecs.ListTaskDefinitionFamiliesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTaskDefinitionFamiliesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListTaskDefinitionFamiliesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTaskDefinitions provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTaskDefinitions(_a0 *ecs.ListTaskDefinitionsInput) (*ecs.ListTaskDefinitionsOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListTaskDefinitionsOutput - if rf, ok := ret.Get(0).(func(*ecs.ListTaskDefinitionsInput) *ecs.ListTaskDefinitionsOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTaskDefinitionsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListTaskDefinitionsInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTaskDefinitionsPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListTaskDefinitionsPages(_a0 *ecs.ListTaskDefinitionsInput, _a1 func(*ecs.ListTaskDefinitionsOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListTaskDefinitionsInput, func(*ecs.ListTaskDefinitionsOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListTaskDefinitionsPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListTaskDefinitionsPagesWithContext(_a0 context.Context, _a1 *ecs.ListTaskDefinitionsInput, _a2 func(*ecs.ListTaskDefinitionsOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTaskDefinitionsInput, func(*ecs.ListTaskDefinitionsOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListTaskDefinitionsRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTaskDefinitionsRequest(_a0 *ecs.ListTaskDefinitionsInput) (*request.Request, *ecs.ListTaskDefinitionsOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListTaskDefinitionsInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListTaskDefinitionsOutput - if rf, ok := ret.Get(1).(func(*ecs.ListTaskDefinitionsInput) *ecs.ListTaskDefinitionsOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListTaskDefinitionsOutput) - } - } - - return r0, r1 -} - -// ListTaskDefinitionsWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListTaskDefinitionsWithContext(_a0 context.Context, _a1 *ecs.ListTaskDefinitionsInput, _a2 ...request.Option) (*ecs.ListTaskDefinitionsOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListTaskDefinitionsOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTaskDefinitionsInput, ...request.Option) *ecs.ListTaskDefinitionsOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTaskDefinitionsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListTaskDefinitionsInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTasks provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTasks(_a0 *ecs.ListTasksInput) (*ecs.ListTasksOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.ListTasksOutput - if rf, ok := ret.Get(0).(func(*ecs.ListTasksInput) *ecs.ListTasksOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTasksOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.ListTasksInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ListTasksPages provides a mock function with given fields: _a0, _a1 -func (_m *ECSAPI) ListTasksPages(_a0 *ecs.ListTasksInput, _a1 func(*ecs.ListTasksOutput, bool) bool) error { - ret := _m.Called(_a0, _a1) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.ListTasksInput, func(*ecs.ListTasksOutput, bool) bool) error); ok { - r0 = rf(_a0, _a1) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListTasksPagesWithContext provides a mock function with given fields: _a0, _a1, _a2, _a3 -func (_m *ECSAPI) ListTasksPagesWithContext(_a0 context.Context, _a1 *ecs.ListTasksInput, _a2 func(*ecs.ListTasksOutput, bool) bool, _a3 ...request.Option) error { - _va := make([]interface{}, len(_a3)) - for _i := range _a3 { - _va[_i] = _a3[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1, _a2) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTasksInput, func(*ecs.ListTasksOutput, bool) bool, ...request.Option) error); ok { - r0 = rf(_a0, _a1, _a2, _a3...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ListTasksRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) ListTasksRequest(_a0 *ecs.ListTasksInput) (*request.Request, *ecs.ListTasksOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.ListTasksInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.ListTasksOutput - if rf, ok := ret.Get(1).(func(*ecs.ListTasksInput) *ecs.ListTasksOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.ListTasksOutput) - } - } - - return r0, r1 -} - -// ListTasksWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) ListTasksWithContext(_a0 context.Context, _a1 *ecs.ListTasksInput, _a2 ...request.Option) (*ecs.ListTasksOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.ListTasksOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.ListTasksInput, ...request.Option) *ecs.ListTasksOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.ListTasksOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.ListTasksInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutAccountSetting provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutAccountSetting(_a0 *ecs.PutAccountSettingInput) (*ecs.PutAccountSettingOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.PutAccountSettingOutput - if rf, ok := ret.Get(0).(func(*ecs.PutAccountSettingInput) *ecs.PutAccountSettingOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutAccountSettingOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.PutAccountSettingInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutAccountSettingDefault provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutAccountSettingDefault(_a0 *ecs.PutAccountSettingDefaultInput) (*ecs.PutAccountSettingDefaultOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.PutAccountSettingDefaultOutput - if rf, ok := ret.Get(0).(func(*ecs.PutAccountSettingDefaultInput) *ecs.PutAccountSettingDefaultOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutAccountSettingDefaultOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.PutAccountSettingDefaultInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutAccountSettingDefaultRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutAccountSettingDefaultRequest(_a0 *ecs.PutAccountSettingDefaultInput) (*request.Request, *ecs.PutAccountSettingDefaultOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.PutAccountSettingDefaultInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.PutAccountSettingDefaultOutput - if rf, ok := ret.Get(1).(func(*ecs.PutAccountSettingDefaultInput) *ecs.PutAccountSettingDefaultOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.PutAccountSettingDefaultOutput) - } - } - - return r0, r1 -} - -// PutAccountSettingDefaultWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) PutAccountSettingDefaultWithContext(_a0 context.Context, _a1 *ecs.PutAccountSettingDefaultInput, _a2 ...request.Option) (*ecs.PutAccountSettingDefaultOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.PutAccountSettingDefaultOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.PutAccountSettingDefaultInput, ...request.Option) *ecs.PutAccountSettingDefaultOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutAccountSettingDefaultOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.PutAccountSettingDefaultInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutAccountSettingRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutAccountSettingRequest(_a0 *ecs.PutAccountSettingInput) (*request.Request, *ecs.PutAccountSettingOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.PutAccountSettingInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.PutAccountSettingOutput - if rf, ok := ret.Get(1).(func(*ecs.PutAccountSettingInput) *ecs.PutAccountSettingOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.PutAccountSettingOutput) - } - } - - return r0, r1 -} - -// PutAccountSettingWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) PutAccountSettingWithContext(_a0 context.Context, _a1 *ecs.PutAccountSettingInput, _a2 ...request.Option) (*ecs.PutAccountSettingOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.PutAccountSettingOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.PutAccountSettingInput, ...request.Option) *ecs.PutAccountSettingOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutAccountSettingOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.PutAccountSettingInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutAttributes provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutAttributes(_a0 *ecs.PutAttributesInput) (*ecs.PutAttributesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.PutAttributesOutput - if rf, ok := ret.Get(0).(func(*ecs.PutAttributesInput) *ecs.PutAttributesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutAttributesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.PutAttributesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutAttributesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutAttributesRequest(_a0 *ecs.PutAttributesInput) (*request.Request, *ecs.PutAttributesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.PutAttributesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.PutAttributesOutput - if rf, ok := ret.Get(1).(func(*ecs.PutAttributesInput) *ecs.PutAttributesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.PutAttributesOutput) - } - } - - return r0, r1 -} - -// PutAttributesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) PutAttributesWithContext(_a0 context.Context, _a1 *ecs.PutAttributesInput, _a2 ...request.Option) (*ecs.PutAttributesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.PutAttributesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.PutAttributesInput, ...request.Option) *ecs.PutAttributesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutAttributesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.PutAttributesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutClusterCapacityProviders provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutClusterCapacityProviders(_a0 *ecs.PutClusterCapacityProvidersInput) (*ecs.PutClusterCapacityProvidersOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.PutClusterCapacityProvidersOutput - if rf, ok := ret.Get(0).(func(*ecs.PutClusterCapacityProvidersInput) *ecs.PutClusterCapacityProvidersOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutClusterCapacityProvidersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.PutClusterCapacityProvidersInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// PutClusterCapacityProvidersRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) PutClusterCapacityProvidersRequest(_a0 *ecs.PutClusterCapacityProvidersInput) (*request.Request, *ecs.PutClusterCapacityProvidersOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.PutClusterCapacityProvidersInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.PutClusterCapacityProvidersOutput - if rf, ok := ret.Get(1).(func(*ecs.PutClusterCapacityProvidersInput) *ecs.PutClusterCapacityProvidersOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.PutClusterCapacityProvidersOutput) - } - } - - return r0, r1 -} - -// PutClusterCapacityProvidersWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) PutClusterCapacityProvidersWithContext(_a0 context.Context, _a1 *ecs.PutClusterCapacityProvidersInput, _a2 ...request.Option) (*ecs.PutClusterCapacityProvidersOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.PutClusterCapacityProvidersOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.PutClusterCapacityProvidersInput, ...request.Option) *ecs.PutClusterCapacityProvidersOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.PutClusterCapacityProvidersOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.PutClusterCapacityProvidersInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RegisterContainerInstance provides a mock function with given fields: _a0 -func (_m *ECSAPI) RegisterContainerInstance(_a0 *ecs.RegisterContainerInstanceInput) (*ecs.RegisterContainerInstanceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.RegisterContainerInstanceOutput - if rf, ok := ret.Get(0).(func(*ecs.RegisterContainerInstanceInput) *ecs.RegisterContainerInstanceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.RegisterContainerInstanceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.RegisterContainerInstanceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RegisterContainerInstanceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) RegisterContainerInstanceRequest(_a0 *ecs.RegisterContainerInstanceInput) (*request.Request, *ecs.RegisterContainerInstanceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.RegisterContainerInstanceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.RegisterContainerInstanceOutput - if rf, ok := ret.Get(1).(func(*ecs.RegisterContainerInstanceInput) *ecs.RegisterContainerInstanceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.RegisterContainerInstanceOutput) - } - } - - return r0, r1 -} - -// RegisterContainerInstanceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) RegisterContainerInstanceWithContext(_a0 context.Context, _a1 *ecs.RegisterContainerInstanceInput, _a2 ...request.Option) (*ecs.RegisterContainerInstanceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.RegisterContainerInstanceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.RegisterContainerInstanceInput, ...request.Option) *ecs.RegisterContainerInstanceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.RegisterContainerInstanceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.RegisterContainerInstanceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RegisterTaskDefinition provides a mock function with given fields: _a0 -func (_m *ECSAPI) RegisterTaskDefinition(_a0 *ecs.RegisterTaskDefinitionInput) (*ecs.RegisterTaskDefinitionOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.RegisterTaskDefinitionOutput - if rf, ok := ret.Get(0).(func(*ecs.RegisterTaskDefinitionInput) *ecs.RegisterTaskDefinitionOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.RegisterTaskDefinitionOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.RegisterTaskDefinitionInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RegisterTaskDefinitionRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) RegisterTaskDefinitionRequest(_a0 *ecs.RegisterTaskDefinitionInput) (*request.Request, *ecs.RegisterTaskDefinitionOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.RegisterTaskDefinitionInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.RegisterTaskDefinitionOutput - if rf, ok := ret.Get(1).(func(*ecs.RegisterTaskDefinitionInput) *ecs.RegisterTaskDefinitionOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.RegisterTaskDefinitionOutput) - } - } - - return r0, r1 -} - -// RegisterTaskDefinitionWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) RegisterTaskDefinitionWithContext(_a0 context.Context, _a1 *ecs.RegisterTaskDefinitionInput, _a2 ...request.Option) (*ecs.RegisterTaskDefinitionOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.RegisterTaskDefinitionOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.RegisterTaskDefinitionInput, ...request.Option) *ecs.RegisterTaskDefinitionOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.RegisterTaskDefinitionOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.RegisterTaskDefinitionInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RunTask provides a mock function with given fields: _a0 -func (_m *ECSAPI) RunTask(_a0 *ecs.RunTaskInput) (*ecs.RunTaskOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.RunTaskOutput - if rf, ok := ret.Get(0).(func(*ecs.RunTaskInput) *ecs.RunTaskOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.RunTaskOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.RunTaskInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RunTaskRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) RunTaskRequest(_a0 *ecs.RunTaskInput) (*request.Request, *ecs.RunTaskOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.RunTaskInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.RunTaskOutput - if rf, ok := ret.Get(1).(func(*ecs.RunTaskInput) *ecs.RunTaskOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.RunTaskOutput) - } - } - - return r0, r1 -} - -// RunTaskWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) RunTaskWithContext(_a0 context.Context, _a1 *ecs.RunTaskInput, _a2 ...request.Option) (*ecs.RunTaskOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.RunTaskOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.RunTaskInput, ...request.Option) *ecs.RunTaskOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.RunTaskOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.RunTaskInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// StartTask provides a mock function with given fields: _a0 -func (_m *ECSAPI) StartTask(_a0 *ecs.StartTaskInput) (*ecs.StartTaskOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.StartTaskOutput - if rf, ok := ret.Get(0).(func(*ecs.StartTaskInput) *ecs.StartTaskOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.StartTaskOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.StartTaskInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// StartTaskRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) StartTaskRequest(_a0 *ecs.StartTaskInput) (*request.Request, *ecs.StartTaskOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.StartTaskInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.StartTaskOutput - if rf, ok := ret.Get(1).(func(*ecs.StartTaskInput) *ecs.StartTaskOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.StartTaskOutput) - } - } - - return r0, r1 -} - -// StartTaskWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) StartTaskWithContext(_a0 context.Context, _a1 *ecs.StartTaskInput, _a2 ...request.Option) (*ecs.StartTaskOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.StartTaskOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.StartTaskInput, ...request.Option) *ecs.StartTaskOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.StartTaskOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.StartTaskInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// StopTask provides a mock function with given fields: _a0 -func (_m *ECSAPI) StopTask(_a0 *ecs.StopTaskInput) (*ecs.StopTaskOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.StopTaskOutput - if rf, ok := ret.Get(0).(func(*ecs.StopTaskInput) *ecs.StopTaskOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.StopTaskOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.StopTaskInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// StopTaskRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) StopTaskRequest(_a0 *ecs.StopTaskInput) (*request.Request, *ecs.StopTaskOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.StopTaskInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.StopTaskOutput - if rf, ok := ret.Get(1).(func(*ecs.StopTaskInput) *ecs.StopTaskOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.StopTaskOutput) - } - } - - return r0, r1 -} - -// StopTaskWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) StopTaskWithContext(_a0 context.Context, _a1 *ecs.StopTaskInput, _a2 ...request.Option) (*ecs.StopTaskOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.StopTaskOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.StopTaskInput, ...request.Option) *ecs.StopTaskOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.StopTaskOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.StopTaskInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SubmitAttachmentStateChanges provides a mock function with given fields: _a0 -func (_m *ECSAPI) SubmitAttachmentStateChanges(_a0 *ecs.SubmitAttachmentStateChangesInput) (*ecs.SubmitAttachmentStateChangesOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.SubmitAttachmentStateChangesOutput - if rf, ok := ret.Get(0).(func(*ecs.SubmitAttachmentStateChangesInput) *ecs.SubmitAttachmentStateChangesOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.SubmitAttachmentStateChangesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.SubmitAttachmentStateChangesInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SubmitAttachmentStateChangesRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) SubmitAttachmentStateChangesRequest(_a0 *ecs.SubmitAttachmentStateChangesInput) (*request.Request, *ecs.SubmitAttachmentStateChangesOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.SubmitAttachmentStateChangesInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.SubmitAttachmentStateChangesOutput - if rf, ok := ret.Get(1).(func(*ecs.SubmitAttachmentStateChangesInput) *ecs.SubmitAttachmentStateChangesOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.SubmitAttachmentStateChangesOutput) - } - } - - return r0, r1 -} - -// SubmitAttachmentStateChangesWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) SubmitAttachmentStateChangesWithContext(_a0 context.Context, _a1 *ecs.SubmitAttachmentStateChangesInput, _a2 ...request.Option) (*ecs.SubmitAttachmentStateChangesOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.SubmitAttachmentStateChangesOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.SubmitAttachmentStateChangesInput, ...request.Option) *ecs.SubmitAttachmentStateChangesOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.SubmitAttachmentStateChangesOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.SubmitAttachmentStateChangesInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SubmitContainerStateChange provides a mock function with given fields: _a0 -func (_m *ECSAPI) SubmitContainerStateChange(_a0 *ecs.SubmitContainerStateChangeInput) (*ecs.SubmitContainerStateChangeOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.SubmitContainerStateChangeOutput - if rf, ok := ret.Get(0).(func(*ecs.SubmitContainerStateChangeInput) *ecs.SubmitContainerStateChangeOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.SubmitContainerStateChangeOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.SubmitContainerStateChangeInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SubmitContainerStateChangeRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) SubmitContainerStateChangeRequest(_a0 *ecs.SubmitContainerStateChangeInput) (*request.Request, *ecs.SubmitContainerStateChangeOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.SubmitContainerStateChangeInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.SubmitContainerStateChangeOutput - if rf, ok := ret.Get(1).(func(*ecs.SubmitContainerStateChangeInput) *ecs.SubmitContainerStateChangeOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.SubmitContainerStateChangeOutput) - } - } - - return r0, r1 -} - -// SubmitContainerStateChangeWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) SubmitContainerStateChangeWithContext(_a0 context.Context, _a1 *ecs.SubmitContainerStateChangeInput, _a2 ...request.Option) (*ecs.SubmitContainerStateChangeOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.SubmitContainerStateChangeOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.SubmitContainerStateChangeInput, ...request.Option) *ecs.SubmitContainerStateChangeOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.SubmitContainerStateChangeOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.SubmitContainerStateChangeInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SubmitTaskStateChange provides a mock function with given fields: _a0 -func (_m *ECSAPI) SubmitTaskStateChange(_a0 *ecs.SubmitTaskStateChangeInput) (*ecs.SubmitTaskStateChangeOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.SubmitTaskStateChangeOutput - if rf, ok := ret.Get(0).(func(*ecs.SubmitTaskStateChangeInput) *ecs.SubmitTaskStateChangeOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.SubmitTaskStateChangeOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.SubmitTaskStateChangeInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SubmitTaskStateChangeRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) SubmitTaskStateChangeRequest(_a0 *ecs.SubmitTaskStateChangeInput) (*request.Request, *ecs.SubmitTaskStateChangeOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.SubmitTaskStateChangeInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.SubmitTaskStateChangeOutput - if rf, ok := ret.Get(1).(func(*ecs.SubmitTaskStateChangeInput) *ecs.SubmitTaskStateChangeOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.SubmitTaskStateChangeOutput) - } - } - - return r0, r1 -} - -// SubmitTaskStateChangeWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) SubmitTaskStateChangeWithContext(_a0 context.Context, _a1 *ecs.SubmitTaskStateChangeInput, _a2 ...request.Option) (*ecs.SubmitTaskStateChangeOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.SubmitTaskStateChangeOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.SubmitTaskStateChangeInput, ...request.Option) *ecs.SubmitTaskStateChangeOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.SubmitTaskStateChangeOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.SubmitTaskStateChangeInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// TagResource provides a mock function with given fields: _a0 -func (_m *ECSAPI) TagResource(_a0 *ecs.TagResourceInput) (*ecs.TagResourceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.TagResourceOutput - if rf, ok := ret.Get(0).(func(*ecs.TagResourceInput) *ecs.TagResourceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.TagResourceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.TagResourceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// TagResourceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) TagResourceRequest(_a0 *ecs.TagResourceInput) (*request.Request, *ecs.TagResourceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.TagResourceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.TagResourceOutput - if rf, ok := ret.Get(1).(func(*ecs.TagResourceInput) *ecs.TagResourceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.TagResourceOutput) - } - } - - return r0, r1 -} - -// TagResourceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) TagResourceWithContext(_a0 context.Context, _a1 *ecs.TagResourceInput, _a2 ...request.Option) (*ecs.TagResourceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.TagResourceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.TagResourceInput, ...request.Option) *ecs.TagResourceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.TagResourceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.TagResourceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UntagResource provides a mock function with given fields: _a0 -func (_m *ECSAPI) UntagResource(_a0 *ecs.UntagResourceInput) (*ecs.UntagResourceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UntagResourceOutput - if rf, ok := ret.Get(0).(func(*ecs.UntagResourceInput) *ecs.UntagResourceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UntagResourceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UntagResourceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UntagResourceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UntagResourceRequest(_a0 *ecs.UntagResourceInput) (*request.Request, *ecs.UntagResourceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UntagResourceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UntagResourceOutput - if rf, ok := ret.Get(1).(func(*ecs.UntagResourceInput) *ecs.UntagResourceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UntagResourceOutput) - } - } - - return r0, r1 -} - -// UntagResourceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UntagResourceWithContext(_a0 context.Context, _a1 *ecs.UntagResourceInput, _a2 ...request.Option) (*ecs.UntagResourceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UntagResourceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UntagResourceInput, ...request.Option) *ecs.UntagResourceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UntagResourceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UntagResourceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateCapacityProvider provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateCapacityProvider(_a0 *ecs.UpdateCapacityProviderInput) (*ecs.UpdateCapacityProviderOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateCapacityProviderOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateCapacityProviderInput) *ecs.UpdateCapacityProviderOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateCapacityProviderOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateCapacityProviderInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateCapacityProviderRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateCapacityProviderRequest(_a0 *ecs.UpdateCapacityProviderInput) (*request.Request, *ecs.UpdateCapacityProviderOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateCapacityProviderInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateCapacityProviderOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateCapacityProviderInput) *ecs.UpdateCapacityProviderOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateCapacityProviderOutput) - } - } - - return r0, r1 -} - -// UpdateCapacityProviderWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateCapacityProviderWithContext(_a0 context.Context, _a1 *ecs.UpdateCapacityProviderInput, _a2 ...request.Option) (*ecs.UpdateCapacityProviderOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateCapacityProviderOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateCapacityProviderInput, ...request.Option) *ecs.UpdateCapacityProviderOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateCapacityProviderOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateCapacityProviderInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateCluster provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateCluster(_a0 *ecs.UpdateClusterInput) (*ecs.UpdateClusterOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateClusterOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateClusterInput) *ecs.UpdateClusterOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateClusterOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateClusterInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateClusterRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateClusterRequest(_a0 *ecs.UpdateClusterInput) (*request.Request, *ecs.UpdateClusterOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateClusterInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateClusterOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateClusterInput) *ecs.UpdateClusterOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateClusterOutput) - } - } - - return r0, r1 -} - -// UpdateClusterSettings provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateClusterSettings(_a0 *ecs.UpdateClusterSettingsInput) (*ecs.UpdateClusterSettingsOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateClusterSettingsOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateClusterSettingsInput) *ecs.UpdateClusterSettingsOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateClusterSettingsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateClusterSettingsInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateClusterSettingsRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateClusterSettingsRequest(_a0 *ecs.UpdateClusterSettingsInput) (*request.Request, *ecs.UpdateClusterSettingsOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateClusterSettingsInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateClusterSettingsOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateClusterSettingsInput) *ecs.UpdateClusterSettingsOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateClusterSettingsOutput) - } - } - - return r0, r1 -} - -// UpdateClusterSettingsWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateClusterSettingsWithContext(_a0 context.Context, _a1 *ecs.UpdateClusterSettingsInput, _a2 ...request.Option) (*ecs.UpdateClusterSettingsOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateClusterSettingsOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateClusterSettingsInput, ...request.Option) *ecs.UpdateClusterSettingsOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateClusterSettingsOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateClusterSettingsInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateClusterWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateClusterWithContext(_a0 context.Context, _a1 *ecs.UpdateClusterInput, _a2 ...request.Option) (*ecs.UpdateClusterOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateClusterOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateClusterInput, ...request.Option) *ecs.UpdateClusterOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateClusterOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateClusterInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateContainerAgent provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateContainerAgent(_a0 *ecs.UpdateContainerAgentInput) (*ecs.UpdateContainerAgentOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateContainerAgentOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateContainerAgentInput) *ecs.UpdateContainerAgentOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateContainerAgentOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateContainerAgentInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateContainerAgentRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateContainerAgentRequest(_a0 *ecs.UpdateContainerAgentInput) (*request.Request, *ecs.UpdateContainerAgentOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateContainerAgentInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateContainerAgentOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateContainerAgentInput) *ecs.UpdateContainerAgentOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateContainerAgentOutput) - } - } - - return r0, r1 -} - -// UpdateContainerAgentWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateContainerAgentWithContext(_a0 context.Context, _a1 *ecs.UpdateContainerAgentInput, _a2 ...request.Option) (*ecs.UpdateContainerAgentOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateContainerAgentOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateContainerAgentInput, ...request.Option) *ecs.UpdateContainerAgentOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateContainerAgentOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateContainerAgentInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateContainerInstancesState provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateContainerInstancesState(_a0 *ecs.UpdateContainerInstancesStateInput) (*ecs.UpdateContainerInstancesStateOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateContainerInstancesStateOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateContainerInstancesStateInput) *ecs.UpdateContainerInstancesStateOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateContainerInstancesStateOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateContainerInstancesStateInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateContainerInstancesStateRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateContainerInstancesStateRequest(_a0 *ecs.UpdateContainerInstancesStateInput) (*request.Request, *ecs.UpdateContainerInstancesStateOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateContainerInstancesStateInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateContainerInstancesStateOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateContainerInstancesStateInput) *ecs.UpdateContainerInstancesStateOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateContainerInstancesStateOutput) - } - } - - return r0, r1 -} - -// UpdateContainerInstancesStateWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateContainerInstancesStateWithContext(_a0 context.Context, _a1 *ecs.UpdateContainerInstancesStateInput, _a2 ...request.Option) (*ecs.UpdateContainerInstancesStateOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateContainerInstancesStateOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateContainerInstancesStateInput, ...request.Option) *ecs.UpdateContainerInstancesStateOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateContainerInstancesStateOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateContainerInstancesStateInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateService provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateService(_a0 *ecs.UpdateServiceInput) (*ecs.UpdateServiceOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateServiceOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateServiceInput) *ecs.UpdateServiceOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateServiceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateServiceInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateServicePrimaryTaskSet provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateServicePrimaryTaskSet(_a0 *ecs.UpdateServicePrimaryTaskSetInput) (*ecs.UpdateServicePrimaryTaskSetOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateServicePrimaryTaskSetOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateServicePrimaryTaskSetInput) *ecs.UpdateServicePrimaryTaskSetOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateServicePrimaryTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateServicePrimaryTaskSetInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateServicePrimaryTaskSetRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateServicePrimaryTaskSetRequest(_a0 *ecs.UpdateServicePrimaryTaskSetInput) (*request.Request, *ecs.UpdateServicePrimaryTaskSetOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateServicePrimaryTaskSetInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateServicePrimaryTaskSetOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateServicePrimaryTaskSetInput) *ecs.UpdateServicePrimaryTaskSetOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateServicePrimaryTaskSetOutput) - } - } - - return r0, r1 -} - -// UpdateServicePrimaryTaskSetWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateServicePrimaryTaskSetWithContext(_a0 context.Context, _a1 *ecs.UpdateServicePrimaryTaskSetInput, _a2 ...request.Option) (*ecs.UpdateServicePrimaryTaskSetOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateServicePrimaryTaskSetOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateServicePrimaryTaskSetInput, ...request.Option) *ecs.UpdateServicePrimaryTaskSetOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateServicePrimaryTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateServicePrimaryTaskSetInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateServiceRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateServiceRequest(_a0 *ecs.UpdateServiceInput) (*request.Request, *ecs.UpdateServiceOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateServiceInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateServiceOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateServiceInput) *ecs.UpdateServiceOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateServiceOutput) - } - } - - return r0, r1 -} - -// UpdateServiceWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateServiceWithContext(_a0 context.Context, _a1 *ecs.UpdateServiceInput, _a2 ...request.Option) (*ecs.UpdateServiceOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateServiceOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateServiceInput, ...request.Option) *ecs.UpdateServiceOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateServiceOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateServiceInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateTaskSet provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateTaskSet(_a0 *ecs.UpdateTaskSetInput) (*ecs.UpdateTaskSetOutput, error) { - ret := _m.Called(_a0) - - var r0 *ecs.UpdateTaskSetOutput - if rf, ok := ret.Get(0).(func(*ecs.UpdateTaskSetInput) *ecs.UpdateTaskSetOutput); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*ecs.UpdateTaskSetInput) error); ok { - r1 = rf(_a0) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UpdateTaskSetRequest provides a mock function with given fields: _a0 -func (_m *ECSAPI) UpdateTaskSetRequest(_a0 *ecs.UpdateTaskSetInput) (*request.Request, *ecs.UpdateTaskSetOutput) { - ret := _m.Called(_a0) - - var r0 *request.Request - if rf, ok := ret.Get(0).(func(*ecs.UpdateTaskSetInput) *request.Request); ok { - r0 = rf(_a0) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*request.Request) - } - } - - var r1 *ecs.UpdateTaskSetOutput - if rf, ok := ret.Get(1).(func(*ecs.UpdateTaskSetInput) *ecs.UpdateTaskSetOutput); ok { - r1 = rf(_a0) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).(*ecs.UpdateTaskSetOutput) - } - } - - return r0, r1 -} - -// UpdateTaskSetWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) UpdateTaskSetWithContext(_a0 context.Context, _a1 *ecs.UpdateTaskSetInput, _a2 ...request.Option) (*ecs.UpdateTaskSetOutput, error) { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 *ecs.UpdateTaskSetOutput - if rf, ok := ret.Get(0).(func(context.Context, *ecs.UpdateTaskSetInput, ...request.Option) *ecs.UpdateTaskSetOutput); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*ecs.UpdateTaskSetOutput) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(context.Context, *ecs.UpdateTaskSetInput, ...request.Option) error); ok { - r1 = rf(_a0, _a1, _a2...) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// WaitUntilServicesInactive provides a mock function with given fields: _a0 -func (_m *ECSAPI) WaitUntilServicesInactive(_a0 *ecs.DescribeServicesInput) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.DescribeServicesInput) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilServicesInactiveWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) WaitUntilServicesInactiveWithContext(_a0 context.Context, _a1 *ecs.DescribeServicesInput, _a2 ...request.WaiterOption) error { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeServicesInput, ...request.WaiterOption) error); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilServicesStable provides a mock function with given fields: _a0 -func (_m *ECSAPI) WaitUntilServicesStable(_a0 *ecs.DescribeServicesInput) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.DescribeServicesInput) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilServicesStableWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) WaitUntilServicesStableWithContext(_a0 context.Context, _a1 *ecs.DescribeServicesInput, _a2 ...request.WaiterOption) error { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeServicesInput, ...request.WaiterOption) error); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilTasksRunning provides a mock function with given fields: _a0 -func (_m *ECSAPI) WaitUntilTasksRunning(_a0 *ecs.DescribeTasksInput) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.DescribeTasksInput) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilTasksRunningWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) WaitUntilTasksRunningWithContext(_a0 context.Context, _a1 *ecs.DescribeTasksInput, _a2 ...request.WaiterOption) error { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeTasksInput, ...request.WaiterOption) error); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilTasksStopped provides a mock function with given fields: _a0 -func (_m *ECSAPI) WaitUntilTasksStopped(_a0 *ecs.DescribeTasksInput) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func(*ecs.DescribeTasksInput) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// WaitUntilTasksStoppedWithContext provides a mock function with given fields: _a0, _a1, _a2 -func (_m *ECSAPI) WaitUntilTasksStoppedWithContext(_a0 context.Context, _a1 *ecs.DescribeTasksInput, _a2 ...request.WaiterOption) error { - _va := make([]interface{}, len(_a2)) - for _i := range _a2 { - _va[_i] = _a2[_i] - } - var _ca []interface{} - _ca = append(_ca, _a0, _a1) - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *ecs.DescribeTasksInput, ...request.WaiterOption) error); ok { - r0 = rf(_a0, _a1, _a2...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewECSAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewECSAPI creates a new instance of ECSAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewECSAPI(t mockConstructorTestingTNewECSAPI) *ECSAPI { - mock := &ECSAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/devnet/cmd/scale-down-ecs-service/mocks_test.go b/devnet/cmd/scale-down-ecs-service/mocks_test.go new file mode 100644 index 0000000000..99498f0366 --- /dev/null +++ b/devnet/cmd/scale-down-ecs-service/mocks_test.go @@ -0,0 +1,97 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/ChainSafe/gossamer/devnet/cmd/scale-down-ecs-service/internal (interfaces: ECSAPI) + +// Package main is a generated GoMock package. +package main + +import ( + context "context" + reflect "reflect" + + request "github.com/aws/aws-sdk-go/aws/request" + ecs "github.com/aws/aws-sdk-go/service/ecs" + gomock "github.com/golang/mock/gomock" +) + +// MockECSAPI is a mock of ECSAPI interface. +type MockECSAPI struct { + ctrl *gomock.Controller + recorder *MockECSAPIMockRecorder +} + +// MockECSAPIMockRecorder is the mock recorder for MockECSAPI. +type MockECSAPIMockRecorder struct { + mock *MockECSAPI +} + +// NewMockECSAPI creates a new mock instance. +func NewMockECSAPI(ctrl *gomock.Controller) *MockECSAPI { + mock := &MockECSAPI{ctrl: ctrl} + mock.recorder = &MockECSAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockECSAPI) EXPECT() *MockECSAPIMockRecorder { + return m.recorder +} + +// DescribeServicesWithContext mocks base method. +func (m *MockECSAPI) DescribeServicesWithContext(arg0 context.Context, arg1 *ecs.DescribeServicesInput, arg2 ...request.Option) (*ecs.DescribeServicesOutput, error) { + m.ctrl.T.Helper() + varargs := []interface{}{arg0, arg1} + for _, a := range arg2 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "DescribeServicesWithContext", varargs...) + ret0, _ := ret[0].(*ecs.DescribeServicesOutput) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DescribeServicesWithContext indicates an expected call of DescribeServicesWithContext. +func (mr *MockECSAPIMockRecorder) DescribeServicesWithContext(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{arg0, arg1}, arg2...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DescribeServicesWithContext", reflect.TypeOf((*MockECSAPI)(nil).DescribeServicesWithContext), varargs...) +} + +// ListServicesWithContext mocks base method. +func (m *MockECSAPI) ListServicesWithContext(arg0 context.Context, arg1 *ecs.ListServicesInput, arg2 ...request.Option) (*ecs.ListServicesOutput, error) { + m.ctrl.T.Helper() + varargs := []interface{}{arg0, arg1} + for _, a := range arg2 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "ListServicesWithContext", varargs...) + ret0, _ := ret[0].(*ecs.ListServicesOutput) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ListServicesWithContext indicates an expected call of ListServicesWithContext. +func (mr *MockECSAPIMockRecorder) ListServicesWithContext(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{arg0, arg1}, arg2...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListServicesWithContext", reflect.TypeOf((*MockECSAPI)(nil).ListServicesWithContext), varargs...) +} + +// UpdateServiceWithContext mocks base method. +func (m *MockECSAPI) UpdateServiceWithContext(arg0 context.Context, arg1 *ecs.UpdateServiceInput, arg2 ...request.Option) (*ecs.UpdateServiceOutput, error) { + m.ctrl.T.Helper() + varargs := []interface{}{arg0, arg1} + for _, a := range arg2 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "UpdateServiceWithContext", varargs...) + ret0, _ := ret[0].(*ecs.UpdateServiceOutput) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// UpdateServiceWithContext indicates an expected call of UpdateServiceWithContext. +func (mr *MockECSAPIMockRecorder) UpdateServiceWithContext(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + varargs := append([]interface{}{arg0, arg1}, arg2...) + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateServiceWithContext", reflect.TypeOf((*MockECSAPI)(nil).UpdateServiceWithContext), varargs...) +} diff --git a/devnet/cmd/scale-down-ecs-service/service_scaler.go b/devnet/cmd/scale-down-ecs-service/service_scaler.go index 6ed6b14f1c..4d726feacb 100644 --- a/devnet/cmd/scale-down-ecs-service/service_scaler.go +++ b/devnet/cmd/scale-down-ecs-service/service_scaler.go @@ -9,18 +9,18 @@ import ( "regexp" "time" + "github.com/ChainSafe/gossamer/devnet/cmd/scale-down-ecs-service/internal" "github.com/aws/aws-sdk-go/aws" "github.com/aws/aws-sdk-go/service/ecs" - "github.com/aws/aws-sdk-go/service/ecs/ecsiface" ) type serviceScaler struct { tickerDuration time.Duration cluster string - ecs ecsiface.ECSAPI + ecs internal.ECSAPI } -func newServiceScaler(tickerDuration time.Duration, cluster string, ecs ecsiface.ECSAPI) *serviceScaler { +func newServiceScaler(tickerDuration time.Duration, cluster string, ecs internal.ECSAPI) *serviceScaler { return &serviceScaler{ tickerDuration: tickerDuration, cluster: cluster, @@ -102,7 +102,6 @@ func (ss serviceScaler) waitForRunningCount(ctx context.Context, serviceArns []* return err } } - return } func (ss serviceScaler) scaleServices(ctx context.Context, servicesRegex string) (err error) { diff --git a/devnet/cmd/scale-down-ecs-service/service_scaler_test.go b/devnet/cmd/scale-down-ecs-service/service_scaler_test.go index c6b50936a2..cec1f46d1a 100644 --- a/devnet/cmd/scale-down-ecs-service/service_scaler_test.go +++ b/devnet/cmd/scale-down-ecs-service/service_scaler_test.go @@ -10,45 +10,50 @@ import ( "testing" "time" - "github.com/ChainSafe/gossamer/devnet/cmd/scale-down-ecs-service/mocks" + "github.com/ChainSafe/gossamer/devnet/cmd/scale-down-ecs-service/internal" "github.com/aws/aws-sdk-go/aws" + request "github.com/aws/aws-sdk-go/aws/request" "github.com/aws/aws-sdk-go/service/ecs" - "github.com/aws/aws-sdk-go/service/ecs/ecsiface" - "github.com/stretchr/testify/mock" + "github.com/golang/mock/gomock" ) -//go:generate mockery --srcpkg=github.com/aws/aws-sdk-go/service/ecs/ecsiface --name ECSAPI --case underscore +//go:generate mockgen -destination=mocks_test.go -package=$GOPACKAGE github.com/ChainSafe/gossamer/devnet/cmd/scale-down-ecs-service/internal ECSAPI func Test_serviceScaler_findServiceArns(t *testing.T) { - mockECS := mocks.NewECSAPI(t) - mockECS. - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + ctrl := gomock.NewController(t) + + mockECS := NewMockECSAPI(ctrl) + mockECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someCluster"), }).Return(&ecs.ListServicesOutput{ ServiceArns: []*string{ aws.String("someArn0"), aws.String("someArn1"), }, - NextToken: aws.String("someNextToken")}, nil).Once(). - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + NextToken: aws.String("someNextToken")}, nil) + mockECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someCluster"), NextToken: aws.String("someNextToken"), }).Return(&ecs.ListServicesOutput{ ServiceArns: []*string{ aws.String("someArn2"), aws.String("someArn3"), - }}, nil).Once(). - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + }}, nil) + mockECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someErrCluster"), - }).Return(nil, fmt.Errorf("someErr")).Once(). - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + }).Return(nil, fmt.Errorf("someErr")) + mockECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someEmptyCluster"), - }).Return(&ecs.ListServicesOutput{}, nil).Once() + }).Return(&ecs.ListServicesOutput{}, nil) type fields struct { tickerDuration time.Duration cluster string - ecs ecsiface.ECSAPI + ecs internal.ECSAPI } type args struct { @@ -135,28 +140,32 @@ func Test_serviceScaler_findServiceArns(t *testing.T) { } func Test_serviceScaler_drainServices(t *testing.T) { - mockECS := mocks.NewECSAPI(t) - mockECS. - On("UpdateServiceWithContext", mock.Anything, &ecs.UpdateServiceInput{ + ctrl := gomock.NewController(t) + + mockECS := NewMockECSAPI(ctrl) + mockECS.EXPECT(). + UpdateServiceWithContext(gomock.Any(), &ecs.UpdateServiceInput{ Cluster: aws.String("someCluster"), Service: aws.String("someArn0"), DesiredCount: aws.Int64(0), - }).Return(&ecs.UpdateServiceOutput{}, nil).Once(). - On("UpdateServiceWithContext", mock.Anything, &ecs.UpdateServiceInput{ + }).Return(&ecs.UpdateServiceOutput{}, nil) + mockECS.EXPECT(). + UpdateServiceWithContext(gomock.Any(), &ecs.UpdateServiceInput{ Cluster: aws.String("someCluster"), Service: aws.String("someArn1"), DesiredCount: aws.Int64(0), - }).Return(&ecs.UpdateServiceOutput{}, nil).Once(). - On("UpdateServiceWithContext", mock.Anything, &ecs.UpdateServiceInput{ + }).Return(&ecs.UpdateServiceOutput{}, nil) + mockECS.EXPECT(). + UpdateServiceWithContext(gomock.Any(), &ecs.UpdateServiceInput{ Cluster: aws.String("someErrCluster"), Service: aws.String("someArn0"), DesiredCount: aws.Int64(0), - }).Return(nil, fmt.Errorf("some Error")).Once() + }).Return(nil, fmt.Errorf("some Error")) type fields struct { tickerDuration time.Duration cluster string - ecs ecsiface.ECSAPI + ecs internal.ECSAPI } type args struct { ctx context.Context @@ -213,9 +222,11 @@ func Test_serviceScaler_drainServices(t *testing.T) { } func Test_serviceScaler_waitForRunningCount(t *testing.T) { - mockECS := mocks.NewECSAPI(t) - mockECS. - On("DescribeServicesWithContext", mock.Anything, &ecs.DescribeServicesInput{ + ctrl := gomock.NewController(t) + + mockECS := NewMockECSAPI(ctrl) + mockECS.EXPECT(). + DescribeServicesWithContext(gomock.Any(), &ecs.DescribeServicesInput{ Cluster: aws.String("someCluster"), Services: []*string{ aws.String("someArn0"), @@ -231,47 +242,50 @@ func Test_serviceScaler_waitForRunningCount(t *testing.T) { RunningCount: aws.Int64(0), ServiceArn: aws.String("someArn1"), }, - }}, nil).Once(). - On("DescribeServicesWithContext", mock.Anything, &ecs.DescribeServicesInput{ + }}, nil) + mockECS.EXPECT(). + DescribeServicesWithContext(gomock.Any(), &ecs.DescribeServicesInput{ Cluster: aws.String("someErrorCluster"), Services: []*string{ aws.String("someArn0"), aws.String("someArn1"), }, - }).Return(nil, fmt.Errorf("someError")).Once() + }).Return(nil, fmt.Errorf("someError")) ctx, cancel := context.WithCancel(context.Background()) - mockECSCancel := mocks.NewECSAPI(t) - mockECSCancel. - On("DescribeServicesWithContext", mock.Anything, &ecs.DescribeServicesInput{ + mockECSCancel := NewMockECSAPI(ctrl) + mockECSCancel.EXPECT(). + DescribeServicesWithContext(gomock.Any(), &ecs.DescribeServicesInput{ Cluster: aws.String("someCluster"), Services: []*string{ aws.String("someArn0"), aws.String("someArn1"), }, - }).Return(&ecs.DescribeServicesOutput{ - Services: []*ecs.Service{ - { - RunningCount: aws.Int64(1), - ServiceArn: aws.String("someArn0"), - }, - { - RunningCount: aws.Int64(1), - ServiceArn: aws.String("someArn1"), - }, - }}, nil).Run( - func(args mock.Arguments) { - go func() { - // should trigger before 10ms ticker - <-time.After(2 * time.Millisecond) - cancel() - }() - }).Once() + }).DoAndReturn(func(_ context.Context, _ *ecs.DescribeServicesInput, _ ...request.Option) ( + *ecs.DescribeServicesOutput, error) { + go func() { + // should trigger before 10ms ticker + <-time.After(2 * time.Millisecond) + cancel() + }() + + return &ecs.DescribeServicesOutput{ + Services: []*ecs.Service{ + { + RunningCount: aws.Int64(1), + ServiceArn: aws.String("someArn0"), + }, + { + RunningCount: aws.Int64(1), + ServiceArn: aws.String("someArn1"), + }, + }}, nil + }) type fields struct { tickerDuration time.Duration cluster string - ecs ecsiface.ECSAPI + ecs internal.ECSAPI } type args struct { ctx context.Context @@ -349,7 +363,7 @@ func Test_newServiceScaler(t *testing.T) { type args struct { tickerDuration time.Duration cluster string - ecs ecsiface.ECSAPI + ecs internal.ECSAPI } tests := []struct { name string @@ -371,26 +385,31 @@ func Test_newServiceScaler(t *testing.T) { } func Test_serviceScaler_scaleServices(t *testing.T) { - mockECS := mocks.NewECSAPI(t) - mockECS. - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + ctrl := gomock.NewController(t) + + mockECS := NewMockECSAPI(ctrl) + mockECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someCluster"), }).Return(&ecs.ListServicesOutput{ ServiceArns: []*string{ aws.String("someArn0"), aws.String("someArn1"), - }}, nil).Once(). - On("UpdateServiceWithContext", mock.Anything, &ecs.UpdateServiceInput{ + }}, nil) + mockECS.EXPECT(). + UpdateServiceWithContext(gomock.Any(), &ecs.UpdateServiceInput{ Cluster: aws.String("someCluster"), Service: aws.String("someArn0"), DesiredCount: aws.Int64(0), - }).Return(&ecs.UpdateServiceOutput{}, nil).Once(). - On("UpdateServiceWithContext", mock.Anything, &ecs.UpdateServiceInput{ + }).Return(&ecs.UpdateServiceOutput{}, nil) + mockECS.EXPECT(). + UpdateServiceWithContext(gomock.Any(), &ecs.UpdateServiceInput{ Cluster: aws.String("someCluster"), Service: aws.String("someArn1"), DesiredCount: aws.Int64(0), - }).Return(&ecs.UpdateServiceOutput{}, nil).Once(). - On("DescribeServicesWithContext", mock.Anything, &ecs.DescribeServicesInput{ + }).Return(&ecs.UpdateServiceOutput{}, nil) + mockECS.EXPECT(). + DescribeServicesWithContext(gomock.Any(), &ecs.DescribeServicesInput{ Cluster: aws.String("someCluster"), Services: []*string{ aws.String("someArn0"), @@ -406,33 +425,34 @@ func Test_serviceScaler_scaleServices(t *testing.T) { RunningCount: aws.Int64(0), ServiceArn: aws.String("someArn1"), }, - }}, nil).Once() + }}, nil) - findServiceArnsErrECS := mocks.NewECSAPI(t) - findServiceArnsErrECS. - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + findServiceArnsErrECS := NewMockECSAPI(ctrl) + findServiceArnsErrECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someCluster"), - }).Return(nil, fmt.Errorf("someError")).Once() + }).Return(nil, fmt.Errorf("someError")) - updateServicesErrECS := mocks.NewECSAPI(t) - updateServicesErrECS. - On("ListServicesWithContext", mock.Anything, &ecs.ListServicesInput{ + updateServicesErrECS := NewMockECSAPI(ctrl) + updateServicesErrECS.EXPECT(). + ListServicesWithContext(gomock.Any(), &ecs.ListServicesInput{ Cluster: aws.String("someCluster"), }).Return(&ecs.ListServicesOutput{ ServiceArns: []*string{ aws.String("someArn0"), aws.String("someArn1"), - }}, nil).Once(). - On("UpdateServiceWithContext", mock.Anything, &ecs.UpdateServiceInput{ + }}, nil) + updateServicesErrECS.EXPECT(). + UpdateServiceWithContext(gomock.Any(), &ecs.UpdateServiceInput{ Cluster: aws.String("someCluster"), Service: aws.String("someArn0"), DesiredCount: aws.Int64(0), - }).Return(nil, fmt.Errorf("someError")).Once() + }).Return(nil, fmt.Errorf("someError")) type fields struct { tickerDuration time.Duration cluster string - ecs ecsiface.ECSAPI + ecs internal.ECSAPI } type args struct { ctx context.Context diff --git a/devnet/go.mod b/devnet/go.mod index 36a288d0e5..f20b1a4166 100644 --- a/devnet/go.mod +++ b/devnet/go.mod @@ -4,16 +4,12 @@ go 1.18 require ( github.com/aws/aws-sdk-go v1.42.22 + github.com/golang/mock v1.6.0 github.com/jessevdk/go-flags v1.5.0 - github.com/stretchr/testify v1.7.0 gopkg.in/yaml.v2 v2.4.0 ) require ( - github.com/davecgh/go-spew v1.1.0 // indirect github.com/jmespath/go-jmespath v0.4.0 // indirect - github.com/pmezard/go-difflib v1.0.0 // indirect - github.com/stretchr/objx v0.1.0 // indirect - golang.org/x/sys v0.0.0-20210423082822-04245dca01da // indirect - gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c // indirect + golang.org/x/sys v0.0.0-20210510120138-977fb7262007 // indirect ) diff --git a/devnet/go.sum b/devnet/go.sum index 067f68c629..ba9564a1e8 100644 --- a/devnet/go.sum +++ b/devnet/go.sum @@ -2,6 +2,8 @@ github.com/aws/aws-sdk-go v1.42.22 h1:EwcM7/+Ytg6xK+jbeM2+f9OELHqPiEiEKetT/GgAr7 github.com/aws/aws-sdk-go v1.42.22/go.mod h1:585smgzpB/KqRA+K3y/NL/oYRqQvpNJYvLm+LY1U59Q= github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/golang/mock v1.6.0 h1:ErTB+efbowRARo13NNdxyJji2egdxLGQhRaY+DUumQc= +github.com/golang/mock v1.6.0/go.mod h1:p6yTPP+5HYm5mzsMV8JkE6ZKdX+/wYM6Hr+LicevLPs= github.com/jessevdk/go-flags v1.5.0 h1:1jKYvbxEjfUl0fmqTCOfonvskHHXMjBySTLW4y9LFvc= github.com/jessevdk/go-flags v1.5.0/go.mod h1:Fw0T6WPc1dYxT4mKEZRfG5kJhaTDP9pj1c2EWnYs/m4= github.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg= @@ -11,22 +13,37 @@ github.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfC github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= -github.com/stretchr/objx v0.1.0 h1:4G4v2dO3VZwixGIRoQ5Lfboy6nUhCyYzaqnIAPPhYs4= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= -github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= -github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20210405180319-a5a99cb37ef4/go.mod h1:p54w0d4576C0XHj96bSt6lcn1PtDYWL6XObtHCRCNQM= golang.org/x/net v0.0.0-20210614182718-04defd469f4e/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20210320140829-1e4c9ba3b0c4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= -golang.org/x/sys v0.0.0-20210423082822-04245dca01da h1:b3NXsE2LusjYGGjL5bxEVZZORm/YEFFrWFjR8eFrw/c= +golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210510120138-977fb7262007 h1:gG67DSER+11cZvqIMb8S8bt0vZtiN6xWYARwirrOSfE= +golang.org/x/sys v0.0.0-20210510120138-977fb7262007/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.1.1/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= -gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo= -gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/dot/rpc/http_test.go b/dot/rpc/http_test.go index 25a19186c8..530bafb441 100644 --- a/dot/rpc/http_test.go +++ b/dot/rpc/http_test.go @@ -30,7 +30,6 @@ import ( "github.com/ChainSafe/gossamer/lib/runtime/wasmer" "github.com/btcsuite/btcutil/base58" "github.com/golang/mock/gomock" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) @@ -98,6 +97,8 @@ func TestUnsafeRPCProtection(t *testing.T) { } } func TestRPCUnsafeExpose(t *testing.T) { + ctrl := gomock.NewController(t) + data := []byte(fmt.Sprintf( `{"jsonrpc":"2.0","method":"%s","params":["%s"],"id":1}`, "system_addReservedPeer", @@ -107,8 +108,8 @@ func TestRPCUnsafeExpose(t *testing.T) { _, err := buf.Write(data) require.NoError(t, err) - netmock := mocks.NewNetworkAPI(t) - netmock.On("AddReservedPeers", mock.AnythingOfType("string")).Return(nil) + netmock := mocks.NewMockNetworkAPI(ctrl) + netmock.EXPECT().AddReservedPeers(gomock.Any()).Return(nil) cfg := &HTTPServerConfig{ Modules: []string{"system"}, @@ -175,6 +176,8 @@ func TestUnsafeRPCJustToLocalhost(t *testing.T) { } func TestRPCExternalEnable_UnsafeExternalNotEnabled(t *testing.T) { + ctrl := gomock.NewController(t) + unsafeData := []byte(fmt.Sprintf( `{"jsonrpc":"2.0","method":"%s","params":["%s"],"id":1}`, "system_addReservedPeer", @@ -188,8 +191,8 @@ func TestRPCExternalEnable_UnsafeExternalNotEnabled(t *testing.T) { safebuf := new(bytes.Buffer) safebuf.Write(safeData) - netmock := mocks.NewNetworkAPI(t) - netmock.On("NetworkState").Return(common.NetworkState{ + netmock := mocks.NewMockNetworkAPI(ctrl) + netmock.EXPECT().NetworkState().Return(common.NetworkState{ PeerID: "peer id", }) diff --git a/dot/rpc/modules/api.go b/dot/rpc/modules/api.go index 9c4dfbebf5..0f75118ef1 100644 --- a/dot/rpc/modules/api.go +++ b/dot/rpc/modules/api.go @@ -16,8 +16,6 @@ import ( "github.com/ChainSafe/gossamer/lib/trie" ) -//go:generate mockery --name StorageAPI --structname StorageAPI --case underscore --keeptree - // StorageAPI is the interface for the storage state type StorageAPI interface { GetStorage(root *common.Hash, key []byte) ([]byte, error) @@ -31,8 +29,6 @@ type StorageAPI interface { UnregisterStorageObserver(observer state.Observer) } -//go:generate mockery --name BlockAPI --structname BlockAPI --case underscore --keeptree - // BlockAPI is the interface for the block state type BlockAPI interface { GetHeader(hash common.Hash) (*types.Header, error) @@ -53,8 +49,6 @@ type BlockAPI interface { GetRuntime(blockHash common.Hash) (instance state.Runtime, err error) } -//go:generate mockery --name NetworkAPI --structname NetworkAPI --case underscore --keeptree - // NetworkAPI interface for network state methods type NetworkAPI interface { Health() common.Health @@ -68,8 +62,6 @@ type NetworkAPI interface { RemoveReservedPeers(addrs ...string) error } -//go:generate mockery --name BlockProducerAPI --structname BlockProducerAPI --case underscore --keeptree - // BlockProducerAPI is the interface for BlockProducer methods type BlockProducerAPI interface { Pause() error @@ -78,15 +70,11 @@ type BlockProducerAPI interface { SlotDuration() uint64 } -//go:generate mockery --name TransactionStateAPI --structname TransactionStateAPI --case underscore --keeptree - // TransactionStateAPI ... type TransactionStateAPI interface { Pending() []*transaction.ValidTransaction } -//go:generate mockery --name CoreAPI --structname CoreAPI --case underscore --keeptree - // CoreAPI is the interface for the core methods type CoreAPI interface { InsertKey(kp core.KeyPair, keystoreType string) error @@ -103,8 +91,6 @@ type RPCAPI interface { Methods() []string } -//go:generate mockery --name SystemAPI --structname SystemAPI --case underscore --keeptree - // SystemAPI is the interface for handling system methods type SystemAPI interface { SystemName() string @@ -114,8 +100,6 @@ type SystemAPI interface { ChainName() string } -//go:generate mockery --name BlockFinalityAPI --structname BlockFinalityAPI --case underscore --keeptree - // BlockFinalityAPI is the interface for handling block finalisation methods type BlockFinalityAPI interface { GetSetID() uint64 @@ -125,8 +109,6 @@ type BlockFinalityAPI interface { PreCommits() []ed25519.PublicKeyBytes } -//go:generate mockery --name RuntimeStorageAPI --structname RuntimeStorageAPI --case underscore --keeptree - // RuntimeStorageAPI is the interface to interacts with the node storage type RuntimeStorageAPI interface { SetLocal(k, v []byte) error @@ -135,15 +117,11 @@ type RuntimeStorageAPI interface { GetPersistent(k []byte) ([]byte, error) } -//go:generate mockery --name SyncStateAPI --structname SyncStateAPI --case underscore --keeptree - // SyncStateAPI is the interface to interact with sync state. type SyncStateAPI interface { GenSyncSpec(raw bool) (*genesis.Genesis, error) } -//go:generate mockgen -destination=mock_sync_api_test.go -package $GOPACKAGE . SyncAPI - // SyncAPI is the interface to interact with the sync service type SyncAPI interface { HighestBlock() uint diff --git a/dot/rpc/modules/api_mocks.go b/dot/rpc/modules/api_mocks.go index 4385e5ee03..38516bf219 100644 --- a/dot/rpc/modules/api_mocks.go +++ b/dot/rpc/modules/api_mocks.go @@ -4,64 +4,58 @@ package modules import ( - "testing" - modulesmocks "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" "github.com/ChainSafe/gossamer/dot/types" "github.com/ChainSafe/gossamer/lib/common" "github.com/ChainSafe/gossamer/lib/runtime" - "github.com/stretchr/testify/mock" + "github.com/golang/mock/gomock" ) -// NewMockeryStorageAPI creates and return an rpc StorageAPI interface mock -func NewMockeryStorageAPI(t *testing.T) *modulesmocks.StorageAPI { - m := modulesmocks.NewStorageAPI(t) - m.On("GetStorage", mock.AnythingOfType("*common.Hash"), mock.AnythingOfType("[]uint8")).Return(nil, nil).Maybe() - m.On("GetStorageFromChild", mock.AnythingOfType("*common.Hash"), mock.AnythingOfType("[]uint8"), - mock.AnythingOfType("[]uint8")).Return(nil, nil).Maybe() - m.On("Entries", mock.AnythingOfType("*common.Hash")).Return(nil, nil).Maybe() - m.On("GetStorageByBlockHash", mock.AnythingOfType("common.Hash"), mock.AnythingOfType("[]uint8")). - Return(nil, nil).Maybe() - m.On("RegisterStorageObserver", mock.Anything).Maybe() - m.On("UnregisterStorageObserver", mock.Anything).Maybe() - m.On("GetStateRootFromBlock", mock.AnythingOfType("*common.Hash")).Return(nil, nil).Maybe() - m.On("GetKeysWithPrefix", mock.AnythingOfType("*common.Hash"), mock.AnythingOfType("[]uint8")).Return(nil, nil).Maybe() +// NewMockAnyStorageAPI creates and return an rpc StorageAPI interface mock +func NewMockAnyStorageAPI(ctrl *gomock.Controller) *modulesmocks.MockStorageAPI { + m := modulesmocks.NewMockStorageAPI(ctrl) + m.EXPECT().GetStorage(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes() + m.EXPECT().GetStorageFromChild(gomock.Any(), gomock.Any(), gomock.Any()). + Return(nil, nil).AnyTimes() + m.EXPECT().Entries(gomock.Any()).Return(nil, nil).AnyTimes() + m.EXPECT().GetStorageByBlockHash(gomock.Any(), gomock.Any()). + Return(nil, nil).AnyTimes() + m.EXPECT().RegisterStorageObserver(gomock.Any()).AnyTimes() + m.EXPECT().UnregisterStorageObserver(gomock.Any()).AnyTimes() + m.EXPECT().GetStateRootFromBlock(gomock.Any()).Return(nil, nil).AnyTimes() + m.EXPECT().GetKeysWithPrefix(gomock.Any(), gomock.Any()).Return(nil, nil).AnyTimes() return m } -// NewMockeryBlockAPI creates and return an rpc BlockAPI interface mock -func NewMockeryBlockAPI(t *testing.T) *modulesmocks.BlockAPI { - m := modulesmocks.NewBlockAPI(t) - m.On("GetHeader", mock.AnythingOfType("common.Hash")).Return(nil, nil).Maybe() - m.On("BestBlockHash").Return(common.Hash{}).Maybe() - m.On("GetBlockByHash", mock.AnythingOfType("common.Hash")).Return(nil, nil).Maybe() - m.On("GetHashByNumber", mock.AnythingOfType("uint")).Return(nil, nil).Maybe() - m.On("GetFinalisedHash", mock.AnythingOfType("uint64"), mock.AnythingOfType("uint64")). - Return(common.Hash{}, nil).Maybe() - m.On("GetHighestFinalisedHash").Return(common.Hash{}, nil).Maybe() - m.On("GetImportedBlockNotifierChannel").Return(make(chan *types.Block, 5)).Maybe() - m.On("FreeImportedBlockNotifierChannel", mock.AnythingOfType("chan *types.Block")).Maybe() - m.On("GetFinalisedNotifierChannel").Return(make(chan *types.FinalisationInfo, 5)).Maybe() - m.On("FreeFinalisedNotifierChannel", mock.AnythingOfType("chan *types.FinalisationInfo")).Maybe() - m.On("GetJustification", mock.AnythingOfType("common.Hash")).Return(make([]byte, 10), nil).Maybe() - m.On("HasJustification", mock.AnythingOfType("common.Hash")).Return(true, nil).Maybe() - m.On("SubChain", mock.AnythingOfType("common.Hash"), mock.AnythingOfType("common.Hash")). - Return(make([]common.Hash, 0), nil).Maybe() - m.On("RegisterRuntimeUpdatedChannel", mock.AnythingOfType("chan<- runtime.Version")). - Return(uint32(0), nil).Maybe() - +// NewMockAnyBlockAPI creates and return an rpc BlockAPI interface mock +func NewMockAnyBlockAPI(ctrl *gomock.Controller) *modulesmocks.MockBlockAPI { + m := modulesmocks.NewMockBlockAPI(ctrl) + m.EXPECT().GetHeader(gomock.Any()).Return(nil, nil).AnyTimes() + m.EXPECT().BestBlockHash().Return(common.Hash{}).AnyTimes() + m.EXPECT().GetBlockByHash(gomock.Any()).Return(nil, nil).AnyTimes() + m.EXPECT().GetHashByNumber(gomock.Any()).Return(common.Hash{}, nil).AnyTimes() + m.EXPECT().GetFinalisedHash(gomock.Any(), gomock.Any()). + Return(common.Hash{}, nil).AnyTimes() + m.EXPECT().GetHighestFinalisedHash().Return(common.Hash{}, nil).AnyTimes() + m.EXPECT().GetImportedBlockNotifierChannel().Return(make(chan *types.Block, 5)).AnyTimes() + m.EXPECT().FreeImportedBlockNotifierChannel(gomock.Any()).AnyTimes() + m.EXPECT().GetFinalisedNotifierChannel().Return(make(chan *types.FinalisationInfo, 5)).AnyTimes() + m.EXPECT().FreeFinalisedNotifierChannel(gomock.Any()).AnyTimes() + m.EXPECT().GetJustification(gomock.Any()).Return(make([]byte, 10), nil).AnyTimes() + m.EXPECT().HasJustification(gomock.Any()).Return(true, nil).AnyTimes() + m.EXPECT().RegisterRuntimeUpdatedChannel(gomock.Any()). + Return(uint32(0), nil).AnyTimes() return m } -// NewMockCoreAPI creates and return an rpc CoreAPI interface mock -func NewMockCoreAPI(t *testing.T) *modulesmocks.CoreAPI { - m := modulesmocks.NewCoreAPI(t) - m.On("InsertKey", mock.AnythingOfType("core.Keypair"), mock.AnythingOfType("string")).Return(nil).Maybe() - m.On("HasKey", mock.AnythingOfType("string"), mock.AnythingOfType("string")).Return(false, nil).Maybe() - m.On("GetRuntimeVersion", mock.AnythingOfType("*common.Hash")). - Return(runtime.Version{SpecName: []byte(`mock-spec`)}, nil).Maybe() - m.On("IsBlockProducer").Return(false).Maybe() - m.On("HandleSubmittedExtrinsic", mock.AnythingOfType("types.Extrinsic")).Return(nil).Maybe() - m.On("GetMetadata", mock.AnythingOfType("*common.Hash")).Return(nil, nil).Maybe() +// NewMockAnyAPI creates and return an rpc CoreAPI interface mock +func NewMockAnyAPI(ctrl *gomock.Controller) *modulesmocks.MockCoreAPI { + m := modulesmocks.NewMockCoreAPI(ctrl) + m.EXPECT().InsertKey(gomock.Any(), gomock.Any()).Return(nil).AnyTimes() + m.EXPECT().HasKey(gomock.Any(), gomock.Any()).Return(false, nil).AnyTimes() + m.EXPECT().GetRuntimeVersion(gomock.Any()). + Return(runtime.Version{SpecName: []byte(`mock-spec`)}, nil).AnyTimes() + m.EXPECT().HandleSubmittedExtrinsic(gomock.Any()).Return(nil).AnyTimes() + m.EXPECT().GetMetadata(gomock.Any()).Return(nil, nil).AnyTimes() return m } diff --git a/dot/rpc/modules/author_test.go b/dot/rpc/modules/author_test.go index b405322891..e211cf4e8f 100644 --- a/dot/rpc/modules/author_test.go +++ b/dot/rpc/modules/author_test.go @@ -18,9 +18,9 @@ import ( "github.com/ChainSafe/gossamer/lib/crypto/sr25519" "github.com/ChainSafe/gossamer/lib/keystore" "github.com/ChainSafe/gossamer/lib/transaction" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) @@ -36,21 +36,24 @@ func TestAuthorModule_HasSessionKeys(t *testing.T) { "9a9d2a24213896ff06895db16aade8b6502f3a71cf56374cc38520426026696d6f6e8034309a9d2a24213896ff06895" + "db16aade8b6502f3a71cf56374cc3852042602661756469") - coreMockAPIUnmarshalErr := mocks.NewCoreAPI(t) - coreMockAPIUnmarshalErr.On("DecodeSessionKeys", []byte{0x4, 0x1}).Return([]byte{0x4, 0x1}, nil) + ctrl := gomock.NewController(t) - coreMockAPIOk := mocks.NewCoreAPI(t) - coreMockAPIOk.On("DecodeSessionKeys", pkeys).Return(data, nil) - coreMockAPIOk.On("HasKey", mock.AnythingOfType("string"), mock.AnythingOfType("string")).Return(true, nil) + coreMockAPIUnmarshalErr := mocks.NewMockCoreAPI(ctrl) + coreMockAPIUnmarshalErr.EXPECT().DecodeSessionKeys([]byte{0x4, 0x1}).Return([]byte{0x4, 0x1}, nil) - coreMockAPIErr := mocks.NewCoreAPI(t) - coreMockAPIErr.On("DecodeSessionKeys", pkeys).Return(data, nil) - coreMockAPIErr.On("HasKey", mock.AnythingOfType("string"), - mock.AnythingOfType("string")). + coreMockAPIOk := mocks.NewMockCoreAPI(ctrl) + coreMockAPIOk.EXPECT().DecodeSessionKeys(pkeys).Return(data, nil) + coreMockAPIOk.EXPECT().HasKey(gomock.Any(), gomock.Any()). + Return(true, nil).AnyTimes() + + coreMockAPIErr := mocks.NewMockCoreAPI(ctrl) + coreMockAPIErr.EXPECT().DecodeSessionKeys(pkeys).Return(data, nil) + coreMockAPIErr.EXPECT().HasKey(gomock.Any(), + gomock.Any()). Return(false, errors.New("HasKey err")) - coreMockAPIInvalidDec := mocks.NewCoreAPI(t) - coreMockAPIInvalidDec.On("DecodeSessionKeys", pkeys).Return([]byte{0x0}, nil) + coreMockAPIInvalidDec := mocks.NewMockCoreAPI(ctrl) + coreMockAPIInvalidDec.EXPECT().DecodeSessionKeys(pkeys).Return([]byte{0x0}, nil) type fields struct { authorModule *AuthorModule @@ -148,12 +151,15 @@ func TestAuthorModule_SubmitExtrinsic(t *testing.T) { 201, 74, 231, 222, 101, 85, 108, 102, 39, 31, 190, 210, 14, 215, 124, 19, 160, 180, 203, 54, 110, 167, 163, 149, 45, 12, 108, 80, 221, 65, 238, 57, 237, 199, 16, 10, 33, 185, 8, 244, 184, 243, 139, 5, 87, 252, 245, 24, 225, 37, 154, 163, 143} - errMockCoreAPI := mocks.NewCoreAPI(t) - errMockCoreAPI.On("HandleSubmittedExtrinsic", + + ctrl := gomock.NewController(t) + + errMockCoreAPI := mocks.NewMockCoreAPI(ctrl) + errMockCoreAPI.EXPECT().HandleSubmittedExtrinsic( types.Extrinsic(common.MustHexToBytes(fmt.Sprintf("0x%x", testInvalidExt)))).Return(fmt.Errorf("some error")) - mockCoreAPI := mocks.NewCoreAPI(t) - mockCoreAPI.On("HandleSubmittedExtrinsic", + mockCoreAPI := mocks.NewMockCoreAPI(ctrl) + mockCoreAPI.EXPECT().HandleSubmittedExtrinsic( types.Extrinsic(common.MustHexToBytes(fmt.Sprintf("0x%x", testExt)))).Return(nil) type fields struct { logger Infoer @@ -223,11 +229,13 @@ func TestAuthorModule_SubmitExtrinsic(t *testing.T) { } func TestAuthorModule_PendingExtrinsics(t *testing.T) { - emptyMockTransactionStateAPI := mocks.NewTransactionStateAPI(t) - emptyMockTransactionStateAPI.On("Pending").Return([]*transaction.ValidTransaction{}) + ctrl := gomock.NewController(t) - mockTransactionStateAPI := mocks.NewTransactionStateAPI(t) - mockTransactionStateAPI.On("Pending").Return([]*transaction.ValidTransaction{ + emptyMockTransactionStateAPI := mocks.NewMockTransactionStateAPI(ctrl) + emptyMockTransactionStateAPI.EXPECT().Pending().Return([]*transaction.ValidTransaction{}) + + mockTransactionStateAPI := mocks.NewMockTransactionStateAPI(ctrl) + mockTransactionStateAPI.EXPECT().Pending().Return([]*transaction.ValidTransaction{ { Extrinsic: types.NewExtrinsic([]byte("someExtrinsic")), }, @@ -308,11 +316,13 @@ func TestAuthorModule_InsertKey(t *testing.T) { require.NoError(t, err) _ = kp3.Public().Hex() - mockCoreAPIHappyBabe := mocks.NewCoreAPI(t) - mockCoreAPIHappyBabe.On("InsertKey", kp1, "babe").Return(nil) + ctrl := gomock.NewController(t) + + mockCoreAPIHappyBabe := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIHappyBabe.EXPECT().InsertKey(kp1, "babe").Return(nil) - mockCoreAPIHappyGran := mocks.NewCoreAPI(t) - mockCoreAPIHappyGran.On("InsertKey", kp2, "gran").Return(nil) + mockCoreAPIHappyGran := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIHappyGran.EXPECT().InsertKey(kp2, "gran").Return(nil) type fields struct { logger Infoer @@ -407,14 +417,16 @@ func TestAuthorModule_HasKey(t *testing.T) { kr, err := keystore.NewSr25519Keyring() require.NoError(t, err) - mockCoreAPITrue := mocks.NewCoreAPI(t) - mockCoreAPITrue.On("HasKey", kr.Alice().Public().Hex(), "babe").Return(true, nil) + ctrl := gomock.NewController(t) + + mockCoreAPITrue := mocks.NewMockCoreAPI(ctrl) + mockCoreAPITrue.EXPECT().HasKey(kr.Alice().Public().Hex(), "babe").Return(true, nil) - mockCoreAPIFalse := mocks.NewCoreAPI(t) - mockCoreAPIFalse.On("HasKey", kr.Alice().Public().Hex(), "babe").Return(false, nil) + mockCoreAPIFalse := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIFalse.EXPECT().HasKey(kr.Alice().Public().Hex(), "babe").Return(false, nil) - mockCoreAPIErr := mocks.NewCoreAPI(t) - mockCoreAPIErr.On("HasKey", kr.Alice().Public().Hex(), "babe").Return(false, fmt.Errorf("some error")) + mockCoreAPIErr := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIErr.EXPECT().HasKey(kr.Alice().Public().Hex(), "babe").Return(false, fmt.Errorf("some error")) type fields struct { logger Infoer diff --git a/dot/rpc/modules/chain_test.go b/dot/rpc/modules/chain_test.go index 39af727b10..72584b977f 100644 --- a/dot/rpc/modules/chain_test.go +++ b/dot/rpc/modules/chain_test.go @@ -11,6 +11,7 @@ import ( "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" "github.com/ChainSafe/gossamer/dot/types" "github.com/ChainSafe/gossamer/lib/common" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" @@ -21,17 +22,19 @@ func TestChainModule_GetBlock(t *testing.T) { inputHash := common.MustHexToHash("0x0102000000000000000000000000000000000000000000000000000000000000") emptyBlock := types.NewEmptyBlock() - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockAPI.On("GetBlockByHash", inputHash).Return(&emptyBlock, nil) - mockBlockAPI.On("BestBlockHash").Return(testHash, nil) + ctrl := gomock.NewController(t) - mockBlockAPIGetHashErr := mocks.NewBlockAPI(t) - mockBlockAPIGetHashErr.On("GetBlockByHash", inputHash).Return(nil, errors.New("GetJustification error")) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockAPI.EXPECT().GetBlockByHash(inputHash).Return(&emptyBlock, nil) + mockBlockAPI.EXPECT().BestBlockHash().Return(testHash) + + mockBlockAPIGetHashErr := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIGetHashErr.EXPECT().GetBlockByHash(inputHash).Return(nil, errors.New("GetJustification error")) bodyBlock := types.NewEmptyBlock() bodyBlock.Body = types.BytesArrayToExtrinsics([][]byte{{1}}) - mockBlockAPIWithBody := mocks.NewBlockAPI(t) - mockBlockAPIWithBody.On("GetBlockByHash", inputHash).Return(&bodyBlock, nil) + mockBlockAPIWithBody := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIWithBody.EXPECT().GetBlockByHash(inputHash).Return(&bodyBlock, nil) chainModule := NewChainModule(mockBlockAPI) type fields struct { @@ -115,15 +118,19 @@ func TestChainModule_GetBlock(t *testing.T) { } func TestChainModule_GetBlockHash(t *testing.T) { + ctrl := gomock.NewController(t) + testHash := common.NewHash([]byte{0x01, 0x02}) i := []interface{}{"a"} - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockAPI.On("BestBlockHash").Return(testHash, nil) - mockBlockAPI.On("GetHashByNumber", uint(21)).Return(testHash, nil) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockAPI.EXPECT().BestBlockHash().Return(testHash) + mockBlockAPI.EXPECT().GetHashByNumber(uint(21)). + Return(testHash, nil).AnyTimes() - mockBlockAPIErr := mocks.NewBlockAPI(t) - mockBlockAPIErr.On("GetHashByNumber", uint(21)).Return(nil, errors.New("GetBlockHash Error")) + mockBlockAPIErr := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIErr.EXPECT().GetHashByNumber(uint(21)). + Return(common.Hash{}, errors.New("GetBlockHash Error")) expRes := ChainHashResponse(testHash.String()) type fields struct { @@ -222,12 +229,15 @@ func TestChainModule_GetBlockHash(t *testing.T) { } func TestChainModule_GetFinalizedHead(t *testing.T) { + ctrl := gomock.NewController(t) + testHash := common.NewHash([]byte{0x01, 0x02}) - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockAPI.On("GetHighestFinalisedHash").Return(testHash, nil) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockAPI.EXPECT().GetHighestFinalisedHash().Return(testHash, nil) - mockBlockAPIErr := mocks.NewBlockAPI(t) - mockBlockAPIErr.On("GetHighestFinalisedHash").Return(nil, errors.New("GetHighestFinalisedHash Error")) + mockBlockAPIErr := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIErr.EXPECT().GetHighestFinalisedHash(). + Return(common.Hash{}, errors.New("GetHighestFinalisedHash Error")) expRes := ChainHashResponse(common.BytesToHex(testHash[:])) type fields struct { @@ -283,12 +293,15 @@ func TestChainModule_GetFinalizedHead(t *testing.T) { } func TestChainModule_GetFinalizedHeadByRound(t *testing.T) { + ctrl := gomock.NewController(t) + testHash := common.NewHash([]byte{0x01, 0x02}) - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockAPI.On("GetFinalisedHash", uint64(21), uint64(21)).Return(testHash, nil) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockAPI.EXPECT().GetFinalisedHash(uint64(21), uint64(21)).Return(testHash, nil) - mockBlockAPIErr := mocks.NewBlockAPI(t) - mockBlockAPIErr.On("GetFinalisedHash", uint64(21), uint64(21)).Return(nil, errors.New("GetFinalisedHash Error")) + mockBlockAPIErr := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIErr.EXPECT().GetFinalisedHash(uint64(21), uint64(21)). + Return(common.Hash{}, errors.New("GetFinalisedHash Error")) expRes := ChainHashResponse(common.BytesToHex(testHash[:])) type fields struct { @@ -350,16 +363,18 @@ func TestChainModule_GetFinalizedHeadByRound(t *testing.T) { } func TestChainModule_GetHeader(t *testing.T) { + ctrl := gomock.NewController(t) + emptyHeader := types.NewEmptyHeader() testHash := common.NewHash([]byte{0x01, 0x02}) inputHash, err := common.HexToHash("0x0102000000000000000000000000000000000000000000000000000000000000") require.NoError(t, err) - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockAPI.On("GetHeader", inputHash).Return(emptyHeader, nil) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockAPI.EXPECT().GetHeader(inputHash).Return(emptyHeader, nil) - mockBlockAPIErr := mocks.NewBlockAPI(t) - mockBlockAPIErr.On("GetHeader", inputHash).Return(nil, errors.New("GetFinalisedHash Error")) + mockBlockAPIErr := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIErr.EXPECT().GetHeader(inputHash).Return(nil, errors.New("GetFinalisedHash Error")) expRes, err := HeaderToJSON(*emptyHeader) require.NoError(t, err) @@ -417,9 +432,11 @@ func TestChainModule_GetHeader(t *testing.T) { } func TestChainModule_ErrSubscriptionTransport(t *testing.T) { + ctrl := gomock.NewController(t) + req := &EmptyRequest{} res := &ChainBlockHeaderResponse{} - cm := NewChainModule(mocks.NewBlockAPI(t)) + cm := NewChainModule(mocks.NewMockBlockAPI(ctrl)) err := cm.SubscribeFinalizedHeads(nil, req, res) require.ErrorIs(t, err, ErrSubscriptionTransport) diff --git a/dot/rpc/modules/childstate_test.go b/dot/rpc/modules/childstate_test.go index 67606e5ac9..8bcb7aba45 100644 --- a/dot/rpc/modules/childstate_test.go +++ b/dot/rpc/modules/childstate_test.go @@ -12,6 +12,7 @@ import ( "github.com/ChainSafe/gossamer/lib/common" rtstorage "github.com/ChainSafe/gossamer/lib/runtime/storage" "github.com/ChainSafe/gossamer/lib/trie" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" @@ -41,6 +42,8 @@ func createTestTrieState(t *testing.T) (*trie.Trie, common.Hash) { } func TestChildStateModule_GetKeys(t *testing.T) { + ctrl := gomock.NewController(t) + tr, sr := createTestTrieState(t) expKeys := tr.GetKeysWithPrefix([]byte{}) @@ -49,22 +52,23 @@ func TestChildStateModule_GetKeys(t *testing.T) { expHexKeys[idx] = common.BytesToHex(k) } - mockStorageAPI := apimocks.NewStorageAPI(t) - mockErrorStorageAPI1 := apimocks.NewStorageAPI(t) - mockErrorStorageAPI2 := apimocks.NewStorageAPI(t) - mockBlockAPI := apimocks.NewBlockAPI(t) + mockStorageAPI := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI1 := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI2 := apimocks.NewMockStorageAPI(ctrl) + mockBlockAPI := apimocks.NewMockBlockAPI(ctrl) hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockBlockAPI.On("BestBlockHash").Return(hash) + mockBlockAPI.EXPECT().BestBlockHash().Return(hash).AnyTimes() - mockStorageAPI.On("GetStateRootFromBlock", &hash).Return(&sr, nil) - mockStorageAPI.On("GetStorageChild", &sr, []byte(":child_storage_key")).Return(tr, nil) + mockStorageAPI.EXPECT().GetStateRootFromBlock(&hash).Return(&sr, nil).AnyTimes() + mockStorageAPI.EXPECT().GetStorageChild(&sr, []byte(":child_storage_key")). + Return(tr, nil).AnyTimes() - mockErrorStorageAPI1.On("GetStateRootFromBlock", &common.Hash{}).Return(nil, nil) - mockErrorStorageAPI1.On("GetStorageChild", (*common.Hash)(nil), []byte(nil)). + mockErrorStorageAPI1.EXPECT().GetStateRootFromBlock(&common.Hash{}).Return(nil, nil) + mockErrorStorageAPI1.EXPECT().GetStorageChild((*common.Hash)(nil), []byte(nil)). Return(nil, errors.New("GetStorageChild error")) - mockErrorStorageAPI2.On("GetStateRootFromBlock", &hash).Return(nil, errors.New("GetStateRootFromBlock error")) + mockErrorStorageAPI2.EXPECT().GetStateRootFromBlock(&hash).Return(nil, errors.New("GetStateRootFromBlock error")) childStateModule := NewChildStateModule(mockStorageAPI, mockBlockAPI) type fields struct { @@ -157,25 +161,27 @@ func TestChildStateModule_GetKeys(t *testing.T) { } func TestChildStateModule_GetStorageSize(t *testing.T) { + ctrl := gomock.NewController(t) + _, sr := createTestTrieState(t) - mockStorageAPI := apimocks.NewStorageAPI(t) - mockErrorStorageAPI1 := apimocks.NewStorageAPI(t) - mockErrorStorageAPI2 := apimocks.NewStorageAPI(t) - mockBlockAPI := apimocks.NewBlockAPI(t) + mockStorageAPI := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI1 := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI2 := apimocks.NewMockStorageAPI(ctrl) + mockBlockAPI := apimocks.NewMockBlockAPI(ctrl) hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockBlockAPI.On("BestBlockHash").Return(hash) + mockBlockAPI.EXPECT().BestBlockHash().Return(hash) - mockStorageAPI.On("GetStateRootFromBlock", &hash).Return(&sr, nil) - mockStorageAPI.On("GetStorageFromChild", &sr, []byte(":child_storage_key"), []byte(":child_first")). - Return([]byte(""), nil) + mockStorageAPI.EXPECT().GetStateRootFromBlock(&hash).Return(&sr, nil).AnyTimes() + mockStorageAPI.EXPECT().GetStorageFromChild(&sr, []byte(":child_storage_key"), []byte(":child_first")). + Return([]byte(""), nil).AnyTimes() - mockErrorStorageAPI1.On("GetStateRootFromBlock", &hash).Return(nil, nil) - mockErrorStorageAPI1.On("GetStorageFromChild", (*common.Hash)(nil), []byte(nil), []byte(nil)). + mockErrorStorageAPI1.EXPECT().GetStateRootFromBlock(&hash).Return(nil, nil) + mockErrorStorageAPI1.EXPECT().GetStorageFromChild((*common.Hash)(nil), []byte(nil), []byte(nil)). Return(nil, errors.New("GetStorageChild error")) - mockErrorStorageAPI2.On("GetStateRootFromBlock", &hash).Return(nil, errors.New("GetStateRootFromBlock error")) + mockErrorStorageAPI2.EXPECT().GetStateRootFromBlock(&hash).Return(nil, errors.New("GetStateRootFromBlock error")) childStateModule := NewChildStateModule(mockStorageAPI, mockBlockAPI) type fields struct { @@ -268,25 +274,27 @@ func TestChildStateModule_GetStorageSize(t *testing.T) { } func TestChildStateModule_GetStorageHash(t *testing.T) { + ctrl := gomock.NewController(t) + _, sr := createTestTrieState(t) - mockStorageAPI := apimocks.NewStorageAPI(t) - mockErrorStorageAPI1 := apimocks.NewStorageAPI(t) - mockErrorStorageAPI2 := apimocks.NewStorageAPI(t) - mockBlockAPI := apimocks.NewBlockAPI(t) + mockStorageAPI := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI1 := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI2 := apimocks.NewMockStorageAPI(ctrl) + mockBlockAPI := apimocks.NewMockBlockAPI(ctrl) hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockBlockAPI.On("BestBlockHash").Return(hash) + mockBlockAPI.EXPECT().BestBlockHash().Return(hash) - mockStorageAPI.On("GetStateRootFromBlock", &hash).Return(&sr, nil) - mockStorageAPI.On("GetStorageFromChild", &sr, []byte(":child_storage_key"), []byte(":child_first")). - Return([]byte(""), nil) + mockStorageAPI.EXPECT().GetStateRootFromBlock(&hash).Return(&sr, nil).AnyTimes() + mockStorageAPI.EXPECT().GetStorageFromChild(&sr, []byte(":child_storage_key"), []byte(":child_first")). + Return([]byte(""), nil).AnyTimes() - mockErrorStorageAPI1.On("GetStateRootFromBlock", &hash).Return(nil, nil) - mockErrorStorageAPI1.On("GetStorageFromChild", (*common.Hash)(nil), []byte(nil), []byte(nil)). + mockErrorStorageAPI1.EXPECT().GetStateRootFromBlock(&hash).Return(nil, nil) + mockErrorStorageAPI1.EXPECT().GetStorageFromChild((*common.Hash)(nil), []byte(nil), []byte(nil)). Return(nil, errors.New("GetStorageChild error")) - mockErrorStorageAPI2.On("GetStateRootFromBlock", &hash).Return(nil, errors.New("GetStateRootFromBlock error")) + mockErrorStorageAPI2.EXPECT().GetStateRootFromBlock(&hash).Return(nil, errors.New("GetStateRootFromBlock error")) childStateModule := NewChildStateModule(mockStorageAPI, mockBlockAPI) type fields struct { @@ -379,25 +387,27 @@ func TestChildStateModule_GetStorageHash(t *testing.T) { } func TestChildStateModule_GetStorage(t *testing.T) { + ctrl := gomock.NewController(t) + _, sr := createTestTrieState(t) - mockStorageAPI := apimocks.NewStorageAPI(t) - mockErrorStorageAPI1 := apimocks.NewStorageAPI(t) - mockErrorStorageAPI2 := apimocks.NewStorageAPI(t) - mockBlockAPI := apimocks.NewBlockAPI(t) + mockStorageAPI := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI1 := apimocks.NewMockStorageAPI(ctrl) + mockErrorStorageAPI2 := apimocks.NewMockStorageAPI(ctrl) + mockBlockAPI := apimocks.NewMockBlockAPI(ctrl) hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockBlockAPI.On("BestBlockHash").Return(hash) + mockBlockAPI.EXPECT().BestBlockHash().Return(hash) - mockStorageAPI.On("GetStateRootFromBlock", &hash).Return(&sr, nil) - mockStorageAPI.On("GetStorageFromChild", &sr, []byte(":child_storage_key"), []byte(":child_first")). - Return([]byte("test"), nil) + mockStorageAPI.EXPECT().GetStateRootFromBlock(&hash).Return(&sr, nil).AnyTimes() + mockStorageAPI.EXPECT().GetStorageFromChild(&sr, []byte(":child_storage_key"), []byte(":child_first")). + Return([]byte("test"), nil).AnyTimes() - mockErrorStorageAPI1.On("GetStateRootFromBlock", &hash).Return(nil, nil) - mockErrorStorageAPI1.On("GetStorageFromChild", (*common.Hash)(nil), []byte(nil), []byte(nil)). + mockErrorStorageAPI1.EXPECT().GetStateRootFromBlock(&hash).Return(nil, nil) + mockErrorStorageAPI1.EXPECT().GetStorageFromChild((*common.Hash)(nil), []byte(nil), []byte(nil)). Return(nil, errors.New("GetStorageChild error")) - mockErrorStorageAPI2.On("GetStateRootFromBlock", &hash).Return(nil, errors.New("GetStateRootFromBlock error")) + mockErrorStorageAPI2.EXPECT().GetStateRootFromBlock(&hash).Return(nil, errors.New("GetStateRootFromBlock error")) childStateModule := NewChildStateModule(mockStorageAPI, mockBlockAPI) type fields struct { diff --git a/dot/rpc/modules/dev_integration_test.go b/dot/rpc/modules/dev_integration_test.go index a267fa715e..eeebd1ef63 100644 --- a/dot/rpc/modules/dev_integration_test.go +++ b/dot/rpc/modules/dev_integration_test.go @@ -12,7 +12,6 @@ import ( "github.com/ChainSafe/gossamer/dot/state" "github.com/ChainSafe/gossamer/dot/types" "github.com/ChainSafe/gossamer/lib/babe" - babemocks "github.com/ChainSafe/gossamer/lib/babe/mocks" "github.com/ChainSafe/gossamer/lib/common" "github.com/ChainSafe/gossamer/lib/crypto/sr25519" "github.com/ChainSafe/gossamer/lib/keystore" @@ -51,6 +50,8 @@ func newState(t *testing.T) (*state.BlockState, *state.EpochState) { } func newBABEService(t *testing.T) *babe.Service { + ctrl := gomock.NewController(t) + kr, err := keystore.NewSr25519Keyring() require.NoError(t, err) @@ -74,7 +75,7 @@ func newBABEService(t *testing.T) *babe.Service { EpochState: es, Keypair: kr.Alice().(*sr25519.Keypair), IsDev: true, - BlockImportHandler: babemocks.NewBlockImportHandler(t), + BlockImportHandler: NewMockBlockImportHandler(ctrl), } babe, err := babe.NewService(cfg) diff --git a/dot/rpc/modules/dev_test.go b/dot/rpc/modules/dev_test.go index 681efb03b3..ede9a16975 100644 --- a/dot/rpc/modules/dev_test.go +++ b/dot/rpc/modules/dev_test.go @@ -9,6 +9,7 @@ import ( "testing" "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" ) @@ -53,8 +54,10 @@ func Test_uint64ToHex(t *testing.T) { } func TestDevModule_EpochLength(t *testing.T) { - mockBlockProducerAPI := mocks.NewBlockProducerAPI(t) - mockBlockProducerAPI.On("EpochLength").Return(uint64(23)) + ctrl := gomock.NewController(t) + + mockBlockProducerAPI := mocks.NewMockBlockProducerAPI(ctrl) + mockBlockProducerAPI.EXPECT().EpochLength().Return(uint64(23)) devModule := NewDevModule(mockBlockProducerAPI, nil) type fields struct { @@ -103,8 +106,10 @@ func TestDevModule_EpochLength(t *testing.T) { } func TestDevModule_SlotDuration(t *testing.T) { - mockBlockProducerAPI := mocks.NewBlockProducerAPI(t) - mockBlockProducerAPI.On("SlotDuration").Return(uint64(23)) + ctrl := gomock.NewController(t) + + mockBlockProducerAPI := mocks.NewMockBlockProducerAPI(ctrl) + mockBlockProducerAPI.EXPECT().SlotDuration().Return(uint64(23)) type fields struct { networkAPI NetworkAPI @@ -152,22 +157,24 @@ func TestDevModule_SlotDuration(t *testing.T) { } func TestDevModule_Control(t *testing.T) { - mockBlockProducerAPI := mocks.NewBlockProducerAPI(t) - mockErrorBlockProducerAPI := mocks.NewBlockProducerAPI(t) - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockErrorNetworkAPI := mocks.NewNetworkAPI(t) + ctrl := gomock.NewController(t) + + mockBlockProducerAPI := mocks.NewMockBlockProducerAPI(ctrl) + mockErrorBlockProducerAPI := mocks.NewMockBlockProducerAPI(ctrl) + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockErrorNetworkAPI := mocks.NewMockNetworkAPI(ctrl) - mockErrorBlockProducerAPI.On("Pause").Return(errors.New("babe pause error")) - mockBlockProducerAPI.On("Pause").Return(nil) + mockErrorBlockProducerAPI.EXPECT().Pause().Return(errors.New("babe pause error")) + mockBlockProducerAPI.EXPECT().Pause().Return(nil) - mockErrorBlockProducerAPI.On("Resume").Return(errors.New("babe resume error")) - mockBlockProducerAPI.On("Resume").Return(nil) + mockErrorBlockProducerAPI.EXPECT().Resume().Return(errors.New("babe resume error")) + mockBlockProducerAPI.EXPECT().Resume().Return(nil) - mockErrorNetworkAPI.On("Stop").Return(errors.New("network stop error")) - mockNetworkAPI.On("Stop").Return(nil) + mockErrorNetworkAPI.EXPECT().Stop().Return(errors.New("network stop error")) + mockNetworkAPI.EXPECT().Stop().Return(nil) - mockErrorNetworkAPI.On("Start").Return(errors.New("network start error")) - mockNetworkAPI.On("Start").Return(nil) + mockErrorNetworkAPI.EXPECT().Start().Return(errors.New("network start error")) + mockNetworkAPI.EXPECT().Start().Return(nil) type fields struct { networkAPI NetworkAPI diff --git a/dot/rpc/modules/grandpa_integration_test.go b/dot/rpc/modules/grandpa_integration_test.go index 6f96382414..04e91b6137 100644 --- a/dot/rpc/modules/grandpa_integration_test.go +++ b/dot/rpc/modules/grandpa_integration_test.go @@ -13,6 +13,7 @@ import ( "github.com/ChainSafe/gossamer/lib/crypto/ed25519" "github.com/ChainSafe/gossamer/lib/grandpa" "github.com/ChainSafe/gossamer/lib/keystore" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" @@ -51,6 +52,8 @@ func TestGrandpaProveFinality(t *testing.T) { } func TestRoundState(t *testing.T) { + ctrl := gomock.NewController(t) + var voters grandpa.Voters for _, k := range kr.Keys { @@ -60,19 +63,19 @@ func TestRoundState(t *testing.T) { }) } - grandpamock := rpcmocks.NewBlockFinalityAPI(t) - grandpamock.On("GetVoters").Return(voters) - grandpamock.On("GetSetID").Return(uint64(0)) - grandpamock.On("GetRound").Return(uint64(2)) + grandpamock := rpcmocks.NewMockBlockFinalityAPI(ctrl) + grandpamock.EXPECT().GetVoters().Return(voters) + grandpamock.EXPECT().GetSetID().Return(uint64(0)) + grandpamock.EXPECT().GetRound().Return(uint64(2)) - grandpamock.On("PreVotes").Return([]ed25519.PublicKeyBytes{ + grandpamock.EXPECT().PreVotes().Return([]ed25519.PublicKeyBytes{ kr.Alice().Public().(*ed25519.PublicKey).AsBytes(), kr.Bob().Public().(*ed25519.PublicKey).AsBytes(), kr.Charlie().Public().(*ed25519.PublicKey).AsBytes(), kr.Dave().Public().(*ed25519.PublicKey).AsBytes(), }) - grandpamock.On("PreCommits").Return([]ed25519.PublicKeyBytes{ + grandpamock.EXPECT().PreCommits().Return([]ed25519.PublicKeyBytes{ kr.Alice().Public().(*ed25519.PublicKey).AsBytes(), kr.Bob().Public().(*ed25519.PublicKey).AsBytes(), }) diff --git a/dot/rpc/modules/grandpa_test.go b/dot/rpc/modules/grandpa_test.go index d94a9d07b5..7c46af87c7 100644 --- a/dot/rpc/modules/grandpa_test.go +++ b/dot/rpc/modules/grandpa_test.go @@ -112,6 +112,8 @@ func TestGrandpaModule_ProveFinality(t *testing.T) { } func TestGrandpaModule_RoundState(t *testing.T) { + ctrl := gomock.NewController(t) + var kr, _ = keystore.NewEd25519Keyring() var voters grandpa.Voters @@ -122,18 +124,18 @@ func TestGrandpaModule_RoundState(t *testing.T) { }) } - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockFinalityAPI := mocks.NewBlockFinalityAPI(t) - mockBlockFinalityAPI.On("GetVoters").Return(voters) - mockBlockFinalityAPI.On("GetSetID").Return(uint64(0)) - mockBlockFinalityAPI.On("GetRound").Return(uint64(2)) - mockBlockFinalityAPI.On("PreVotes").Return([]ed25519.PublicKeyBytes{ + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockFinalityAPI := mocks.NewMockBlockFinalityAPI(ctrl) + mockBlockFinalityAPI.EXPECT().GetVoters().Return(voters) + mockBlockFinalityAPI.EXPECT().GetSetID().Return(uint64(0)) + mockBlockFinalityAPI.EXPECT().GetRound().Return(uint64(2)) + mockBlockFinalityAPI.EXPECT().PreVotes().Return([]ed25519.PublicKeyBytes{ kr.Alice().Public().(*ed25519.PublicKey).AsBytes(), kr.Bob().Public().(*ed25519.PublicKey).AsBytes(), kr.Charlie().Public().(*ed25519.PublicKey).AsBytes(), kr.Dave().Public().(*ed25519.PublicKey).AsBytes(), }) - mockBlockFinalityAPI.On("PreCommits").Return([]ed25519.PublicKeyBytes{ + mockBlockFinalityAPI.EXPECT().PreCommits().Return([]ed25519.PublicKeyBytes{ kr.Alice().Public().(*ed25519.PublicKey).AsBytes(), kr.Bob().Public().(*ed25519.PublicKey).AsBytes(), }) diff --git a/dot/rpc/modules/mocks/block_api.go b/dot/rpc/modules/mocks/block_api.go deleted file mode 100644 index c5730809a5..0000000000 --- a/dot/rpc/modules/mocks/block_api.go +++ /dev/null @@ -1,332 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - common "github.com/ChainSafe/gossamer/lib/common" - mock "github.com/stretchr/testify/mock" - - runtime "github.com/ChainSafe/gossamer/lib/runtime" - - state "github.com/ChainSafe/gossamer/dot/state" - - types "github.com/ChainSafe/gossamer/dot/types" -) - -// BlockAPI is an autogenerated mock type for the BlockAPI type -type BlockAPI struct { - mock.Mock -} - -// BestBlockHash provides a mock function with given fields: -func (_m *BlockAPI) BestBlockHash() common.Hash { - ret := _m.Called() - - var r0 common.Hash - if rf, ok := ret.Get(0).(func() common.Hash); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - return r0 -} - -// FreeFinalisedNotifierChannel provides a mock function with given fields: ch -func (_m *BlockAPI) FreeFinalisedNotifierChannel(ch chan *types.FinalisationInfo) { - _m.Called(ch) -} - -// FreeImportedBlockNotifierChannel provides a mock function with given fields: ch -func (_m *BlockAPI) FreeImportedBlockNotifierChannel(ch chan *types.Block) { - _m.Called(ch) -} - -// GetBlockByHash provides a mock function with given fields: hash -func (_m *BlockAPI) GetBlockByHash(hash common.Hash) (*types.Block, error) { - ret := _m.Called(hash) - - var r0 *types.Block - if rf, ok := ret.Get(0).(func(common.Hash) *types.Block); ok { - r0 = rf(hash) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*types.Block) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(common.Hash) error); ok { - r1 = rf(hash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetFinalisedHash provides a mock function with given fields: _a0, _a1 -func (_m *BlockAPI) GetFinalisedHash(_a0 uint64, _a1 uint64) (common.Hash, error) { - ret := _m.Called(_a0, _a1) - - var r0 common.Hash - if rf, ok := ret.Get(0).(func(uint64, uint64) common.Hash); ok { - r0 = rf(_a0, _a1) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(uint64, uint64) error); ok { - r1 = rf(_a0, _a1) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetFinalisedNotifierChannel provides a mock function with given fields: -func (_m *BlockAPI) GetFinalisedNotifierChannel() chan *types.FinalisationInfo { - ret := _m.Called() - - var r0 chan *types.FinalisationInfo - if rf, ok := ret.Get(0).(func() chan *types.FinalisationInfo); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(chan *types.FinalisationInfo) - } - } - - return r0 -} - -// GetHashByNumber provides a mock function with given fields: blockNumber -func (_m *BlockAPI) GetHashByNumber(blockNumber uint) (common.Hash, error) { - ret := _m.Called(blockNumber) - - var r0 common.Hash - if rf, ok := ret.Get(0).(func(uint) common.Hash); ok { - r0 = rf(blockNumber) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(uint) error); ok { - r1 = rf(blockNumber) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetHeader provides a mock function with given fields: hash -func (_m *BlockAPI) GetHeader(hash common.Hash) (*types.Header, error) { - ret := _m.Called(hash) - - var r0 *types.Header - if rf, ok := ret.Get(0).(func(common.Hash) *types.Header); ok { - r0 = rf(hash) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*types.Header) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(common.Hash) error); ok { - r1 = rf(hash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetHighestFinalisedHash provides a mock function with given fields: -func (_m *BlockAPI) GetHighestFinalisedHash() (common.Hash, error) { - ret := _m.Called() - - var r0 common.Hash - if rf, ok := ret.Get(0).(func() common.Hash); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetImportedBlockNotifierChannel provides a mock function with given fields: -func (_m *BlockAPI) GetImportedBlockNotifierChannel() chan *types.Block { - ret := _m.Called() - - var r0 chan *types.Block - if rf, ok := ret.Get(0).(func() chan *types.Block); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(chan *types.Block) - } - } - - return r0 -} - -// GetJustification provides a mock function with given fields: hash -func (_m *BlockAPI) GetJustification(hash common.Hash) ([]byte, error) { - ret := _m.Called(hash) - - var r0 []byte - if rf, ok := ret.Get(0).(func(common.Hash) []byte); ok { - r0 = rf(hash) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(common.Hash) error); ok { - r1 = rf(hash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetRuntime provides a mock function with given fields: blockHash -func (_m *BlockAPI) GetRuntime(blockHash common.Hash) (state.Runtime, error) { - ret := _m.Called(blockHash) - - var r0 state.Runtime - if rf, ok := ret.Get(0).(func(common.Hash) state.Runtime); ok { - r0 = rf(blockHash) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(state.Runtime) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(common.Hash) error); ok { - r1 = rf(blockHash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// HasJustification provides a mock function with given fields: hash -func (_m *BlockAPI) HasJustification(hash common.Hash) (bool, error) { - ret := _m.Called(hash) - - var r0 bool - if rf, ok := ret.Get(0).(func(common.Hash) bool); ok { - r0 = rf(hash) - } else { - r0 = ret.Get(0).(bool) - } - - var r1 error - if rf, ok := ret.Get(1).(func(common.Hash) error); ok { - r1 = rf(hash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RangeInMemory provides a mock function with given fields: start, end -func (_m *BlockAPI) RangeInMemory(start common.Hash, end common.Hash) ([]common.Hash, error) { - ret := _m.Called(start, end) - - var r0 []common.Hash - if rf, ok := ret.Get(0).(func(common.Hash, common.Hash) []common.Hash); ok { - r0 = rf(start, end) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]common.Hash) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(common.Hash, common.Hash) error); ok { - r1 = rf(start, end) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RegisterRuntimeUpdatedChannel provides a mock function with given fields: ch -func (_m *BlockAPI) RegisterRuntimeUpdatedChannel(ch chan<- runtime.Version) (uint32, error) { - ret := _m.Called(ch) - - var r0 uint32 - if rf, ok := ret.Get(0).(func(chan<- runtime.Version) uint32); ok { - r0 = rf(ch) - } else { - r0 = ret.Get(0).(uint32) - } - - var r1 error - if rf, ok := ret.Get(1).(func(chan<- runtime.Version) error); ok { - r1 = rf(ch) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// UnregisterRuntimeUpdatedChannel provides a mock function with given fields: id -func (_m *BlockAPI) UnregisterRuntimeUpdatedChannel(id uint32) bool { - ret := _m.Called(id) - - var r0 bool - if rf, ok := ret.Get(0).(func(uint32) bool); ok { - r0 = rf(id) - } else { - r0 = ret.Get(0).(bool) - } - - return r0 -} - -type mockConstructorTestingTNewBlockAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewBlockAPI creates a new instance of BlockAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewBlockAPI(t mockConstructorTestingTNewBlockAPI) *BlockAPI { - mock := &BlockAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/block_finality_api.go b/dot/rpc/modules/mocks/block_finality_api.go deleted file mode 100644 index d6af6f9ffd..0000000000 --- a/dot/rpc/modules/mocks/block_finality_api.go +++ /dev/null @@ -1,106 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - ed25519 "github.com/ChainSafe/gossamer/lib/crypto/ed25519" - mock "github.com/stretchr/testify/mock" - - types "github.com/ChainSafe/gossamer/dot/types" -) - -// BlockFinalityAPI is an autogenerated mock type for the BlockFinalityAPI type -type BlockFinalityAPI struct { - mock.Mock -} - -// GetRound provides a mock function with given fields: -func (_m *BlockFinalityAPI) GetRound() uint64 { - ret := _m.Called() - - var r0 uint64 - if rf, ok := ret.Get(0).(func() uint64); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(uint64) - } - - return r0 -} - -// GetSetID provides a mock function with given fields: -func (_m *BlockFinalityAPI) GetSetID() uint64 { - ret := _m.Called() - - var r0 uint64 - if rf, ok := ret.Get(0).(func() uint64); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(uint64) - } - - return r0 -} - -// GetVoters provides a mock function with given fields: -func (_m *BlockFinalityAPI) GetVoters() types.GrandpaVoters { - ret := _m.Called() - - var r0 types.GrandpaVoters - if rf, ok := ret.Get(0).(func() types.GrandpaVoters); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(types.GrandpaVoters) - } - } - - return r0 -} - -// PreCommits provides a mock function with given fields: -func (_m *BlockFinalityAPI) PreCommits() []ed25519.PublicKeyBytes { - ret := _m.Called() - - var r0 []ed25519.PublicKeyBytes - if rf, ok := ret.Get(0).(func() []ed25519.PublicKeyBytes); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]ed25519.PublicKeyBytes) - } - } - - return r0 -} - -// PreVotes provides a mock function with given fields: -func (_m *BlockFinalityAPI) PreVotes() []ed25519.PublicKeyBytes { - ret := _m.Called() - - var r0 []ed25519.PublicKeyBytes - if rf, ok := ret.Get(0).(func() []ed25519.PublicKeyBytes); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]ed25519.PublicKeyBytes) - } - } - - return r0 -} - -type mockConstructorTestingTNewBlockFinalityAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewBlockFinalityAPI creates a new instance of BlockFinalityAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewBlockFinalityAPI(t mockConstructorTestingTNewBlockFinalityAPI) *BlockFinalityAPI { - mock := &BlockFinalityAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/block_producer_api.go b/dot/rpc/modules/mocks/block_producer_api.go deleted file mode 100644 index 39dc6122a4..0000000000 --- a/dot/rpc/modules/mocks/block_producer_api.go +++ /dev/null @@ -1,81 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import mock "github.com/stretchr/testify/mock" - -// BlockProducerAPI is an autogenerated mock type for the BlockProducerAPI type -type BlockProducerAPI struct { - mock.Mock -} - -// EpochLength provides a mock function with given fields: -func (_m *BlockProducerAPI) EpochLength() uint64 { - ret := _m.Called() - - var r0 uint64 - if rf, ok := ret.Get(0).(func() uint64); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(uint64) - } - - return r0 -} - -// Pause provides a mock function with given fields: -func (_m *BlockProducerAPI) Pause() error { - ret := _m.Called() - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// Resume provides a mock function with given fields: -func (_m *BlockProducerAPI) Resume() error { - ret := _m.Called() - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// SlotDuration provides a mock function with given fields: -func (_m *BlockProducerAPI) SlotDuration() uint64 { - ret := _m.Called() - - var r0 uint64 - if rf, ok := ret.Get(0).(func() uint64); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(uint64) - } - - return r0 -} - -type mockConstructorTestingTNewBlockProducerAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewBlockProducerAPI creates a new instance of BlockProducerAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewBlockProducerAPI(t mockConstructorTestingTNewBlockProducerAPI) *BlockProducerAPI { - mock := &BlockProducerAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/core_api.go b/dot/rpc/modules/mocks/core_api.go deleted file mode 100644 index 48b4b36028..0000000000 --- a/dot/rpc/modules/mocks/core_api.go +++ /dev/null @@ -1,182 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - core "github.com/ChainSafe/gossamer/dot/core" - common "github.com/ChainSafe/gossamer/lib/common" - - mock "github.com/stretchr/testify/mock" - - runtime "github.com/ChainSafe/gossamer/lib/runtime" - - types "github.com/ChainSafe/gossamer/dot/types" -) - -// CoreAPI is an autogenerated mock type for the CoreAPI type -type CoreAPI struct { - mock.Mock -} - -// DecodeSessionKeys provides a mock function with given fields: enc -func (_m *CoreAPI) DecodeSessionKeys(enc []byte) ([]byte, error) { - ret := _m.Called(enc) - - var r0 []byte - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(enc) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(enc) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetMetadata provides a mock function with given fields: bhash -func (_m *CoreAPI) GetMetadata(bhash *common.Hash) ([]byte, error) { - ret := _m.Called(bhash) - - var r0 []byte - if rf, ok := ret.Get(0).(func(*common.Hash) []byte); ok { - r0 = rf(bhash) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash) error); ok { - r1 = rf(bhash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetReadProofAt provides a mock function with given fields: block, keys -func (_m *CoreAPI) GetReadProofAt(block common.Hash, keys [][]byte) (common.Hash, [][]byte, error) { - ret := _m.Called(block, keys) - - var r0 common.Hash - if rf, ok := ret.Get(0).(func(common.Hash, [][]byte) common.Hash); ok { - r0 = rf(block, keys) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - var r1 [][]byte - if rf, ok := ret.Get(1).(func(common.Hash, [][]byte) [][]byte); ok { - r1 = rf(block, keys) - } else { - if ret.Get(1) != nil { - r1 = ret.Get(1).([][]byte) - } - } - - var r2 error - if rf, ok := ret.Get(2).(func(common.Hash, [][]byte) error); ok { - r2 = rf(block, keys) - } else { - r2 = ret.Error(2) - } - - return r0, r1, r2 -} - -// GetRuntimeVersion provides a mock function with given fields: bhash -func (_m *CoreAPI) GetRuntimeVersion(bhash *common.Hash) (runtime.Version, error) { - ret := _m.Called(bhash) - - var r0 runtime.Version - if rf, ok := ret.Get(0).(func(*common.Hash) runtime.Version); ok { - r0 = rf(bhash) - } else { - r0 = ret.Get(0).(runtime.Version) - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash) error); ok { - r1 = rf(bhash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// HandleSubmittedExtrinsic provides a mock function with given fields: _a0 -func (_m *CoreAPI) HandleSubmittedExtrinsic(_a0 types.Extrinsic) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func(types.Extrinsic) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// HasKey provides a mock function with given fields: pubKeyStr, keyType -func (_m *CoreAPI) HasKey(pubKeyStr string, keyType string) (bool, error) { - ret := _m.Called(pubKeyStr, keyType) - - var r0 bool - if rf, ok := ret.Get(0).(func(string, string) bool); ok { - r0 = rf(pubKeyStr, keyType) - } else { - r0 = ret.Get(0).(bool) - } - - var r1 error - if rf, ok := ret.Get(1).(func(string, string) error); ok { - r1 = rf(pubKeyStr, keyType) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// InsertKey provides a mock function with given fields: kp, keystoreType -func (_m *CoreAPI) InsertKey(kp core.KeyPair, keystoreType string) error { - ret := _m.Called(kp, keystoreType) - - var r0 error - if rf, ok := ret.Get(0).(func(core.KeyPair, string) error); ok { - r0 = rf(kp, keystoreType) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewCoreAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewCoreAPI creates a new instance of CoreAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewCoreAPI(t mockConstructorTestingTNewCoreAPI) *CoreAPI { - mock := &CoreAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/mocks.go b/dot/rpc/modules/mocks/mocks.go new file mode 100644 index 0000000000..266f7f6269 --- /dev/null +++ b/dot/rpc/modules/mocks/mocks.go @@ -0,0 +1,1130 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/ChainSafe/gossamer/dot/rpc/modules (interfaces: StorageAPI,BlockAPI,NetworkAPI,BlockProducerAPI,TransactionStateAPI,CoreAPI,SystemAPI,BlockFinalityAPI,RuntimeStorageAPI,SyncStateAPI) + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + core "github.com/ChainSafe/gossamer/dot/core" + state "github.com/ChainSafe/gossamer/dot/state" + types "github.com/ChainSafe/gossamer/dot/types" + common "github.com/ChainSafe/gossamer/lib/common" + ed25519 "github.com/ChainSafe/gossamer/lib/crypto/ed25519" + genesis "github.com/ChainSafe/gossamer/lib/genesis" + runtime "github.com/ChainSafe/gossamer/lib/runtime" + transaction "github.com/ChainSafe/gossamer/lib/transaction" + trie "github.com/ChainSafe/gossamer/lib/trie" + gomock "github.com/golang/mock/gomock" +) + +// MockStorageAPI is a mock of StorageAPI interface. +type MockStorageAPI struct { + ctrl *gomock.Controller + recorder *MockStorageAPIMockRecorder +} + +// MockStorageAPIMockRecorder is the mock recorder for MockStorageAPI. +type MockStorageAPIMockRecorder struct { + mock *MockStorageAPI +} + +// NewMockStorageAPI creates a new mock instance. +func NewMockStorageAPI(ctrl *gomock.Controller) *MockStorageAPI { + mock := &MockStorageAPI{ctrl: ctrl} + mock.recorder = &MockStorageAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockStorageAPI) EXPECT() *MockStorageAPIMockRecorder { + return m.recorder +} + +// Entries mocks base method. +func (m *MockStorageAPI) Entries(arg0 *common.Hash) (map[string][]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Entries", arg0) + ret0, _ := ret[0].(map[string][]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Entries indicates an expected call of Entries. +func (mr *MockStorageAPIMockRecorder) Entries(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Entries", reflect.TypeOf((*MockStorageAPI)(nil).Entries), arg0) +} + +// GetKeysWithPrefix mocks base method. +func (m *MockStorageAPI) GetKeysWithPrefix(arg0 *common.Hash, arg1 []byte) ([][]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetKeysWithPrefix", arg0, arg1) + ret0, _ := ret[0].([][]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetKeysWithPrefix indicates an expected call of GetKeysWithPrefix. +func (mr *MockStorageAPIMockRecorder) GetKeysWithPrefix(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetKeysWithPrefix", reflect.TypeOf((*MockStorageAPI)(nil).GetKeysWithPrefix), arg0, arg1) +} + +// GetStateRootFromBlock mocks base method. +func (m *MockStorageAPI) GetStateRootFromBlock(arg0 *common.Hash) (*common.Hash, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetStateRootFromBlock", arg0) + ret0, _ := ret[0].(*common.Hash) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetStateRootFromBlock indicates an expected call of GetStateRootFromBlock. +func (mr *MockStorageAPIMockRecorder) GetStateRootFromBlock(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStateRootFromBlock", reflect.TypeOf((*MockStorageAPI)(nil).GetStateRootFromBlock), arg0) +} + +// GetStorage mocks base method. +func (m *MockStorageAPI) GetStorage(arg0 *common.Hash, arg1 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetStorage", arg0, arg1) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetStorage indicates an expected call of GetStorage. +func (mr *MockStorageAPIMockRecorder) GetStorage(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStorage", reflect.TypeOf((*MockStorageAPI)(nil).GetStorage), arg0, arg1) +} + +// GetStorageByBlockHash mocks base method. +func (m *MockStorageAPI) GetStorageByBlockHash(arg0 *common.Hash, arg1 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetStorageByBlockHash", arg0, arg1) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetStorageByBlockHash indicates an expected call of GetStorageByBlockHash. +func (mr *MockStorageAPIMockRecorder) GetStorageByBlockHash(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStorageByBlockHash", reflect.TypeOf((*MockStorageAPI)(nil).GetStorageByBlockHash), arg0, arg1) +} + +// GetStorageChild mocks base method. +func (m *MockStorageAPI) GetStorageChild(arg0 *common.Hash, arg1 []byte) (*trie.Trie, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetStorageChild", arg0, arg1) + ret0, _ := ret[0].(*trie.Trie) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetStorageChild indicates an expected call of GetStorageChild. +func (mr *MockStorageAPIMockRecorder) GetStorageChild(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStorageChild", reflect.TypeOf((*MockStorageAPI)(nil).GetStorageChild), arg0, arg1) +} + +// GetStorageFromChild mocks base method. +func (m *MockStorageAPI) GetStorageFromChild(arg0 *common.Hash, arg1, arg2 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetStorageFromChild", arg0, arg1, arg2) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetStorageFromChild indicates an expected call of GetStorageFromChild. +func (mr *MockStorageAPIMockRecorder) GetStorageFromChild(arg0, arg1, arg2 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetStorageFromChild", reflect.TypeOf((*MockStorageAPI)(nil).GetStorageFromChild), arg0, arg1, arg2) +} + +// RegisterStorageObserver mocks base method. +func (m *MockStorageAPI) RegisterStorageObserver(arg0 state.Observer) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "RegisterStorageObserver", arg0) +} + +// RegisterStorageObserver indicates an expected call of RegisterStorageObserver. +func (mr *MockStorageAPIMockRecorder) RegisterStorageObserver(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterStorageObserver", reflect.TypeOf((*MockStorageAPI)(nil).RegisterStorageObserver), arg0) +} + +// UnregisterStorageObserver mocks base method. +func (m *MockStorageAPI) UnregisterStorageObserver(arg0 state.Observer) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "UnregisterStorageObserver", arg0) +} + +// UnregisterStorageObserver indicates an expected call of UnregisterStorageObserver. +func (mr *MockStorageAPIMockRecorder) UnregisterStorageObserver(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnregisterStorageObserver", reflect.TypeOf((*MockStorageAPI)(nil).UnregisterStorageObserver), arg0) +} + +// MockBlockAPI is a mock of BlockAPI interface. +type MockBlockAPI struct { + ctrl *gomock.Controller + recorder *MockBlockAPIMockRecorder +} + +// MockBlockAPIMockRecorder is the mock recorder for MockBlockAPI. +type MockBlockAPIMockRecorder struct { + mock *MockBlockAPI +} + +// NewMockBlockAPI creates a new mock instance. +func NewMockBlockAPI(ctrl *gomock.Controller) *MockBlockAPI { + mock := &MockBlockAPI{ctrl: ctrl} + mock.recorder = &MockBlockAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockBlockAPI) EXPECT() *MockBlockAPIMockRecorder { + return m.recorder +} + +// BestBlockHash mocks base method. +func (m *MockBlockAPI) BestBlockHash() common.Hash { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "BestBlockHash") + ret0, _ := ret[0].(common.Hash) + return ret0 +} + +// BestBlockHash indicates an expected call of BestBlockHash. +func (mr *MockBlockAPIMockRecorder) BestBlockHash() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BestBlockHash", reflect.TypeOf((*MockBlockAPI)(nil).BestBlockHash)) +} + +// FreeFinalisedNotifierChannel mocks base method. +func (m *MockBlockAPI) FreeFinalisedNotifierChannel(arg0 chan *types.FinalisationInfo) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "FreeFinalisedNotifierChannel", arg0) +} + +// FreeFinalisedNotifierChannel indicates an expected call of FreeFinalisedNotifierChannel. +func (mr *MockBlockAPIMockRecorder) FreeFinalisedNotifierChannel(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FreeFinalisedNotifierChannel", reflect.TypeOf((*MockBlockAPI)(nil).FreeFinalisedNotifierChannel), arg0) +} + +// FreeImportedBlockNotifierChannel mocks base method. +func (m *MockBlockAPI) FreeImportedBlockNotifierChannel(arg0 chan *types.Block) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "FreeImportedBlockNotifierChannel", arg0) +} + +// FreeImportedBlockNotifierChannel indicates an expected call of FreeImportedBlockNotifierChannel. +func (mr *MockBlockAPIMockRecorder) FreeImportedBlockNotifierChannel(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FreeImportedBlockNotifierChannel", reflect.TypeOf((*MockBlockAPI)(nil).FreeImportedBlockNotifierChannel), arg0) +} + +// GetBlockByHash mocks base method. +func (m *MockBlockAPI) GetBlockByHash(arg0 common.Hash) (*types.Block, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetBlockByHash", arg0) + ret0, _ := ret[0].(*types.Block) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetBlockByHash indicates an expected call of GetBlockByHash. +func (mr *MockBlockAPIMockRecorder) GetBlockByHash(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetBlockByHash", reflect.TypeOf((*MockBlockAPI)(nil).GetBlockByHash), arg0) +} + +// GetFinalisedHash mocks base method. +func (m *MockBlockAPI) GetFinalisedHash(arg0, arg1 uint64) (common.Hash, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetFinalisedHash", arg0, arg1) + ret0, _ := ret[0].(common.Hash) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetFinalisedHash indicates an expected call of GetFinalisedHash. +func (mr *MockBlockAPIMockRecorder) GetFinalisedHash(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFinalisedHash", reflect.TypeOf((*MockBlockAPI)(nil).GetFinalisedHash), arg0, arg1) +} + +// GetFinalisedNotifierChannel mocks base method. +func (m *MockBlockAPI) GetFinalisedNotifierChannel() chan *types.FinalisationInfo { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetFinalisedNotifierChannel") + ret0, _ := ret[0].(chan *types.FinalisationInfo) + return ret0 +} + +// GetFinalisedNotifierChannel indicates an expected call of GetFinalisedNotifierChannel. +func (mr *MockBlockAPIMockRecorder) GetFinalisedNotifierChannel() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFinalisedNotifierChannel", reflect.TypeOf((*MockBlockAPI)(nil).GetFinalisedNotifierChannel)) +} + +// GetHashByNumber mocks base method. +func (m *MockBlockAPI) GetHashByNumber(arg0 uint) (common.Hash, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetHashByNumber", arg0) + ret0, _ := ret[0].(common.Hash) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetHashByNumber indicates an expected call of GetHashByNumber. +func (mr *MockBlockAPIMockRecorder) GetHashByNumber(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHashByNumber", reflect.TypeOf((*MockBlockAPI)(nil).GetHashByNumber), arg0) +} + +// GetHeader mocks base method. +func (m *MockBlockAPI) GetHeader(arg0 common.Hash) (*types.Header, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetHeader", arg0) + ret0, _ := ret[0].(*types.Header) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetHeader indicates an expected call of GetHeader. +func (mr *MockBlockAPIMockRecorder) GetHeader(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHeader", reflect.TypeOf((*MockBlockAPI)(nil).GetHeader), arg0) +} + +// GetHighestFinalisedHash mocks base method. +func (m *MockBlockAPI) GetHighestFinalisedHash() (common.Hash, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetHighestFinalisedHash") + ret0, _ := ret[0].(common.Hash) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetHighestFinalisedHash indicates an expected call of GetHighestFinalisedHash. +func (mr *MockBlockAPIMockRecorder) GetHighestFinalisedHash() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHighestFinalisedHash", reflect.TypeOf((*MockBlockAPI)(nil).GetHighestFinalisedHash)) +} + +// GetImportedBlockNotifierChannel mocks base method. +func (m *MockBlockAPI) GetImportedBlockNotifierChannel() chan *types.Block { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetImportedBlockNotifierChannel") + ret0, _ := ret[0].(chan *types.Block) + return ret0 +} + +// GetImportedBlockNotifierChannel indicates an expected call of GetImportedBlockNotifierChannel. +func (mr *MockBlockAPIMockRecorder) GetImportedBlockNotifierChannel() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetImportedBlockNotifierChannel", reflect.TypeOf((*MockBlockAPI)(nil).GetImportedBlockNotifierChannel)) +} + +// GetJustification mocks base method. +func (m *MockBlockAPI) GetJustification(arg0 common.Hash) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetJustification", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetJustification indicates an expected call of GetJustification. +func (mr *MockBlockAPIMockRecorder) GetJustification(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetJustification", reflect.TypeOf((*MockBlockAPI)(nil).GetJustification), arg0) +} + +// GetRuntime mocks base method. +func (m *MockBlockAPI) GetRuntime(arg0 common.Hash) (state.Runtime, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetRuntime", arg0) + ret0, _ := ret[0].(state.Runtime) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetRuntime indicates an expected call of GetRuntime. +func (mr *MockBlockAPIMockRecorder) GetRuntime(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRuntime", reflect.TypeOf((*MockBlockAPI)(nil).GetRuntime), arg0) +} + +// HasJustification mocks base method. +func (m *MockBlockAPI) HasJustification(arg0 common.Hash) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "HasJustification", arg0) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// HasJustification indicates an expected call of HasJustification. +func (mr *MockBlockAPIMockRecorder) HasJustification(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HasJustification", reflect.TypeOf((*MockBlockAPI)(nil).HasJustification), arg0) +} + +// RangeInMemory mocks base method. +func (m *MockBlockAPI) RangeInMemory(arg0, arg1 common.Hash) ([]common.Hash, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "RangeInMemory", arg0, arg1) + ret0, _ := ret[0].([]common.Hash) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// RangeInMemory indicates an expected call of RangeInMemory. +func (mr *MockBlockAPIMockRecorder) RangeInMemory(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RangeInMemory", reflect.TypeOf((*MockBlockAPI)(nil).RangeInMemory), arg0, arg1) +} + +// RegisterRuntimeUpdatedChannel mocks base method. +func (m *MockBlockAPI) RegisterRuntimeUpdatedChannel(arg0 chan<- runtime.Version) (uint32, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "RegisterRuntimeUpdatedChannel", arg0) + ret0, _ := ret[0].(uint32) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// RegisterRuntimeUpdatedChannel indicates an expected call of RegisterRuntimeUpdatedChannel. +func (mr *MockBlockAPIMockRecorder) RegisterRuntimeUpdatedChannel(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterRuntimeUpdatedChannel", reflect.TypeOf((*MockBlockAPI)(nil).RegisterRuntimeUpdatedChannel), arg0) +} + +// UnregisterRuntimeUpdatedChannel mocks base method. +func (m *MockBlockAPI) UnregisterRuntimeUpdatedChannel(arg0 uint32) bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UnregisterRuntimeUpdatedChannel", arg0) + ret0, _ := ret[0].(bool) + return ret0 +} + +// UnregisterRuntimeUpdatedChannel indicates an expected call of UnregisterRuntimeUpdatedChannel. +func (mr *MockBlockAPIMockRecorder) UnregisterRuntimeUpdatedChannel(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnregisterRuntimeUpdatedChannel", reflect.TypeOf((*MockBlockAPI)(nil).UnregisterRuntimeUpdatedChannel), arg0) +} + +// MockNetworkAPI is a mock of NetworkAPI interface. +type MockNetworkAPI struct { + ctrl *gomock.Controller + recorder *MockNetworkAPIMockRecorder +} + +// MockNetworkAPIMockRecorder is the mock recorder for MockNetworkAPI. +type MockNetworkAPIMockRecorder struct { + mock *MockNetworkAPI +} + +// NewMockNetworkAPI creates a new mock instance. +func NewMockNetworkAPI(ctrl *gomock.Controller) *MockNetworkAPI { + mock := &MockNetworkAPI{ctrl: ctrl} + mock.recorder = &MockNetworkAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockNetworkAPI) EXPECT() *MockNetworkAPIMockRecorder { + return m.recorder +} + +// AddReservedPeers mocks base method. +func (m *MockNetworkAPI) AddReservedPeers(arg0 ...string) error { + m.ctrl.T.Helper() + varargs := []interface{}{} + for _, a := range arg0 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "AddReservedPeers", varargs...) + ret0, _ := ret[0].(error) + return ret0 +} + +// AddReservedPeers indicates an expected call of AddReservedPeers. +func (mr *MockNetworkAPIMockRecorder) AddReservedPeers(arg0 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddReservedPeers", reflect.TypeOf((*MockNetworkAPI)(nil).AddReservedPeers), arg0...) +} + +// Health mocks base method. +func (m *MockNetworkAPI) Health() common.Health { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Health") + ret0, _ := ret[0].(common.Health) + return ret0 +} + +// Health indicates an expected call of Health. +func (mr *MockNetworkAPIMockRecorder) Health() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Health", reflect.TypeOf((*MockNetworkAPI)(nil).Health)) +} + +// NetworkState mocks base method. +func (m *MockNetworkAPI) NetworkState() common.NetworkState { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "NetworkState") + ret0, _ := ret[0].(common.NetworkState) + return ret0 +} + +// NetworkState indicates an expected call of NetworkState. +func (mr *MockNetworkAPIMockRecorder) NetworkState() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NetworkState", reflect.TypeOf((*MockNetworkAPI)(nil).NetworkState)) +} + +// NodeRoles mocks base method. +func (m *MockNetworkAPI) NodeRoles() common.Roles { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "NodeRoles") + ret0, _ := ret[0].(common.Roles) + return ret0 +} + +// NodeRoles indicates an expected call of NodeRoles. +func (mr *MockNetworkAPIMockRecorder) NodeRoles() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NodeRoles", reflect.TypeOf((*MockNetworkAPI)(nil).NodeRoles)) +} + +// Peers mocks base method. +func (m *MockNetworkAPI) Peers() []common.PeerInfo { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Peers") + ret0, _ := ret[0].([]common.PeerInfo) + return ret0 +} + +// Peers indicates an expected call of Peers. +func (mr *MockNetworkAPIMockRecorder) Peers() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Peers", reflect.TypeOf((*MockNetworkAPI)(nil).Peers)) +} + +// RemoveReservedPeers mocks base method. +func (m *MockNetworkAPI) RemoveReservedPeers(arg0 ...string) error { + m.ctrl.T.Helper() + varargs := []interface{}{} + for _, a := range arg0 { + varargs = append(varargs, a) + } + ret := m.ctrl.Call(m, "RemoveReservedPeers", varargs...) + ret0, _ := ret[0].(error) + return ret0 +} + +// RemoveReservedPeers indicates an expected call of RemoveReservedPeers. +func (mr *MockNetworkAPIMockRecorder) RemoveReservedPeers(arg0 ...interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveReservedPeers", reflect.TypeOf((*MockNetworkAPI)(nil).RemoveReservedPeers), arg0...) +} + +// Start mocks base method. +func (m *MockNetworkAPI) Start() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Start") + ret0, _ := ret[0].(error) + return ret0 +} + +// Start indicates an expected call of Start. +func (mr *MockNetworkAPIMockRecorder) Start() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockNetworkAPI)(nil).Start)) +} + +// StartingBlock mocks base method. +func (m *MockNetworkAPI) StartingBlock() int64 { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "StartingBlock") + ret0, _ := ret[0].(int64) + return ret0 +} + +// StartingBlock indicates an expected call of StartingBlock. +func (mr *MockNetworkAPIMockRecorder) StartingBlock() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartingBlock", reflect.TypeOf((*MockNetworkAPI)(nil).StartingBlock)) +} + +// Stop mocks base method. +func (m *MockNetworkAPI) Stop() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Stop") + ret0, _ := ret[0].(error) + return ret0 +} + +// Stop indicates an expected call of Stop. +func (mr *MockNetworkAPIMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockNetworkAPI)(nil).Stop)) +} + +// MockBlockProducerAPI is a mock of BlockProducerAPI interface. +type MockBlockProducerAPI struct { + ctrl *gomock.Controller + recorder *MockBlockProducerAPIMockRecorder +} + +// MockBlockProducerAPIMockRecorder is the mock recorder for MockBlockProducerAPI. +type MockBlockProducerAPIMockRecorder struct { + mock *MockBlockProducerAPI +} + +// NewMockBlockProducerAPI creates a new mock instance. +func NewMockBlockProducerAPI(ctrl *gomock.Controller) *MockBlockProducerAPI { + mock := &MockBlockProducerAPI{ctrl: ctrl} + mock.recorder = &MockBlockProducerAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockBlockProducerAPI) EXPECT() *MockBlockProducerAPIMockRecorder { + return m.recorder +} + +// EpochLength mocks base method. +func (m *MockBlockProducerAPI) EpochLength() uint64 { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "EpochLength") + ret0, _ := ret[0].(uint64) + return ret0 +} + +// EpochLength indicates an expected call of EpochLength. +func (mr *MockBlockProducerAPIMockRecorder) EpochLength() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EpochLength", reflect.TypeOf((*MockBlockProducerAPI)(nil).EpochLength)) +} + +// Pause mocks base method. +func (m *MockBlockProducerAPI) Pause() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Pause") + ret0, _ := ret[0].(error) + return ret0 +} + +// Pause indicates an expected call of Pause. +func (mr *MockBlockProducerAPIMockRecorder) Pause() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Pause", reflect.TypeOf((*MockBlockProducerAPI)(nil).Pause)) +} + +// Resume mocks base method. +func (m *MockBlockProducerAPI) Resume() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Resume") + ret0, _ := ret[0].(error) + return ret0 +} + +// Resume indicates an expected call of Resume. +func (mr *MockBlockProducerAPIMockRecorder) Resume() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Resume", reflect.TypeOf((*MockBlockProducerAPI)(nil).Resume)) +} + +// SlotDuration mocks base method. +func (m *MockBlockProducerAPI) SlotDuration() uint64 { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SlotDuration") + ret0, _ := ret[0].(uint64) + return ret0 +} + +// SlotDuration indicates an expected call of SlotDuration. +func (mr *MockBlockProducerAPIMockRecorder) SlotDuration() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SlotDuration", reflect.TypeOf((*MockBlockProducerAPI)(nil).SlotDuration)) +} + +// MockTransactionStateAPI is a mock of TransactionStateAPI interface. +type MockTransactionStateAPI struct { + ctrl *gomock.Controller + recorder *MockTransactionStateAPIMockRecorder +} + +// MockTransactionStateAPIMockRecorder is the mock recorder for MockTransactionStateAPI. +type MockTransactionStateAPIMockRecorder struct { + mock *MockTransactionStateAPI +} + +// NewMockTransactionStateAPI creates a new mock instance. +func NewMockTransactionStateAPI(ctrl *gomock.Controller) *MockTransactionStateAPI { + mock := &MockTransactionStateAPI{ctrl: ctrl} + mock.recorder = &MockTransactionStateAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockTransactionStateAPI) EXPECT() *MockTransactionStateAPIMockRecorder { + return m.recorder +} + +// Pending mocks base method. +func (m *MockTransactionStateAPI) Pending() []*transaction.ValidTransaction { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Pending") + ret0, _ := ret[0].([]*transaction.ValidTransaction) + return ret0 +} + +// Pending indicates an expected call of Pending. +func (mr *MockTransactionStateAPIMockRecorder) Pending() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Pending", reflect.TypeOf((*MockTransactionStateAPI)(nil).Pending)) +} + +// MockCoreAPI is a mock of CoreAPI interface. +type MockCoreAPI struct { + ctrl *gomock.Controller + recorder *MockCoreAPIMockRecorder +} + +// MockCoreAPIMockRecorder is the mock recorder for MockCoreAPI. +type MockCoreAPIMockRecorder struct { + mock *MockCoreAPI +} + +// NewMockCoreAPI creates a new mock instance. +func NewMockCoreAPI(ctrl *gomock.Controller) *MockCoreAPI { + mock := &MockCoreAPI{ctrl: ctrl} + mock.recorder = &MockCoreAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockCoreAPI) EXPECT() *MockCoreAPIMockRecorder { + return m.recorder +} + +// DecodeSessionKeys mocks base method. +func (m *MockCoreAPI) DecodeSessionKeys(arg0 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DecodeSessionKeys", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DecodeSessionKeys indicates an expected call of DecodeSessionKeys. +func (mr *MockCoreAPIMockRecorder) DecodeSessionKeys(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecodeSessionKeys", reflect.TypeOf((*MockCoreAPI)(nil).DecodeSessionKeys), arg0) +} + +// GetMetadata mocks base method. +func (m *MockCoreAPI) GetMetadata(arg0 *common.Hash) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetMetadata", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetMetadata indicates an expected call of GetMetadata. +func (mr *MockCoreAPIMockRecorder) GetMetadata(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetMetadata", reflect.TypeOf((*MockCoreAPI)(nil).GetMetadata), arg0) +} + +// GetReadProofAt mocks base method. +func (m *MockCoreAPI) GetReadProofAt(arg0 common.Hash, arg1 [][]byte) (common.Hash, [][]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetReadProofAt", arg0, arg1) + ret0, _ := ret[0].(common.Hash) + ret1, _ := ret[1].([][]byte) + ret2, _ := ret[2].(error) + return ret0, ret1, ret2 +} + +// GetReadProofAt indicates an expected call of GetReadProofAt. +func (mr *MockCoreAPIMockRecorder) GetReadProofAt(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetReadProofAt", reflect.TypeOf((*MockCoreAPI)(nil).GetReadProofAt), arg0, arg1) +} + +// GetRuntimeVersion mocks base method. +func (m *MockCoreAPI) GetRuntimeVersion(arg0 *common.Hash) (runtime.Version, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetRuntimeVersion", arg0) + ret0, _ := ret[0].(runtime.Version) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetRuntimeVersion indicates an expected call of GetRuntimeVersion. +func (mr *MockCoreAPIMockRecorder) GetRuntimeVersion(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRuntimeVersion", reflect.TypeOf((*MockCoreAPI)(nil).GetRuntimeVersion), arg0) +} + +// HandleSubmittedExtrinsic mocks base method. +func (m *MockCoreAPI) HandleSubmittedExtrinsic(arg0 types.Extrinsic) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "HandleSubmittedExtrinsic", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// HandleSubmittedExtrinsic indicates an expected call of HandleSubmittedExtrinsic. +func (mr *MockCoreAPIMockRecorder) HandleSubmittedExtrinsic(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleSubmittedExtrinsic", reflect.TypeOf((*MockCoreAPI)(nil).HandleSubmittedExtrinsic), arg0) +} + +// HasKey mocks base method. +func (m *MockCoreAPI) HasKey(arg0, arg1 string) (bool, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "HasKey", arg0, arg1) + ret0, _ := ret[0].(bool) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// HasKey indicates an expected call of HasKey. +func (mr *MockCoreAPIMockRecorder) HasKey(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HasKey", reflect.TypeOf((*MockCoreAPI)(nil).HasKey), arg0, arg1) +} + +// InsertKey mocks base method. +func (m *MockCoreAPI) InsertKey(arg0 core.KeyPair, arg1 string) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "InsertKey", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// InsertKey indicates an expected call of InsertKey. +func (mr *MockCoreAPIMockRecorder) InsertKey(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InsertKey", reflect.TypeOf((*MockCoreAPI)(nil).InsertKey), arg0, arg1) +} + +// MockSystemAPI is a mock of SystemAPI interface. +type MockSystemAPI struct { + ctrl *gomock.Controller + recorder *MockSystemAPIMockRecorder +} + +// MockSystemAPIMockRecorder is the mock recorder for MockSystemAPI. +type MockSystemAPIMockRecorder struct { + mock *MockSystemAPI +} + +// NewMockSystemAPI creates a new mock instance. +func NewMockSystemAPI(ctrl *gomock.Controller) *MockSystemAPI { + mock := &MockSystemAPI{ctrl: ctrl} + mock.recorder = &MockSystemAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSystemAPI) EXPECT() *MockSystemAPIMockRecorder { + return m.recorder +} + +// ChainName mocks base method. +func (m *MockSystemAPI) ChainName() string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ChainName") + ret0, _ := ret[0].(string) + return ret0 +} + +// ChainName indicates an expected call of ChainName. +func (mr *MockSystemAPIMockRecorder) ChainName() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChainName", reflect.TypeOf((*MockSystemAPI)(nil).ChainName)) +} + +// ChainType mocks base method. +func (m *MockSystemAPI) ChainType() string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ChainType") + ret0, _ := ret[0].(string) + return ret0 +} + +// ChainType indicates an expected call of ChainType. +func (mr *MockSystemAPIMockRecorder) ChainType() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ChainType", reflect.TypeOf((*MockSystemAPI)(nil).ChainType)) +} + +// Properties mocks base method. +func (m *MockSystemAPI) Properties() map[string]interface{} { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Properties") + ret0, _ := ret[0].(map[string]interface{}) + return ret0 +} + +// Properties indicates an expected call of Properties. +func (mr *MockSystemAPIMockRecorder) Properties() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Properties", reflect.TypeOf((*MockSystemAPI)(nil).Properties)) +} + +// SystemName mocks base method. +func (m *MockSystemAPI) SystemName() string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SystemName") + ret0, _ := ret[0].(string) + return ret0 +} + +// SystemName indicates an expected call of SystemName. +func (mr *MockSystemAPIMockRecorder) SystemName() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemName", reflect.TypeOf((*MockSystemAPI)(nil).SystemName)) +} + +// SystemVersion mocks base method. +func (m *MockSystemAPI) SystemVersion() string { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SystemVersion") + ret0, _ := ret[0].(string) + return ret0 +} + +// SystemVersion indicates an expected call of SystemVersion. +func (mr *MockSystemAPIMockRecorder) SystemVersion() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SystemVersion", reflect.TypeOf((*MockSystemAPI)(nil).SystemVersion)) +} + +// MockBlockFinalityAPI is a mock of BlockFinalityAPI interface. +type MockBlockFinalityAPI struct { + ctrl *gomock.Controller + recorder *MockBlockFinalityAPIMockRecorder +} + +// MockBlockFinalityAPIMockRecorder is the mock recorder for MockBlockFinalityAPI. +type MockBlockFinalityAPIMockRecorder struct { + mock *MockBlockFinalityAPI +} + +// NewMockBlockFinalityAPI creates a new mock instance. +func NewMockBlockFinalityAPI(ctrl *gomock.Controller) *MockBlockFinalityAPI { + mock := &MockBlockFinalityAPI{ctrl: ctrl} + mock.recorder = &MockBlockFinalityAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockBlockFinalityAPI) EXPECT() *MockBlockFinalityAPIMockRecorder { + return m.recorder +} + +// GetRound mocks base method. +func (m *MockBlockFinalityAPI) GetRound() uint64 { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetRound") + ret0, _ := ret[0].(uint64) + return ret0 +} + +// GetRound indicates an expected call of GetRound. +func (mr *MockBlockFinalityAPIMockRecorder) GetRound() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRound", reflect.TypeOf((*MockBlockFinalityAPI)(nil).GetRound)) +} + +// GetSetID mocks base method. +func (m *MockBlockFinalityAPI) GetSetID() uint64 { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetSetID") + ret0, _ := ret[0].(uint64) + return ret0 +} + +// GetSetID indicates an expected call of GetSetID. +func (mr *MockBlockFinalityAPIMockRecorder) GetSetID() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSetID", reflect.TypeOf((*MockBlockFinalityAPI)(nil).GetSetID)) +} + +// GetVoters mocks base method. +func (m *MockBlockFinalityAPI) GetVoters() types.GrandpaVoters { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetVoters") + ret0, _ := ret[0].(types.GrandpaVoters) + return ret0 +} + +// GetVoters indicates an expected call of GetVoters. +func (mr *MockBlockFinalityAPIMockRecorder) GetVoters() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVoters", reflect.TypeOf((*MockBlockFinalityAPI)(nil).GetVoters)) +} + +// PreCommits mocks base method. +func (m *MockBlockFinalityAPI) PreCommits() []ed25519.PublicKeyBytes { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "PreCommits") + ret0, _ := ret[0].([]ed25519.PublicKeyBytes) + return ret0 +} + +// PreCommits indicates an expected call of PreCommits. +func (mr *MockBlockFinalityAPIMockRecorder) PreCommits() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreCommits", reflect.TypeOf((*MockBlockFinalityAPI)(nil).PreCommits)) +} + +// PreVotes mocks base method. +func (m *MockBlockFinalityAPI) PreVotes() []ed25519.PublicKeyBytes { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "PreVotes") + ret0, _ := ret[0].([]ed25519.PublicKeyBytes) + return ret0 +} + +// PreVotes indicates an expected call of PreVotes. +func (mr *MockBlockFinalityAPIMockRecorder) PreVotes() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreVotes", reflect.TypeOf((*MockBlockFinalityAPI)(nil).PreVotes)) +} + +// MockRuntimeStorageAPI is a mock of RuntimeStorageAPI interface. +type MockRuntimeStorageAPI struct { + ctrl *gomock.Controller + recorder *MockRuntimeStorageAPIMockRecorder +} + +// MockRuntimeStorageAPIMockRecorder is the mock recorder for MockRuntimeStorageAPI. +type MockRuntimeStorageAPIMockRecorder struct { + mock *MockRuntimeStorageAPI +} + +// NewMockRuntimeStorageAPI creates a new mock instance. +func NewMockRuntimeStorageAPI(ctrl *gomock.Controller) *MockRuntimeStorageAPI { + mock := &MockRuntimeStorageAPI{ctrl: ctrl} + mock.recorder = &MockRuntimeStorageAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockRuntimeStorageAPI) EXPECT() *MockRuntimeStorageAPIMockRecorder { + return m.recorder +} + +// GetLocal mocks base method. +func (m *MockRuntimeStorageAPI) GetLocal(arg0 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetLocal", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetLocal indicates an expected call of GetLocal. +func (mr *MockRuntimeStorageAPIMockRecorder) GetLocal(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLocal", reflect.TypeOf((*MockRuntimeStorageAPI)(nil).GetLocal), arg0) +} + +// GetPersistent mocks base method. +func (m *MockRuntimeStorageAPI) GetPersistent(arg0 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetPersistent", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GetPersistent indicates an expected call of GetPersistent. +func (mr *MockRuntimeStorageAPIMockRecorder) GetPersistent(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPersistent", reflect.TypeOf((*MockRuntimeStorageAPI)(nil).GetPersistent), arg0) +} + +// SetLocal mocks base method. +func (m *MockRuntimeStorageAPI) SetLocal(arg0, arg1 []byte) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SetLocal", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// SetLocal indicates an expected call of SetLocal. +func (mr *MockRuntimeStorageAPIMockRecorder) SetLocal(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetLocal", reflect.TypeOf((*MockRuntimeStorageAPI)(nil).SetLocal), arg0, arg1) +} + +// SetPersistent mocks base method. +func (m *MockRuntimeStorageAPI) SetPersistent(arg0, arg1 []byte) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "SetPersistent", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// SetPersistent indicates an expected call of SetPersistent. +func (mr *MockRuntimeStorageAPIMockRecorder) SetPersistent(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPersistent", reflect.TypeOf((*MockRuntimeStorageAPI)(nil).SetPersistent), arg0, arg1) +} + +// MockSyncStateAPI is a mock of SyncStateAPI interface. +type MockSyncStateAPI struct { + ctrl *gomock.Controller + recorder *MockSyncStateAPIMockRecorder +} + +// MockSyncStateAPIMockRecorder is the mock recorder for MockSyncStateAPI. +type MockSyncStateAPIMockRecorder struct { + mock *MockSyncStateAPI +} + +// NewMockSyncStateAPI creates a new mock instance. +func NewMockSyncStateAPI(ctrl *gomock.Controller) *MockSyncStateAPI { + mock := &MockSyncStateAPI{ctrl: ctrl} + mock.recorder = &MockSyncStateAPIMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockSyncStateAPI) EXPECT() *MockSyncStateAPIMockRecorder { + return m.recorder +} + +// GenSyncSpec mocks base method. +func (m *MockSyncStateAPI) GenSyncSpec(arg0 bool) (*genesis.Genesis, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GenSyncSpec", arg0) + ret0, _ := ret[0].(*genesis.Genesis) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GenSyncSpec indicates an expected call of GenSyncSpec. +func (mr *MockSyncStateAPIMockRecorder) GenSyncSpec(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GenSyncSpec", reflect.TypeOf((*MockSyncStateAPI)(nil).GenSyncSpec), arg0) +} diff --git a/dot/rpc/modules/mocks/network_api.go b/dot/rpc/modules/mocks/network_api.go deleted file mode 100644 index 7dd12bc265..0000000000 --- a/dot/rpc/modules/mocks/network_api.go +++ /dev/null @@ -1,168 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - common "github.com/ChainSafe/gossamer/lib/common" - mock "github.com/stretchr/testify/mock" -) - -// NetworkAPI is an autogenerated mock type for the NetworkAPI type -type NetworkAPI struct { - mock.Mock -} - -// AddReservedPeers provides a mock function with given fields: addrs -func (_m *NetworkAPI) AddReservedPeers(addrs ...string) error { - _va := make([]interface{}, len(addrs)) - for _i := range addrs { - _va[_i] = addrs[_i] - } - var _ca []interface{} - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(...string) error); ok { - r0 = rf(addrs...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// Health provides a mock function with given fields: -func (_m *NetworkAPI) Health() common.Health { - ret := _m.Called() - - var r0 common.Health - if rf, ok := ret.Get(0).(func() common.Health); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(common.Health) - } - - return r0 -} - -// NetworkState provides a mock function with given fields: -func (_m *NetworkAPI) NetworkState() common.NetworkState { - ret := _m.Called() - - var r0 common.NetworkState - if rf, ok := ret.Get(0).(func() common.NetworkState); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(common.NetworkState) - } - - return r0 -} - -// NodeRoles provides a mock function with given fields: -func (_m *NetworkAPI) NodeRoles() common.Roles { - ret := _m.Called() - - var r0 common.Roles - if rf, ok := ret.Get(0).(func() common.Roles); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(common.Roles) - } - - return r0 -} - -// Peers provides a mock function with given fields: -func (_m *NetworkAPI) Peers() []common.PeerInfo { - ret := _m.Called() - - var r0 []common.PeerInfo - if rf, ok := ret.Get(0).(func() []common.PeerInfo); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]common.PeerInfo) - } - } - - return r0 -} - -// RemoveReservedPeers provides a mock function with given fields: addrs -func (_m *NetworkAPI) RemoveReservedPeers(addrs ...string) error { - _va := make([]interface{}, len(addrs)) - for _i := range addrs { - _va[_i] = addrs[_i] - } - var _ca []interface{} - _ca = append(_ca, _va...) - ret := _m.Called(_ca...) - - var r0 error - if rf, ok := ret.Get(0).(func(...string) error); ok { - r0 = rf(addrs...) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// Start provides a mock function with given fields: -func (_m *NetworkAPI) Start() error { - ret := _m.Called() - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// StartingBlock provides a mock function with given fields: -func (_m *NetworkAPI) StartingBlock() int64 { - ret := _m.Called() - - var r0 int64 - if rf, ok := ret.Get(0).(func() int64); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(int64) - } - - return r0 -} - -// Stop provides a mock function with given fields: -func (_m *NetworkAPI) Stop() error { - ret := _m.Called() - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewNetworkAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewNetworkAPI creates a new instance of NetworkAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewNetworkAPI(t mockConstructorTestingTNewNetworkAPI) *NetworkAPI { - mock := &NetworkAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/runtime_storage_api.go b/dot/rpc/modules/mocks/runtime_storage_api.go deleted file mode 100644 index b47fcfade7..0000000000 --- a/dot/rpc/modules/mocks/runtime_storage_api.go +++ /dev/null @@ -1,99 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import mock "github.com/stretchr/testify/mock" - -// RuntimeStorageAPI is an autogenerated mock type for the RuntimeStorageAPI type -type RuntimeStorageAPI struct { - mock.Mock -} - -// GetLocal provides a mock function with given fields: k -func (_m *RuntimeStorageAPI) GetLocal(k []byte) ([]byte, error) { - ret := _m.Called(k) - - var r0 []byte - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(k) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(k) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetPersistent provides a mock function with given fields: k -func (_m *RuntimeStorageAPI) GetPersistent(k []byte) ([]byte, error) { - ret := _m.Called(k) - - var r0 []byte - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(k) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(k) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// SetLocal provides a mock function with given fields: k, v -func (_m *RuntimeStorageAPI) SetLocal(k []byte, v []byte) error { - ret := _m.Called(k, v) - - var r0 error - if rf, ok := ret.Get(0).(func([]byte, []byte) error); ok { - r0 = rf(k, v) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// SetPersistent provides a mock function with given fields: k, v -func (_m *RuntimeStorageAPI) SetPersistent(k []byte, v []byte) error { - ret := _m.Called(k, v) - - var r0 error - if rf, ok := ret.Get(0).(func([]byte, []byte) error); ok { - r0 = rf(k, v) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewRuntimeStorageAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewRuntimeStorageAPI creates a new instance of RuntimeStorageAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewRuntimeStorageAPI(t mockConstructorTestingTNewRuntimeStorageAPI) *RuntimeStorageAPI { - mock := &RuntimeStorageAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/storage_api.go b/dot/rpc/modules/mocks/storage_api.go deleted file mode 100644 index 2b749127f8..0000000000 --- a/dot/rpc/modules/mocks/storage_api.go +++ /dev/null @@ -1,203 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - common "github.com/ChainSafe/gossamer/lib/common" - mock "github.com/stretchr/testify/mock" - - state "github.com/ChainSafe/gossamer/dot/state" - - trie "github.com/ChainSafe/gossamer/lib/trie" -) - -// StorageAPI is an autogenerated mock type for the StorageAPI type -type StorageAPI struct { - mock.Mock -} - -// Entries provides a mock function with given fields: root -func (_m *StorageAPI) Entries(root *common.Hash) (map[string][]byte, error) { - ret := _m.Called(root) - - var r0 map[string][]byte - if rf, ok := ret.Get(0).(func(*common.Hash) map[string][]byte); ok { - r0 = rf(root) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(map[string][]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash) error); ok { - r1 = rf(root) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetKeysWithPrefix provides a mock function with given fields: root, prefix -func (_m *StorageAPI) GetKeysWithPrefix(root *common.Hash, prefix []byte) ([][]byte, error) { - ret := _m.Called(root, prefix) - - var r0 [][]byte - if rf, ok := ret.Get(0).(func(*common.Hash, []byte) [][]byte); ok { - r0 = rf(root, prefix) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([][]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash, []byte) error); ok { - r1 = rf(root, prefix) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetStateRootFromBlock provides a mock function with given fields: bhash -func (_m *StorageAPI) GetStateRootFromBlock(bhash *common.Hash) (*common.Hash, error) { - ret := _m.Called(bhash) - - var r0 *common.Hash - if rf, ok := ret.Get(0).(func(*common.Hash) *common.Hash); ok { - r0 = rf(bhash) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*common.Hash) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash) error); ok { - r1 = rf(bhash) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetStorage provides a mock function with given fields: root, key -func (_m *StorageAPI) GetStorage(root *common.Hash, key []byte) ([]byte, error) { - ret := _m.Called(root, key) - - var r0 []byte - if rf, ok := ret.Get(0).(func(*common.Hash, []byte) []byte); ok { - r0 = rf(root, key) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash, []byte) error); ok { - r1 = rf(root, key) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetStorageByBlockHash provides a mock function with given fields: bhash, key -func (_m *StorageAPI) GetStorageByBlockHash(bhash *common.Hash, key []byte) ([]byte, error) { - ret := _m.Called(bhash, key) - - var r0 []byte - if rf, ok := ret.Get(0).(func(*common.Hash, []byte) []byte); ok { - r0 = rf(bhash, key) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash, []byte) error); ok { - r1 = rf(bhash, key) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetStorageChild provides a mock function with given fields: root, keyToChild -func (_m *StorageAPI) GetStorageChild(root *common.Hash, keyToChild []byte) (*trie.Trie, error) { - ret := _m.Called(root, keyToChild) - - var r0 *trie.Trie - if rf, ok := ret.Get(0).(func(*common.Hash, []byte) *trie.Trie); ok { - r0 = rf(root, keyToChild) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*trie.Trie) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash, []byte) error); ok { - r1 = rf(root, keyToChild) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GetStorageFromChild provides a mock function with given fields: root, keyToChild, key -func (_m *StorageAPI) GetStorageFromChild(root *common.Hash, keyToChild []byte, key []byte) ([]byte, error) { - ret := _m.Called(root, keyToChild, key) - - var r0 []byte - if rf, ok := ret.Get(0).(func(*common.Hash, []byte, []byte) []byte); ok { - r0 = rf(root, keyToChild, key) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*common.Hash, []byte, []byte) error); ok { - r1 = rf(root, keyToChild, key) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RegisterStorageObserver provides a mock function with given fields: observer -func (_m *StorageAPI) RegisterStorageObserver(observer state.Observer) { - _m.Called(observer) -} - -// UnregisterStorageObserver provides a mock function with given fields: observer -func (_m *StorageAPI) UnregisterStorageObserver(observer state.Observer) { - _m.Called(observer) -} - -type mockConstructorTestingTNewStorageAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewStorageAPI creates a new instance of StorageAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewStorageAPI(t mockConstructorTestingTNewStorageAPI) *StorageAPI { - mock := &StorageAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/sync_state_api.go b/dot/rpc/modules/mocks/sync_state_api.go deleted file mode 100644 index a971b4ec32..0000000000 --- a/dot/rpc/modules/mocks/sync_state_api.go +++ /dev/null @@ -1,51 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - genesis "github.com/ChainSafe/gossamer/lib/genesis" - mock "github.com/stretchr/testify/mock" -) - -// SyncStateAPI is an autogenerated mock type for the SyncStateAPI type -type SyncStateAPI struct { - mock.Mock -} - -// GenSyncSpec provides a mock function with given fields: raw -func (_m *SyncStateAPI) GenSyncSpec(raw bool) (*genesis.Genesis, error) { - ret := _m.Called(raw) - - var r0 *genesis.Genesis - if rf, ok := ret.Get(0).(func(bool) *genesis.Genesis); ok { - r0 = rf(raw) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*genesis.Genesis) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(bool) error); ok { - r1 = rf(raw) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -type mockConstructorTestingTNewSyncStateAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewSyncStateAPI creates a new instance of SyncStateAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewSyncStateAPI(t mockConstructorTestingTNewSyncStateAPI) *SyncStateAPI { - mock := &SyncStateAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/system_api.go b/dot/rpc/modules/mocks/system_api.go deleted file mode 100644 index c60482a7f0..0000000000 --- a/dot/rpc/modules/mocks/system_api.go +++ /dev/null @@ -1,97 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import mock "github.com/stretchr/testify/mock" - -// SystemAPI is an autogenerated mock type for the SystemAPI type -type SystemAPI struct { - mock.Mock -} - -// ChainName provides a mock function with given fields: -func (_m *SystemAPI) ChainName() string { - ret := _m.Called() - - var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(string) - } - - return r0 -} - -// ChainType provides a mock function with given fields: -func (_m *SystemAPI) ChainType() string { - ret := _m.Called() - - var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(string) - } - - return r0 -} - -// Properties provides a mock function with given fields: -func (_m *SystemAPI) Properties() map[string]interface{} { - ret := _m.Called() - - var r0 map[string]interface{} - if rf, ok := ret.Get(0).(func() map[string]interface{}); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(map[string]interface{}) - } - } - - return r0 -} - -// SystemName provides a mock function with given fields: -func (_m *SystemAPI) SystemName() string { - ret := _m.Called() - - var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(string) - } - - return r0 -} - -// SystemVersion provides a mock function with given fields: -func (_m *SystemAPI) SystemVersion() string { - ret := _m.Called() - - var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(string) - } - - return r0 -} - -type mockConstructorTestingTNewSystemAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewSystemAPI creates a new instance of SystemAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewSystemAPI(t mockConstructorTestingTNewSystemAPI) *SystemAPI { - mock := &SystemAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks/transaction_state_api.go b/dot/rpc/modules/mocks/transaction_state_api.go deleted file mode 100644 index b13fdb0515..0000000000 --- a/dot/rpc/modules/mocks/transaction_state_api.go +++ /dev/null @@ -1,45 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - mock "github.com/stretchr/testify/mock" - - transaction "github.com/ChainSafe/gossamer/lib/transaction" -) - -// TransactionStateAPI is an autogenerated mock type for the TransactionStateAPI type -type TransactionStateAPI struct { - mock.Mock -} - -// Pending provides a mock function with given fields: -func (_m *TransactionStateAPI) Pending() []*transaction.ValidTransaction { - ret := _m.Called() - - var r0 []*transaction.ValidTransaction - if rf, ok := ret.Get(0).(func() []*transaction.ValidTransaction); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]*transaction.ValidTransaction) - } - } - - return r0 -} - -type mockConstructorTestingTNewTransactionStateAPI interface { - mock.TestingT - Cleanup(func()) -} - -// NewTransactionStateAPI creates a new instance of TransactionStateAPI. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewTransactionStateAPI(t mockConstructorTestingTNewTransactionStateAPI) *TransactionStateAPI { - mock := &TransactionStateAPI{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/rpc/modules/mocks_babe_test.go b/dot/rpc/modules/mocks_babe_test.go new file mode 100644 index 0000000000..e9f0d47fee --- /dev/null +++ b/dot/rpc/modules/mocks_babe_test.go @@ -0,0 +1,50 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/ChainSafe/gossamer/lib/babe (interfaces: BlockImportHandler) + +// Package modules is a generated GoMock package. +package modules + +import ( + reflect "reflect" + + types "github.com/ChainSafe/gossamer/dot/types" + storage "github.com/ChainSafe/gossamer/lib/runtime/storage" + gomock "github.com/golang/mock/gomock" +) + +// MockBlockImportHandler is a mock of BlockImportHandler interface. +type MockBlockImportHandler struct { + ctrl *gomock.Controller + recorder *MockBlockImportHandlerMockRecorder +} + +// MockBlockImportHandlerMockRecorder is the mock recorder for MockBlockImportHandler. +type MockBlockImportHandlerMockRecorder struct { + mock *MockBlockImportHandler +} + +// NewMockBlockImportHandler creates a new mock instance. +func NewMockBlockImportHandler(ctrl *gomock.Controller) *MockBlockImportHandler { + mock := &MockBlockImportHandler{ctrl: ctrl} + mock.recorder = &MockBlockImportHandlerMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockBlockImportHandler) EXPECT() *MockBlockImportHandlerMockRecorder { + return m.recorder +} + +// HandleBlockProduced mocks base method. +func (m *MockBlockImportHandler) HandleBlockProduced(arg0 *types.Block, arg1 *storage.TrieState) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "HandleBlockProduced", arg0, arg1) + ret0, _ := ret[0].(error) + return ret0 +} + +// HandleBlockProduced indicates an expected call of HandleBlockProduced. +func (mr *MockBlockImportHandlerMockRecorder) HandleBlockProduced(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleBlockProduced", reflect.TypeOf((*MockBlockImportHandler)(nil).HandleBlockProduced), arg0, arg1) +} diff --git a/dot/rpc/modules/mocks_generate_test.go b/dot/rpc/modules/mocks_generate_test.go index 6328d72e67..7e77782b14 100644 --- a/dot/rpc/modules/mocks_generate_test.go +++ b/dot/rpc/modules/mocks_generate_test.go @@ -4,3 +4,6 @@ package modules //go:generate mockgen -destination=mocks_test.go -package=$GOPACKAGE . StorageAPI,BlockAPI,Telemetry +//go:generate mockgen -destination=mocks/mocks.go -package mocks . StorageAPI,BlockAPI,NetworkAPI,BlockProducerAPI,TransactionStateAPI,CoreAPI,SystemAPI,BlockFinalityAPI,RuntimeStorageAPI,SyncStateAPI +//go:generate mockgen -destination=mock_sync_api_test.go -package $GOPACKAGE . SyncAPI +//go:generate mockgen -destination=mocks_babe_test.go -package $GOPACKAGE github.com/ChainSafe/gossamer/lib/babe BlockImportHandler diff --git a/dot/rpc/modules/offchain_integration_test.go b/dot/rpc/modules/offchain_integration_test.go index 3cc7514277..ffdac98cfb 100644 --- a/dot/rpc/modules/offchain_integration_test.go +++ b/dot/rpc/modules/offchain_integration_test.go @@ -11,42 +11,76 @@ import ( "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" "github.com/ChainSafe/gossamer/lib/common" - "github.com/stretchr/testify/mock" + "github.com/golang/mock/gomock" + "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) -func TestOffchainStorageGet(t *testing.T) { - testFuncs := map[string]string{ - offchainLocal: "GetLocal", - offchainPersistent: "GetPersistent", - } +func Test_OffchainModule_LocalStorageGet(t *testing.T) { + t.Run("get local error", func(t *testing.T) { + ctrl := gomock.NewController(t) + + runtimeStorage := mocks.NewMockRuntimeStorageAPI(ctrl) + offchainModule := &OffchainModule{ + nodeStorage: runtimeStorage, + } - for kind, test := range testFuncs { - expectedValue := common.BytesToHex([]byte("some-value")) - st := mocks.NewRuntimeStorageAPI(t) - st.On(test, mock.AnythingOfType("[]uint8")).Return([]byte("some-value"), nil).Once() + const keyHex = "0x11111111111111" + request := &OffchainLocalStorageGet{ + Kind: offchainLocal, + Key: keyHex, + } + errTest := errors.New("test error") + runtimeStorage.EXPECT().GetLocal(common.MustHexToBytes(keyHex)). + Return(nil, errTest) + + err := offchainModule.LocalStorageGet(nil, request, nil) + assert.ErrorIs(t, err, errTest) + }) - m := new(OffchainModule) - m.nodeStorage = st + t.Run("local kind", func(t *testing.T) { + ctrl := gomock.NewController(t) - req := &OffchainLocalStorageGet{ - Kind: kind, - Key: "0x11111111111111", + runtimeStorage := mocks.NewMockRuntimeStorageAPI(ctrl) + offchainModule := &OffchainModule{ + nodeStorage: runtimeStorage, } - var res StringResponse - err := m.LocalStorageGet(nil, req, &res) + const keyHex = "0x11111111111111" + request := &OffchainLocalStorageGet{ + Kind: offchainLocal, + Key: keyHex, + } + runtimeStorage.EXPECT().GetLocal(common.MustHexToBytes(keyHex)). + Return([]byte("some-value"), nil) + var response StringResponse + err := offchainModule.LocalStorageGet(nil, request, &response) require.NoError(t, err) - require.Equal(t, res, StringResponse(expectedValue)) + expectedResponse := StringResponse(common.BytesToHex([]byte("some-value"))) + assert.Equal(t, response, expectedResponse) + }) - st. - On(test, mock.AnythingOfType("[]uint8")). - Return(nil, errors.New("problem to retrieve")). - Once() + t.Run("persistent kind", func(t *testing.T) { + ctrl := gomock.NewController(t) - err = m.LocalStorageGet(nil, req, nil) - require.Error(t, err, "problem to retrieve") - } + runtimeStorage := mocks.NewMockRuntimeStorageAPI(ctrl) + offchainModule := &OffchainModule{ + nodeStorage: runtimeStorage, + } + + const keyHex = "0x11111111111111" + request := &OffchainLocalStorageGet{ + Kind: offchainPersistent, + Key: keyHex, + } + runtimeStorage.EXPECT().GetPersistent(common.MustHexToBytes(keyHex)). + Return([]byte("some-value"), nil) + var response StringResponse + err := offchainModule.LocalStorageGet(nil, request, &response) + require.NoError(t, err) + expectedResponse := StringResponse(common.BytesToHex([]byte("some-value"))) + assert.Equal(t, response, expectedResponse) + }) } func TestOffchainStorage_OtherKind(t *testing.T) { @@ -67,37 +101,72 @@ func TestOffchainStorage_OtherKind(t *testing.T) { require.Error(t, err, "storage kind not found: another kind") } -func TestOffchainStorageSet(t *testing.T) { - testFuncs := map[string]string{ - offchainLocal: "SetLocal", - offchainPersistent: "SetPersistent", - } +func Test_OffchainModule_LocalStorageSet(t *testing.T) { + const keyHex, valueHex = "0x11111111111111", "0x22222222222222" - for kind, test := range testFuncs { - st := mocks.NewRuntimeStorageAPI(t) - st.On(test, mock.AnythingOfType("[]uint8"), mock.AnythingOfType("[]uint8")).Return(nil).Once() + t.Run("set local error", func(t *testing.T) { + ctrl := gomock.NewController(t) - m := new(OffchainModule) - m.nodeStorage = st + runtimeStorage := mocks.NewMockRuntimeStorageAPI(ctrl) + offchainModule := &OffchainModule{ + nodeStorage: runtimeStorage, + } + + request := &OffchainLocalStorageSet{ + Kind: offchainLocal, + Key: keyHex, + Value: valueHex, + } + errTest := errors.New("test error") + runtimeStorage.EXPECT().SetLocal( + common.MustHexToBytes(keyHex), common.MustHexToBytes(valueHex)). + Return(errTest) - req := &OffchainLocalStorageSet{ - Kind: kind, - Key: "0x11111111111111", - Value: "0x22222222222222", + err := offchainModule.LocalStorageSet(nil, request, nil) + assert.ErrorIs(t, err, errTest) + }) + + t.Run("local kind", func(t *testing.T) { + ctrl := gomock.NewController(t) + + runtimeStorage := mocks.NewMockRuntimeStorageAPI(ctrl) + offchainModule := &OffchainModule{ + nodeStorage: runtimeStorage, } - var res StringResponse - err := m.LocalStorageSet(nil, req, &res) + request := &OffchainLocalStorageSet{ + Kind: offchainLocal, + Key: keyHex, + Value: valueHex, + } + runtimeStorage.EXPECT().SetLocal( + common.MustHexToBytes(keyHex), common.MustHexToBytes(valueHex)). + Return(nil) + var response StringResponse + err := offchainModule.LocalStorageSet(nil, request, &response) require.NoError(t, err) - require.Empty(t, res) + assert.Empty(t, response) + }) - st. - On(test, mock.AnythingOfType("[]uint8"), mock.AnythingOfType("[]uint8")). - Return(errors.New("problem to store")). - Once() + t.Run("persistent kind", func(t *testing.T) { + ctrl := gomock.NewController(t) - err = m.LocalStorageSet(nil, req, &res) - require.Error(t, err, "problem to store") - require.Empty(t, res) - } + runtimeStorage := mocks.NewMockRuntimeStorageAPI(ctrl) + offchainModule := &OffchainModule{ + nodeStorage: runtimeStorage, + } + + request := &OffchainLocalStorageSet{ + Kind: offchainPersistent, + Key: keyHex, + Value: valueHex, + } + runtimeStorage.EXPECT().SetPersistent( + common.MustHexToBytes(keyHex), common.MustHexToBytes(valueHex)). + Return(nil) + var response StringResponse + err := offchainModule.LocalStorageSet(nil, request, &response) + require.NoError(t, err) + assert.Empty(t, response) + }) } diff --git a/dot/rpc/modules/offchain_test.go b/dot/rpc/modules/offchain_test.go index dfaaa324ad..83edcd5118 100644 --- a/dot/rpc/modules/offchain_test.go +++ b/dot/rpc/modules/offchain_test.go @@ -11,15 +11,18 @@ import ( "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" "github.com/ChainSafe/gossamer/lib/common" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" ) func TestOffchainModule_LocalStorageGet(t *testing.T) { - mockRuntimeStorageAPI := mocks.NewRuntimeStorageAPI(t) - mockRuntimeStorageAPI.On("GetPersistent", common.MustHexToBytes("0x11111111111111")). + ctrl := gomock.NewController(t) + + mockRuntimeStorageAPI := mocks.NewMockRuntimeStorageAPI(ctrl) + mockRuntimeStorageAPI.EXPECT().GetPersistent(common.MustHexToBytes("0x11111111111111")). Return(nil, errors.New("GetPersistent error")) - mockRuntimeStorageAPI.On("GetLocal", common.MustHexToBytes("0x11111111111111")).Return([]byte("some-value"), nil) + mockRuntimeStorageAPI.EXPECT().GetLocal(common.MustHexToBytes("0x11111111111111")).Return([]byte("some-value"), nil) offChainModule := NewOffchainModule(mockRuntimeStorageAPI) type fields struct { @@ -107,11 +110,13 @@ func TestOffchainModule_LocalStorageGet(t *testing.T) { } func TestOffchainModule_LocalStorageSet(t *testing.T) { - mockRuntimeStorageAPI := mocks.NewRuntimeStorageAPI(t) - mockRuntimeStorageAPI.On("SetLocal", + ctrl := gomock.NewController(t) + + mockRuntimeStorageAPI := mocks.NewMockRuntimeStorageAPI(ctrl) + mockRuntimeStorageAPI.EXPECT().SetLocal( common.MustHexToBytes("0x11111111111111"), common.MustHexToBytes("0x22222222222222")). Return(nil) - mockRuntimeStorageAPI.On("SetPersistent", + mockRuntimeStorageAPI.EXPECT().SetPersistent( common.MustHexToBytes("0x11111111111111"), common.MustHexToBytes("0x22222222222222")). Return(errors.New("SetPersistent error")) diff --git a/dot/rpc/modules/payment_integration_test.go b/dot/rpc/modules/payment_integration_test.go index 2aacc59f82..731f7b72cc 100644 --- a/dot/rpc/modules/payment_integration_test.go +++ b/dot/rpc/modules/payment_integration_test.go @@ -12,7 +12,7 @@ import ( "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" "github.com/ChainSafe/gossamer/dot/types" "github.com/ChainSafe/gossamer/pkg/scale" - "github.com/stretchr/testify/mock" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/require" "github.com/ChainSafe/gossamer/lib/common" @@ -24,6 +24,8 @@ func TestPaymentQueryInfo(t *testing.T) { bestBlockHash := state.Block.BestBlockHash() t.Run("When there is no errors", func(t *testing.T) { + ctrl := gomock.NewController(t) + mockedQueryInfo := &types.RuntimeDispatchInfo{ Weight: 0, Class: 0, @@ -36,13 +38,13 @@ func TestPaymentQueryInfo(t *testing.T) { PartialFee: scale.MaxUint128.String(), } - runtimeMock := mocksruntime.NewInstance(t) - runtimeMock.On("PaymentQueryInfo", mock.AnythingOfType("[]uint8")).Return(mockedQueryInfo, nil) + runtimeMock := mocksruntime.NewMockInstance(ctrl) + runtimeMock.EXPECT().PaymentQueryInfo(gomock.Any()).Return(mockedQueryInfo, nil) - blockAPIMock := mocks.NewBlockAPI(t) - blockAPIMock.On("BestBlockHash").Return(bestBlockHash) + blockAPIMock := mocks.NewMockBlockAPI(ctrl) + blockAPIMock.EXPECT().BestBlockHash().Return(bestBlockHash) - blockAPIMock.On("GetRuntime", bestBlockHash).Return(runtimeMock, nil) + blockAPIMock.EXPECT().GetRuntime(bestBlockHash).Return(runtimeMock, nil) mod := &PaymentModule{ blockAPI: blockAPIMock, @@ -60,10 +62,12 @@ func TestPaymentQueryInfo(t *testing.T) { }) t.Run("When could not get runtime", func(t *testing.T) { - blockAPIMock := mocks.NewBlockAPI(t) - blockAPIMock.On("BestBlockHash").Return(bestBlockHash) + ctrl := gomock.NewController(t) + + blockAPIMock := mocks.NewMockBlockAPI(ctrl) + blockAPIMock.EXPECT().BestBlockHash().Return(bestBlockHash) - blockAPIMock.On("GetRuntime", bestBlockHash). + blockAPIMock.EXPECT().GetRuntime(bestBlockHash). Return(nil, errors.New("mocked problems")) mod := &PaymentModule{ @@ -82,11 +86,13 @@ func TestPaymentQueryInfo(t *testing.T) { }) t.Run("When PaymentQueryInfo returns error", func(t *testing.T) { - runtimeMock := mocksruntime.NewInstance(t) - runtimeMock.On("PaymentQueryInfo", mock.AnythingOfType("[]uint8")).Return(nil, errors.New("mocked error")) + ctrl := gomock.NewController(t) - blockAPIMock := mocks.NewBlockAPI(t) - blockAPIMock.On("GetRuntime", common.Hash{1, 2}).Return(runtimeMock, nil) + runtimeMock := mocksruntime.NewMockInstance(ctrl) + runtimeMock.EXPECT().PaymentQueryInfo(gomock.Any()).Return(nil, errors.New("mocked error")) + + blockAPIMock := mocks.NewMockBlockAPI(ctrl) + blockAPIMock.EXPECT().GetRuntime(common.Hash{1, 2}).Return(runtimeMock, nil) mod := &PaymentModule{ blockAPI: blockAPIMock, @@ -105,11 +111,13 @@ func TestPaymentQueryInfo(t *testing.T) { }) t.Run("When PaymentQueryInfo returns a nil info", func(t *testing.T) { - runtimeMock := mocksruntime.NewInstance(t) - runtimeMock.On("PaymentQueryInfo", mock.AnythingOfType("[]uint8")).Return(nil, nil) + ctrl := gomock.NewController(t) + + runtimeMock := mocksruntime.NewMockInstance(ctrl) + runtimeMock.EXPECT().PaymentQueryInfo(gomock.Any()).Return(nil, nil) - blockAPIMock := mocks.NewBlockAPI(t) - blockAPIMock.On("GetRuntime", common.Hash{1, 2}).Return(runtimeMock, nil) + blockAPIMock := mocks.NewMockBlockAPI(ctrl) + blockAPIMock.EXPECT().GetRuntime(common.Hash{1, 2}).Return(runtimeMock, nil) mod := &PaymentModule{ blockAPI: blockAPIMock, diff --git a/dot/rpc/modules/payment_test.go b/dot/rpc/modules/payment_test.go index e0427990be..6d44000d85 100644 --- a/dot/rpc/modules/payment_test.go +++ b/dot/rpc/modules/payment_test.go @@ -14,41 +14,44 @@ import ( "github.com/ChainSafe/gossamer/lib/common" mocksruntime "github.com/ChainSafe/gossamer/lib/runtime/mocks" "github.com/ChainSafe/gossamer/pkg/scale" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestPaymentModule_QueryInfo(t *testing.T) { + ctrl := gomock.NewController(t) + testHash := common.NewHash([]byte{0x01, 0x02}) u, err := scale.NewUint128(new(big.Int).SetBytes([]byte{1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6})) require.NoError(t, err) - runtimeMock := mocksruntime.NewInstance(t) - runtimeMock2 := mocksruntime.NewInstance(t) - runtimeErrorMock := mocksruntime.NewInstance(t) + runtimeMock := mocksruntime.NewMockInstance(ctrl) + runtimeMock2 := mocksruntime.NewMockInstance(ctrl) + runtimeErrorMock := mocksruntime.NewMockInstance(ctrl) - blockAPIMock := mocks.NewBlockAPI(t) - blockAPIMock2 := mocks.NewBlockAPI(t) - blockErrorAPIMock1 := mocks.NewBlockAPI(t) - blockErrorAPIMock2 := mocks.NewBlockAPI(t) + blockAPIMock := mocks.NewMockBlockAPI(ctrl) + blockAPIMock2 := mocks.NewMockBlockAPI(ctrl) + blockErrorAPIMock1 := mocks.NewMockBlockAPI(ctrl) + blockErrorAPIMock2 := mocks.NewMockBlockAPI(ctrl) - blockAPIMock.On("BestBlockHash").Return(testHash, nil) - blockAPIMock.On("GetRuntime", testHash).Return(runtimeMock, nil) + blockAPIMock.EXPECT().BestBlockHash().Return(testHash).AnyTimes() + blockAPIMock.EXPECT().GetRuntime(testHash).Return(runtimeMock, nil).AnyTimes() - blockAPIMock2.On("GetRuntime", testHash).Return(runtimeMock2, nil) + blockAPIMock2.EXPECT().GetRuntime(testHash).Return(runtimeMock2, nil) - blockErrorAPIMock1.On("GetRuntime", testHash).Return(runtimeErrorMock, nil) + blockErrorAPIMock1.EXPECT().GetRuntime(testHash).Return(runtimeErrorMock, nil) - blockErrorAPIMock2.On("GetRuntime", testHash).Return(nil, errors.New("GetRuntime error")) + blockErrorAPIMock2.EXPECT().GetRuntime(testHash).Return(nil, errors.New("GetRuntime error")) - runtimeMock.On("PaymentQueryInfo", common.MustHexToBytes("0x0000")).Return(nil, nil) - runtimeMock2.On("PaymentQueryInfo", common.MustHexToBytes("0x0000")).Return(&types.RuntimeDispatchInfo{ + runtimeMock.EXPECT().PaymentQueryInfo(common.MustHexToBytes("0x0000")).Return(nil, nil).AnyTimes() + runtimeMock2.EXPECT().PaymentQueryInfo(common.MustHexToBytes("0x0000")).Return(&types.RuntimeDispatchInfo{ Weight: uint64(21), Class: 21, PartialFee: u, }, nil) - runtimeErrorMock.On("PaymentQueryInfo", common.MustHexToBytes("0x0000")). + runtimeErrorMock.EXPECT().PaymentQueryInfo(common.MustHexToBytes("0x0000")). Return(nil, errors.New("PaymentQueryInfo error")) paymentModule := NewPaymentModule(blockAPIMock) diff --git a/dot/rpc/modules/state_integration_test.go b/dot/rpc/modules/state_integration_test.go index 372764531f..2cf8383310 100644 --- a/dot/rpc/modules/state_integration_test.go +++ b/dot/rpc/modules/state_integration_test.go @@ -19,7 +19,6 @@ import ( "github.com/ChainSafe/gossamer/lib/common" "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) @@ -491,9 +490,10 @@ func TestStateModule_GetKeysPaged(t *testing.T) { } func TestGetReadProof_WhenCoreAPIReturnsError(t *testing.T) { - coreAPIMock := mocks.NewCoreAPI(t) - coreAPIMock. - On("GetReadProofAt", mock.AnythingOfType("common.Hash"), mock.AnythingOfType("[][]uint8")). + ctrl := gomock.NewController(t) + + coreAPIMock := mocks.NewMockCoreAPI(ctrl) + coreAPIMock.EXPECT().GetReadProofAt(gomock.Any(), gomock.Any()). Return(common.Hash{}, nil, errors.New("mocked error")) sm := new(StateModule) @@ -507,12 +507,14 @@ func TestGetReadProof_WhenCoreAPIReturnsError(t *testing.T) { } func TestGetReadProof_WhenReturnsProof(t *testing.T) { + ctrl := gomock.NewController(t) + expectedBlock := common.BytesToHash([]byte("random hash")) mockedProof := [][]byte{[]byte("proof-1"), []byte("proof-2")} - coreAPIMock := mocks.NewCoreAPI(t) + coreAPIMock := mocks.NewMockCoreAPI(ctrl) coreAPIMock. - On("GetReadProofAt", mock.AnythingOfType("common.Hash"), mock.AnythingOfType("[][]uint8")). + EXPECT().GetReadProofAt(gomock.Any(), gomock.Any()). Return(expectedBlock, mockedProof, nil) sm := new(StateModule) diff --git a/dot/rpc/modules/state_test.go b/dot/rpc/modules/state_test.go index 11c04a7245..ffdb0f3a20 100644 --- a/dot/rpc/modules/state_test.go +++ b/dot/rpc/modules/state_test.go @@ -32,41 +32,44 @@ import ( ) func TestStateModuleGetPairs(t *testing.T) { + ctrl := gomock.NewController(t) + str := "0x01" hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") m := make(map[string][]byte) m["a"] = []byte{21, 22} m["b"] = []byte{23, 24} - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("GetStateRootFromBlock", &hash).Return(&hash, nil) - mockStorageAPI.On("GetKeysWithPrefix", &hash, common.MustHexToBytes(str)).Return([][]byte{{1}, {1}}, nil) - mockStorageAPI.On("GetStorage", &hash, []byte{1}).Return([]byte{21}, nil) + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().GetStateRootFromBlock(&hash).Return(&hash, nil) + mockStorageAPI.EXPECT().GetKeysWithPrefix(&hash, common.MustHexToBytes(str)).Return([][]byte{{1}, {1}}, nil) + mockStorageAPI.EXPECT().GetStorage(&hash, []byte{1}).Return([]byte{21}, nil).AnyTimes() - mockStorageAPINil := mocks.NewStorageAPI(t) - mockStorageAPINil.On("GetStateRootFromBlock", &hash).Return(&hash, nil) - mockStorageAPINil.On("Entries", &hash).Return(m, nil) + mockStorageAPINil := mocks.NewMockStorageAPI(ctrl) + mockStorageAPINil.EXPECT().GetStateRootFromBlock(&hash).Return(&hash, nil) + mockStorageAPINil.EXPECT().Entries(&hash).Return(m, nil) - mockStorageAPIGetKeysEmpty := mocks.NewStorageAPI(t) - mockStorageAPIGetKeysEmpty.On("GetStateRootFromBlock", &hash).Return(&hash, nil) - mockStorageAPIGetKeysEmpty.On("GetKeysWithPrefix", &hash, common.MustHexToBytes(str)).Return([][]byte{}, nil) + mockStorageAPIGetKeysEmpty := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIGetKeysEmpty.EXPECT().GetStateRootFromBlock(&hash).Return(&hash, nil) + mockStorageAPIGetKeysEmpty.EXPECT().GetKeysWithPrefix(&hash, common.MustHexToBytes(str)).Return([][]byte{}, nil) - mockStorageAPIGetKeysErr := mocks.NewStorageAPI(t) - mockStorageAPIGetKeysErr.On("GetStateRootFromBlock", &hash).Return(&hash, nil) - mockStorageAPIGetKeysErr.On("GetKeysWithPrefix", &hash, common.MustHexToBytes(str)). + mockStorageAPIGetKeysErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIGetKeysErr.EXPECT().GetStateRootFromBlock(&hash).Return(&hash, nil) + mockStorageAPIGetKeysErr.EXPECT().GetKeysWithPrefix(&hash, common.MustHexToBytes(str)). Return(nil, errors.New("GetKeysWithPrefix Err")) - mockStorageAPIEntriesErr := mocks.NewStorageAPI(t) - mockStorageAPIEntriesErr.On("GetStateRootFromBlock", &hash).Return(&hash, nil) - mockStorageAPIEntriesErr.On("Entries", &hash).Return(nil, errors.New("entries Err")) + mockStorageAPIEntriesErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIEntriesErr.EXPECT().GetStateRootFromBlock(&hash).Return(&hash, nil) + mockStorageAPIEntriesErr.EXPECT().Entries(&hash).Return(nil, errors.New("entries Err")) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("GetStateRootFromBlock", &hash).Return(nil, errors.New("GetStateRootFromBlock Err")) + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().GetStateRootFromBlock(&hash).Return(nil, errors.New("GetStateRootFromBlock Err")) - mockStorageAPIGetStorageErr := mocks.NewStorageAPI(t) - mockStorageAPIGetStorageErr.On("GetStateRootFromBlock", &hash).Return(&hash, nil) - mockStorageAPIGetStorageErr.On("GetKeysWithPrefix", &hash, common.MustHexToBytes(str)).Return([][]byte{{2}, {2}}, nil) - mockStorageAPIGetStorageErr.On("GetStorage", &hash, []byte{2}).Return(nil, errors.New("GetStorage Err")) + mockStorageAPIGetStorageErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIGetStorageErr.EXPECT().GetStateRootFromBlock(&hash).Return(&hash, nil) + mockStorageAPIGetStorageErr.EXPECT().GetKeysWithPrefix(&hash, common.MustHexToBytes(str)). + Return([][]byte{{2}, {2}}, nil) + mockStorageAPIGetStorageErr.EXPECT().GetStorage(&hash, []byte{2}).Return(nil, errors.New("GetStorage Err")) var expRes StatePairResponse for k, v := range m { @@ -185,16 +188,18 @@ func TestStateModuleGetPairs(t *testing.T) { } func TestStateModuleGetKeysPaged(t *testing.T) { - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("GetKeysWithPrefix", (*common.Hash)(nil), common.MustHexToBytes("0x")). + ctrl := gomock.NewController(t) + + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().GetKeysWithPrefix((*common.Hash)(nil), common.MustHexToBytes("0x")). Return([][]byte{{1}, {2}}, nil) - mockStorageAPI2 := mocks.NewStorageAPI(t) - mockStorageAPI2.On("GetKeysWithPrefix", (*common.Hash)(nil), common.MustHexToBytes("0x")). + mockStorageAPI2 := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI2.EXPECT().GetKeysWithPrefix((*common.Hash)(nil), common.MustHexToBytes("0x")). Return([][]byte{{1, 1, 1}, {1, 1, 1}}, nil) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("GetKeysWithPrefix", (*common.Hash)(nil), common.MustHexToBytes("0x")). + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().GetKeysWithPrefix((*common.Hash)(nil), common.MustHexToBytes("0x")). Return(nil, errors.New("GetKeysWithPrefix Err")) type fields struct { @@ -277,8 +282,10 @@ func TestStateModuleGetKeysPaged(t *testing.T) { // Implement Tests once function is implemented func TestCall(t *testing.T) { - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockStorageAPI := mocks.NewStorageAPI(t) + ctrl := gomock.NewController(t) + + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) sm := NewStateModule(mockNetworkAPI, mockStorageAPI, nil, nil) err := sm.Call(nil, nil, nil) @@ -286,13 +293,15 @@ func TestCall(t *testing.T) { } func TestStateModuleGetMetadata(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockCoreAPI := mocks.NewCoreAPI(t) - mockCoreAPI.On("GetMetadata", &hash).Return(common.MustHexToBytes(testdata.NewTestMetadata()), nil) + mockCoreAPI := mocks.NewMockCoreAPI(ctrl) + mockCoreAPI.EXPECT().GetMetadata(&hash).Return(common.MustHexToBytes(testdata.NewTestMetadata()), nil) - mockCoreAPIErr := mocks.NewCoreAPI(t) - mockCoreAPIErr.On("GetMetadata", &hash).Return(nil, errors.New("GetMetadata Error")) + mockCoreAPIErr := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIErr.EXPECT().GetMetadata(&hash).Return(nil, errors.New("GetMetadata Error")) mockStateModule := NewStateModule(nil, nil, mockCoreAPIErr, nil) @@ -352,6 +361,8 @@ func TestStateModuleGetMetadata(t *testing.T) { } func TestStateModuleGetReadProof(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") keys := []string{"0x1111", "0x2222"} expKeys := make([][]byte, len(keys)) @@ -362,11 +373,12 @@ func TestStateModuleGetReadProof(t *testing.T) { expKeys[i] = bKey } - mockCoreAPI := mocks.NewCoreAPI(t) - mockCoreAPI.On("GetReadProofAt", hash, expKeys).Return(hash, [][]byte{{1, 1, 1}, {1, 1, 1}}, nil) + mockCoreAPI := mocks.NewMockCoreAPI(ctrl) + mockCoreAPI.EXPECT().GetReadProofAt(hash, expKeys).Return(hash, [][]byte{{1, 1, 1}, {1, 1, 1}}, nil) - mockCoreAPIErr := mocks.NewCoreAPI(t) - mockCoreAPIErr.On("GetReadProofAt", hash, expKeys).Return(nil, nil, errors.New("GetReadProofAt Error")) + mockCoreAPIErr := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIErr.EXPECT().GetReadProofAt(hash, expKeys). + Return(common.Hash{}, nil, errors.New("GetReadProofAt Error")).Times(2) type fields struct { networkAPI NetworkAPI @@ -443,6 +455,8 @@ func TestStateModuleGetReadProof(t *testing.T) { } func TestStateModuleGetRuntimeVersion(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") version := runtime.Version{ SpecName: []byte("polkadot"), @@ -457,11 +471,11 @@ func TestStateModuleGetRuntimeVersion(t *testing.T) { TransactionVersion: 5, } - mockCoreAPI := mocks.NewCoreAPI(t) - mockCoreAPI.On("GetRuntimeVersion", &hash).Return(version, nil) + mockCoreAPI := mocks.NewMockCoreAPI(ctrl) + mockCoreAPI.EXPECT().GetRuntimeVersion(&hash).Return(version, nil) - mockCoreAPIErr := mocks.NewCoreAPI(t) - mockCoreAPIErr.On("GetRuntimeVersion", &hash). + mockCoreAPIErr := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIErr.EXPECT().GetRuntimeVersion(&hash). Return(runtime.Version{}, errors.New("GetRuntimeVersion Error")) type fields struct { @@ -524,16 +538,20 @@ func TestStateModuleGetRuntimeVersion(t *testing.T) { } func TestStateModuleGetStorage(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") reqBytes := common.MustHexToBytes("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("GetStorageByBlockHash", &hash, reqBytes).Return([]byte{21}, nil) - mockStorageAPI.On("GetStorage", (*common.Hash)(nil), reqBytes).Return([]byte{21}, nil) + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().GetStorageByBlockHash(&hash, reqBytes).Return([]byte{21}, nil) + mockStorageAPI.EXPECT().GetStorage((*common.Hash)(nil), reqBytes).Return([]byte{21}, nil) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("GetStorageByBlockHash", &hash, reqBytes).Return(nil, errors.New("GetStorageByBlockHash Error")) - mockStorageAPIErr.On("GetStorage", (*common.Hash)(nil), reqBytes).Return(nil, errors.New("GetStorage Error")) + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().GetStorageByBlockHash(&hash, reqBytes). + Return(nil, errors.New("GetStorageByBlockHash Error")) + mockStorageAPIErr.EXPECT().GetStorage((*common.Hash)(nil), reqBytes). + Return(nil, errors.New("GetStorage Error")) type fields struct { networkAPI NetworkAPI @@ -614,16 +632,21 @@ func TestStateModuleGetStorage(t *testing.T) { } func TestStateModuleGetStorageHash(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") reqBytes := common.MustHexToBytes("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("GetStorageByBlockHash", &hash, reqBytes).Return([]byte{21}, nil) - mockStorageAPI.On("GetStorage", (*common.Hash)(nil), reqBytes).Return([]byte{21}, nil) + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().GetStorageByBlockHash(&hash, reqBytes).Return([]byte{21}, nil) + mockStorageAPI.EXPECT().GetStorage((*common.Hash)(nil), reqBytes). + Return([]byte{21}, nil) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("GetStorageByBlockHash", &hash, reqBytes).Return(nil, errors.New("GetStorageByBlockHash Error")) - mockStorageAPIErr.On("GetStorage", (*common.Hash)(nil), reqBytes).Return(nil, errors.New("GetStorage Error")) + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().GetStorageByBlockHash(&hash, reqBytes). + Return(nil, errors.New("GetStorageByBlockHash Error")) + mockStorageAPIErr.EXPECT().GetStorage((*common.Hash)(nil), reqBytes). + Return(nil, errors.New("GetStorage Error")) type fields struct { networkAPI NetworkAPI @@ -704,16 +727,20 @@ func TestStateModuleGetStorageHash(t *testing.T) { } func TestStateModuleGetStorageSize(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") reqBytes := common.MustHexToBytes("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("GetStorageByBlockHash", &hash, reqBytes).Return([]byte{21}, nil) - mockStorageAPI.On("GetStorage", (*common.Hash)(nil), reqBytes).Return([]byte{21}, nil) + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().GetStorageByBlockHash(&hash, reqBytes).Return([]byte{21}, nil) + mockStorageAPI.EXPECT().GetStorage((*common.Hash)(nil), reqBytes).Return([]byte{21}, nil) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("GetStorageByBlockHash", &hash, reqBytes).Return(nil, errors.New("GetStorageByBlockHash Error")) - mockStorageAPIErr.On("GetStorage", (*common.Hash)(nil), reqBytes).Return(nil, errors.New("GetStorage Error")) + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().GetStorageByBlockHash(&hash, reqBytes). + Return(nil, errors.New("GetStorageByBlockHash Error")) + mockStorageAPIErr.EXPECT().GetStorage((*common.Hash)(nil), reqBytes). + Return(nil, errors.New("GetStorage Error")) type fields struct { networkAPI NetworkAPI diff --git a/dot/rpc/modules/sync_state_test.go b/dot/rpc/modules/sync_state_test.go index 58eb558d09..7d8e804dc3 100644 --- a/dot/rpc/modules/sync_state_test.go +++ b/dot/rpc/modules/sync_state_test.go @@ -11,17 +11,20 @@ import ( "github.com/ChainSafe/gossamer/dot/rpc/modules/mocks" "github.com/ChainSafe/gossamer/lib/common" "github.com/ChainSafe/gossamer/lib/genesis" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" ) func TestSyncStateModule_GenSyncSpec(t *testing.T) { + ctrl := gomock.NewController(t) + g := new(genesis.Genesis) - mockSyncStateAPI := mocks.NewSyncStateAPI(t) - mockSyncStateAPI.On("GenSyncSpec", true).Return(g, nil) + mockSyncStateAPI := mocks.NewMockSyncStateAPI(ctrl) + mockSyncStateAPI.EXPECT().GenSyncSpec(true).Return(g, nil) - mockSyncStateAPIErr := mocks.NewSyncStateAPI(t) - mockSyncStateAPIErr.On("GenSyncSpec", true).Return(nil, errors.New("GenSyncSpec error")) + mockSyncStateAPIErr := mocks.NewMockSyncStateAPI(ctrl) + mockSyncStateAPIErr.EXPECT().GenSyncSpec(true).Return(nil, errors.New("GenSyncSpec error")) syncStateModule := NewSyncStateModule(mockSyncStateAPI) type fields struct { @@ -81,14 +84,16 @@ func TestSyncStateModule_GenSyncSpec(t *testing.T) { } func TestNewStateSync(t *testing.T) { + ctrl := gomock.NewController(t) + g1 := &genesis.Genesis{} g2 := &genesis.Genesis{} raw := make(map[string][]byte) - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("Entries", (*common.Hash)(nil)).Return(raw, nil) + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().Entries((*common.Hash)(nil)).Return(raw, nil) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("Entries", (*common.Hash)(nil)).Return(nil, errors.New("entries error")) + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().Entries((*common.Hash)(nil)).Return(nil, errors.New("entries error")) type args struct { gData *genesis.Data diff --git a/dot/rpc/modules/system_integration_test.go b/dot/rpc/modules/system_integration_test.go index 23c03740fd..3f2be61d69 100644 --- a/dot/rpc/modules/system_integration_test.go +++ b/dot/rpc/modules/system_integration_test.go @@ -16,7 +16,6 @@ import ( "github.com/btcsuite/btcd/btcutil/base58" "github.com/golang/mock/gomock" "github.com/multiformats/go-multiaddr" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" "github.com/ChainSafe/gossamer/dot/core" @@ -92,8 +91,10 @@ func newNetworkService(t *testing.T) *network.Service { // Test RPC's System.Health() response func TestSystemModule_Health(t *testing.T) { - networkMock := mocks.NewNetworkAPI(t) - networkMock.On("Health").Return(testHealth) + ctrl := gomock.NewController(t) + + networkMock := mocks.NewMockNetworkAPI(ctrl) + networkMock.EXPECT().Health().Return(testHealth) sys := NewSystemModule(networkMock, nil, nil, nil, nil, nil, nil) @@ -157,8 +158,10 @@ var testGenesisData = &genesis.Data{ } func TestSystemModule_Chain(t *testing.T) { - api := mocks.NewSystemAPI(t) - api.On("ChainName").Return(testGenesisData.Name) + ctrl := gomock.NewController(t) + + api := mocks.NewMockSystemAPI(ctrl) + api.EXPECT().ChainName().Return(testGenesisData.Name) sys := NewSystemModule(nil, api, nil, nil, nil, nil, nil) res := new(string) @@ -168,8 +171,10 @@ func TestSystemModule_Chain(t *testing.T) { } func TestSystemModule_ChainType(t *testing.T) { - api := mocks.NewSystemAPI(t) - api.On("ChainType").Return(testGenesisData.ChainType) + ctrl := gomock.NewController(t) + + api := mocks.NewMockSystemAPI(ctrl) + api.EXPECT().ChainType().Return(testGenesisData.ChainType) sys := NewSystemModule(nil, api, nil, nil, nil, nil, nil) @@ -178,8 +183,10 @@ func TestSystemModule_ChainType(t *testing.T) { require.Equal(t, testGenesisData.ChainType, *res) } func TestSystemModule_Name(t *testing.T) { - api := mocks.NewSystemAPI(t) - api.On("SystemName").Return(testSystemInfo.SystemName) + ctrl := gomock.NewController(t) + + api := mocks.NewMockSystemAPI(ctrl) + api.EXPECT().SystemName().Return(testSystemInfo.SystemName) sys := NewSystemModule(nil, api, nil, nil, nil, nil, nil) res := new(string) @@ -189,8 +196,10 @@ func TestSystemModule_Name(t *testing.T) { } func TestSystemModule_Version(t *testing.T) { - api := mocks.NewSystemAPI(t) - api.On("SystemVersion").Return(testSystemInfo.SystemVersion) + ctrl := gomock.NewController(t) + + api := mocks.NewMockSystemAPI(ctrl) + api.EXPECT().SystemVersion().Return(testSystemInfo.SystemVersion) sys := NewSystemModule(nil, api, nil, nil, nil, nil, nil) @@ -201,8 +210,10 @@ func TestSystemModule_Version(t *testing.T) { } func TestSystemModule_Properties(t *testing.T) { - api := mocks.NewSystemAPI(t) - api.On("Properties").Return(nil) + ctrl := gomock.NewController(t) + + api := mocks.NewMockSystemAPI(ctrl) + api.EXPECT().Properties().Return(nil) sys := NewSystemModule(nil, api, nil, nil, nil, nil, nil) @@ -386,17 +397,19 @@ func newCoreService(t *testing.T, srvc *state.Service) *core.Service { } func TestSyncState(t *testing.T) { + ctrl := gomock.NewController(t) + fakeCommonHash := common.NewHash([]byte("fake")) fakeHeader := &types.Header{ Number: 49, } - blockapiMock := mocks.NewBlockAPI(t) - blockapiMock.On("BestBlockHash").Return(fakeCommonHash) - blockapiMock.On("GetHeader", fakeCommonHash).Return(fakeHeader, nil).Once() + blockapiMock := mocks.NewMockBlockAPI(ctrl) + blockapiMock.EXPECT().BestBlockHash().Return(fakeCommonHash).Times(2) + blockapiMock.EXPECT().GetHeader(fakeCommonHash).Return(fakeHeader, nil) - netapiMock := mocks.NewNetworkAPI(t) - netapiMock.On("StartingBlock").Return(int64(10)) + netapiMock := mocks.NewMockNetworkAPI(ctrl) + netapiMock.EXPECT().StartingBlock().Return(int64(10)) syncapiCtrl := gomock.NewController(t) syncapiMock := NewMockSyncAPI(syncapiCtrl) @@ -419,12 +432,14 @@ func TestSyncState(t *testing.T) { require.Equal(t, expectedSyncState, res) - blockapiMock.On("GetHeader", fakeCommonHash).Return(nil, errors.New("Problems while getting header")).Once() + blockapiMock.EXPECT().GetHeader(fakeCommonHash).Return(nil, errors.New("Problems while getting header")) err = sysmodule.SyncState(nil, nil, nil) require.Error(t, err) } func TestLocalListenAddresses(t *testing.T) { + ctrl := gomock.NewController(t) + ma, err := multiaddr.NewMultiaddr("/ip4/127.0.0.1/tcp/7001/p2p/12D3KooWCYyh5xoAc5oRyiGU4d9ktcqFQ23JjitNFR6bEcbw7YdN") require.NoError(t, err) @@ -433,8 +448,8 @@ func TestLocalListenAddresses(t *testing.T) { Multiaddrs: []multiaddr.Multiaddr{ma}, } - mockNetAPI := mocks.NewNetworkAPI(t) - mockNetAPI.On("NetworkState").Return(mockedNetState).Once() + mockNetAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetAPI.EXPECT().NetworkState().Return(mockedNetState) res := make([]string, 0) @@ -447,12 +462,14 @@ func TestLocalListenAddresses(t *testing.T) { require.Len(t, res, 1) require.Equal(t, res[0], ma.String()) - mockNetAPI.On("NetworkState").Return(common.NetworkState{Multiaddrs: []multiaddr.Multiaddr{}}).Once() + mockNetAPI.EXPECT().NetworkState().Return(common.NetworkState{Multiaddrs: []multiaddr.Multiaddr{}}) err = sysmodule.LocalListenAddresses(nil, nil, &res) require.Error(t, err, "multiaddress list is empty") } func TestLocalPeerId(t *testing.T) { + ctrl := gomock.NewController(t) + peerID := "12D3KooWBrwpqLE9Z23NEs59m2UHUs9sGYWenxjeCk489Xq7SG2h" encoded := base58.Encode([]byte(peerID)) @@ -460,8 +477,8 @@ func TestLocalPeerId(t *testing.T) { PeerID: peerID, } - mocknetAPI := mocks.NewNetworkAPI(t) - mocknetAPI.On("NetworkState").Return(state).Once() + mocknetAPI := mocks.NewMockNetworkAPI(ctrl) + mocknetAPI.EXPECT().NetworkState().Return(state) sysmodules := new(SystemModule) sysmodules.networkAPI = mocknetAPI @@ -473,16 +490,18 @@ func TestLocalPeerId(t *testing.T) { require.Equal(t, res, encoded) state.PeerID = "" - mocknetAPI.On("NetworkState").Return(state).Once() + mocknetAPI.EXPECT().NetworkState().Return(state) err = sysmodules.LocalPeerId(nil, nil, &res) require.Error(t, err) } func TestAddReservedPeer(t *testing.T) { t.Run("Test Add and Remove reserved peers with success", func(t *testing.T) { - networkMock := mocks.NewNetworkAPI(t) - networkMock.On("AddReservedPeers", mock.AnythingOfType("string")).Return(nil).Once() - networkMock.On("RemoveReservedPeers", mock.AnythingOfType("string")).Return(nil).Once() + ctrl := gomock.NewController(t) + + networkMock := mocks.NewMockNetworkAPI(ctrl) + networkMock.EXPECT().AddReservedPeers(gomock.Any()).Return(nil) + networkMock.EXPECT().RemoveReservedPeers(gomock.Any()).Return(nil) multiAddrPeer := "/ip4/198.51.100.19/tcp/30333/p2p/QmSk5HQbn6LhUwDiNMseVUjuRYhEtYj4aUZ6WfWoGURpdV" sysModule := &SystemModule{ @@ -501,9 +520,11 @@ func TestAddReservedPeer(t *testing.T) { }) t.Run("Test Add and Remove reserved peers without success", func(t *testing.T) { - networkMock := mocks.NewNetworkAPI(t) - networkMock.On("AddReservedPeers", mock.AnythingOfType("string")).Return(errors.New("some problems")).Once() - networkMock.On("RemoveReservedPeers", mock.AnythingOfType("string")).Return(errors.New("other problems")).Once() + ctrl := gomock.NewController(t) + + networkMock := mocks.NewMockNetworkAPI(ctrl) + networkMock.EXPECT().AddReservedPeers(gomock.Any()).Return(errors.New("some problems")) + networkMock.EXPECT().RemoveReservedPeers(gomock.Any()).Return(errors.New("other problems")) sysModule := &SystemModule{ networkAPI: networkMock, diff --git a/dot/rpc/modules/system_test.go b/dot/rpc/modules/system_test.go index 020f9cda9d..a950e1362b 100644 --- a/dot/rpc/modules/system_test.go +++ b/dot/rpc/modules/system_test.go @@ -21,8 +21,10 @@ import ( ) func TestSystemModule_ChainTest(t *testing.T) { - mockSystemAPI := mocks.NewSystemAPI(t) - mockSystemAPI.On("ChainName").Return("polkadot", nil) + ctrl := gomock.NewController(t) + + mockSystemAPI := mocks.NewMockSystemAPI(ctrl) + mockSystemAPI.EXPECT().ChainName().Return("polkadot") sm := &SystemModule{ systemAPI: mockSystemAPI, } @@ -36,8 +38,10 @@ func TestSystemModule_ChainTest(t *testing.T) { } func TestSystemModule_NameTest(t *testing.T) { - mockSystemAPI := mocks.NewSystemAPI(t) - mockSystemAPI.On("SystemName").Return("kusama", nil) + ctrl := gomock.NewController(t) + + mockSystemAPI := mocks.NewMockSystemAPI(ctrl) + mockSystemAPI.EXPECT().SystemName().Return("kusama") sm := &SystemModule{ systemAPI: mockSystemAPI, } @@ -51,8 +55,10 @@ func TestSystemModule_NameTest(t *testing.T) { } func TestSystemModule_ChainTypeTest(t *testing.T) { - mockSystemAPI := mocks.NewSystemAPI(t) - mockSystemAPI.On("ChainType").Return("testChainType", nil) + ctrl := gomock.NewController(t) + + mockSystemAPI := mocks.NewMockSystemAPI(ctrl) + mockSystemAPI.EXPECT().ChainType().Return("testChainType") sm := &SystemModule{ systemAPI: mockSystemAPI, } @@ -66,9 +72,11 @@ func TestSystemModule_ChainTypeTest(t *testing.T) { } func TestSystemModule_PropertiesTest(t *testing.T) { + ctrl := gomock.NewController(t) + var emptyMap map[string]interface{} - mockSystemAPI := mocks.NewSystemAPI(t) - mockSystemAPI.On("Properties").Return(emptyMap) + mockSystemAPI := mocks.NewMockSystemAPI(ctrl) + mockSystemAPI.EXPECT().Properties().Return(emptyMap) sm := &SystemModule{ systemAPI: mockSystemAPI, } @@ -81,8 +89,10 @@ func TestSystemModule_PropertiesTest(t *testing.T) { } func TestSystemModule_SystemVersionTest(t *testing.T) { - mockSystemAPI := mocks.NewSystemAPI(t) - mockSystemAPI.On("SystemVersion").Return("1.2.1", nil) + ctrl := gomock.NewController(t) + + mockSystemAPI := mocks.NewMockSystemAPI(ctrl) + mockSystemAPI.EXPECT().SystemVersion().Return("1.2.1") sm := &SystemModule{ systemAPI: mockSystemAPI, } @@ -96,8 +106,10 @@ func TestSystemModule_SystemVersionTest(t *testing.T) { } func TestSystemModule_HealthTest(t *testing.T) { - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("Health").Return(common.Health{}, nil) + ctrl := gomock.NewController(t) + + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().Health().Return(common.Health{}) sm := &SystemModule{ networkAPI: mockNetworkAPI, } @@ -110,8 +122,10 @@ func TestSystemModule_HealthTest(t *testing.T) { } func TestSystemModule_NetworkStateTest(t *testing.T) { - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("NetworkState").Return(common.NetworkState{}, nil) + ctrl := gomock.NewController(t) + + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().NetworkState().Return(common.NetworkState{}) sm := &SystemModule{ networkAPI: mockNetworkAPI, } @@ -124,8 +138,10 @@ func TestSystemModule_NetworkStateTest(t *testing.T) { } func TestSystemModule_PeersTest(t *testing.T) { - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("Peers").Return([]common.PeerInfo{}, nil) + ctrl := gomock.NewController(t) + + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().Peers().Return([]common.PeerInfo{}) sm := &SystemModule{ networkAPI: mockNetworkAPI, } @@ -138,17 +154,19 @@ func TestSystemModule_PeersTest(t *testing.T) { } func TestSystemModule_NodeRolesTest(t *testing.T) { - mockNetworkAPI1 := mocks.NewNetworkAPI(t) - mockNetworkAPI1.On("NodeRoles").Return(common.FullNodeRole, nil) + ctrl := gomock.NewController(t) - mockNetworkAPI2 := mocks.NewNetworkAPI(t) - mockNetworkAPI2.On("NodeRoles").Return(common.LightClientRole, nil) + mockNetworkAPI1 := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI1.EXPECT().NodeRoles().Return(common.FullNodeRole) - mockNetworkAPI3 := mocks.NewNetworkAPI(t) - mockNetworkAPI3.On("NodeRoles").Return(common.AuthorityRole, nil) + mockNetworkAPI2 := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI2.EXPECT().NodeRoles().Return(common.LightClientRole) - mockNetworkAPI4 := mocks.NewNetworkAPI(t) - mockNetworkAPI4.On("NodeRoles").Return(common.Roles(21), nil) + mockNetworkAPI3 := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI3.EXPECT().NodeRoles().Return(common.AuthorityRole) + + mockNetworkAPI4 := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI4.EXPECT().NodeRoles().Return(common.Roles(21)) type args struct { r *http.Request @@ -210,6 +228,8 @@ func TestSystemModule_NodeRolesTest(t *testing.T) { } func TestSystemModule_AccountNextIndex(t *testing.T) { + ctrl := gomock.NewController(t) + storageKeyHex := common.MustHexToBytes("0x26aa394eea5630e07c48ae0c9558cef7b99d880ec681799c0cf30e8886" + "371da93116aec311d8421cece41129ffaac05aa7f9580382edb384b1b43cbcf3d1b1e7f1a1d232cf4139bd48eaafb9656da27d") signedExt := common.MustHexToBytes("0xad018400d43593c715fdd31c61141abd04a99fd6822c8558854ccde39a5684e" + @@ -221,26 +241,28 @@ func TestSystemModule_AccountNextIndex(t *testing.T) { Validity: new(transaction.Validity), } - mockTxStateAPI := mocks.NewTransactionStateAPI(t) - mockTxStateAPI.On("Pending").Return(v, nil) + mockTxStateAPI := mocks.NewMockTransactionStateAPI(ctrl) + mockTxStateAPI.EXPECT().Pending().Return(v).AnyTimes() - mockCoreAPI := mocks.NewCoreAPI(t) - mockCoreAPI.On("GetMetadata", (*common.Hash)(nil)).Return(common.MustHexToBytes(testdata.NewTestMetadata()), nil) + mockCoreAPI := mocks.NewMockCoreAPI(ctrl) + mockCoreAPI.EXPECT().GetMetadata((*common.Hash)(nil)). + Return(common.MustHexToBytes(testdata.NewTestMetadata()), nil).AnyTimes() - mockCoreAPIErr := mocks.NewCoreAPI(t) - mockCoreAPIErr.On("GetMetadata", (*common.Hash)(nil)).Return(nil, errors.New("getMetadata error")) + mockCoreAPIErr := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIErr.EXPECT().GetMetadata((*common.Hash)(nil)). + Return(nil, errors.New("getMetadata error")) // Magic number mismatch - mockCoreAPIMagicNumMismatch := mocks.NewCoreAPI(t) - mockCoreAPIMagicNumMismatch.On("GetMetadata", (*common.Hash)(nil)).Return(storageKeyHex, nil) + mockCoreAPIMagicNumMismatch := mocks.NewMockCoreAPI(ctrl) + mockCoreAPIMagicNumMismatch.EXPECT().GetMetadata((*common.Hash)(nil)).Return(storageKeyHex, nil) - mockStorageAPI := mocks.NewStorageAPI(t) - mockStorageAPI.On("GetStorage", (*common.Hash)(nil), storageKeyHex). + mockStorageAPI := mocks.NewMockStorageAPI(ctrl) + mockStorageAPI.EXPECT().GetStorage((*common.Hash)(nil), storageKeyHex). Return(common.MustHexToBytes("0x0300000000000000000000000000000000000000000000000000000000000000000000"+ "000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"), nil) - mockStorageAPIErr := mocks.NewStorageAPI(t) - mockStorageAPIErr.On("GetStorage", (*common.Hash)(nil), storageKeyHex).Return(nil, errors.New("getStorage error")) + mockStorageAPIErr := mocks.NewMockStorageAPI(ctrl) + mockStorageAPIErr.EXPECT().GetStorage((*common.Hash)(nil), storageKeyHex).Return(nil, errors.New("getStorage error")) type args struct { r *http.Request @@ -316,17 +338,19 @@ func TestSystemModule_AccountNextIndex(t *testing.T) { } func TestSystemModule_SyncState(t *testing.T) { + ctrl := gomock.NewController(t) + hash := common.MustHexToHash("0x3aa96b0149b6ca3688878bdbd19464448624136398e3ce45b9e755d3ab61355a") - mockBlockAPI := mocks.NewBlockAPI(t) - mockBlockAPI.On("BestBlockHash").Return(hash) - mockBlockAPI.On("GetHeader", hash).Return(types.NewEmptyHeader(), nil) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) + mockBlockAPI.EXPECT().BestBlockHash().Return(hash) + mockBlockAPI.EXPECT().GetHeader(hash).Return(types.NewEmptyHeader(), nil) - mockBlockAPIErr := mocks.NewBlockAPI(t) - mockBlockAPIErr.On("BestBlockHash").Return(hash) - mockBlockAPIErr.On("GetHeader", hash).Return(nil, errors.New("GetHeader Err")) + mockBlockAPIErr := mocks.NewMockBlockAPI(ctrl) + mockBlockAPIErr.EXPECT().BestBlockHash().Return(hash) + mockBlockAPIErr.EXPECT().GetHeader(hash).Return(nil, errors.New("GetHeader Err")) - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("StartingBlock").Return(int64(23)) + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().StartingBlock().Return(int64(23)) ctrlSyncAPI := gomock.NewController(t) mockSyncAPI := NewMockSyncAPI(ctrlSyncAPI) @@ -380,8 +404,10 @@ func TestSystemModule_SyncState(t *testing.T) { } func TestSystemModule_LocalListenAddresses(t *testing.T) { - mockNetworkAPIEmpty := mocks.NewNetworkAPI(t) - mockNetworkAPIEmpty.On("NetworkState").Return(common.NetworkState{}) + ctrl := gomock.NewController(t) + + mockNetworkAPIEmpty := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPIEmpty.EXPECT().NetworkState().Return(common.NetworkState{}) addr, err := multiaddr.NewMultiaddr("/ip4/1.2.3.4/tcp/80") require.NoError(t, err) @@ -392,8 +418,8 @@ func TestSystemModule_LocalListenAddresses(t *testing.T) { Multiaddrs: multiAddy, } - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("NetworkState").Return(ns, nil) + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().NetworkState().Return(ns) type args struct { r *http.Request @@ -440,8 +466,10 @@ func TestSystemModule_LocalListenAddresses(t *testing.T) { } func TestSystemModule_LocalPeerId(t *testing.T) { - mockNetworkAPIEmpty := mocks.NewNetworkAPI(t) - mockNetworkAPIEmpty.On("NetworkState").Return(common.NetworkState{}) + ctrl := gomock.NewController(t) + + mockNetworkAPIEmpty := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPIEmpty.EXPECT().NetworkState().Return(common.NetworkState{}) addr, err := multiaddr.NewMultiaddr("/ip4/1.2.3.4/tcp/80") require.NoError(t, err) @@ -452,8 +480,8 @@ func TestSystemModule_LocalPeerId(t *testing.T) { Multiaddrs: multiAddy, } - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("NetworkState").Return(ns, nil) + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().NetworkState().Return(ns) type args struct { r *http.Request @@ -499,11 +527,13 @@ func TestSystemModule_LocalPeerId(t *testing.T) { } func TestSystemModule_AddReservedPeer(t *testing.T) { - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("AddReservedPeers", "jimbo").Return(nil) + ctrl := gomock.NewController(t) + + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().AddReservedPeers("jimbo").Return(nil) - mockNetworkAPIErr := mocks.NewNetworkAPI(t) - mockNetworkAPIErr.On("AddReservedPeers", "jimbo").Return(errors.New("addReservedPeer error")) + mockNetworkAPIErr := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPIErr.EXPECT().AddReservedPeers("jimbo").Return(errors.New("addReservedPeer error")) type args struct { r *http.Request @@ -557,11 +587,13 @@ func TestSystemModule_AddReservedPeer(t *testing.T) { } func TestSystemModule_RemoveReservedPeer(t *testing.T) { - mockNetworkAPI := mocks.NewNetworkAPI(t) - mockNetworkAPI.On("RemoveReservedPeers", "jimbo").Return(nil) + ctrl := gomock.NewController(t) + + mockNetworkAPI := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPI.EXPECT().RemoveReservedPeers("jimbo").Return(nil) - mockNetworkAPIErr := mocks.NewNetworkAPI(t) - mockNetworkAPIErr.On("RemoveReservedPeers", "jimbo").Return(errors.New("removeReservedPeer error")) + mockNetworkAPIErr := mocks.NewMockNetworkAPI(ctrl) + mockNetworkAPIErr.EXPECT().RemoveReservedPeers("jimbo").Return(errors.New("removeReservedPeer error")) type args struct { r *http.Request diff --git a/dot/rpc/subscription/listeners_integration_test.go b/dot/rpc/subscription/listeners_integration_test.go index dbe9043b66..9256b4b264 100644 --- a/dot/rpc/subscription/listeners_integration_test.go +++ b/dot/rpc/subscription/listeners_integration_test.go @@ -24,7 +24,6 @@ import ( "github.com/ChainSafe/gossamer/lib/transaction" "github.com/ChainSafe/gossamer/pkg/scale" gomock "github.com/golang/mock/gomock" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) @@ -78,11 +77,13 @@ func TestStorageObserver_Update(t *testing.T) { } func TestBlockListener_Listen(t *testing.T) { + ctrl := gomock.NewController(t) + wsconn, ws, cancel := setupWSConn(t) defer cancel() - BlockAPI := mocks.NewBlockAPI(t) - BlockAPI.On("FreeImportedBlockNotifierChannel", mock.AnythingOfType("chan *types.Block")) + BlockAPI := mocks.NewMockBlockAPI(ctrl) + BlockAPI.EXPECT().FreeImportedBlockNotifierChannel(gomock.Any()) wsconn.BlockAPI = BlockAPI @@ -124,11 +125,13 @@ func TestBlockListener_Listen(t *testing.T) { } func TestBlockFinalizedListener_Listen(t *testing.T) { + ctrl := gomock.NewController(t) + wsconn, ws, cancel := setupWSConn(t) defer cancel() - BlockAPI := mocks.NewBlockAPI(t) - BlockAPI.On("FreeFinalisedNotifierChannel", mock.AnythingOfType("chan *types.FinalisationInfo")) + BlockAPI := mocks.NewMockBlockAPI(ctrl) + BlockAPI.EXPECT().FreeFinalisedNotifierChannel(gomock.Any()) wsconn.BlockAPI = BlockAPI @@ -170,6 +173,8 @@ func TestBlockFinalizedListener_Listen(t *testing.T) { require.Equal(t, string(expectedResponseBytes)+"\n", string(msg)) } func TestExtrinsicSubmitListener_Listen(t *testing.T) { + ctrl := gomock.NewController(t) + wsconn, ws, cancel := setupWSConn(t) defer cancel() @@ -177,13 +182,12 @@ func TestExtrinsicSubmitListener_Listen(t *testing.T) { notifyFinalizedChan := make(chan *types.FinalisationInfo, 100) txStatusChan := make(chan transaction.Status) - BlockAPI := mocks.NewBlockAPI(t) - BlockAPI.On("FreeImportedBlockNotifierChannel", mock.AnythingOfType("chan *types.Block")) - BlockAPI.On("FreeFinalisedNotifierChannel", mock.AnythingOfType("chan *types.FinalisationInfo")) + BlockAPI := mocks.NewMockBlockAPI(ctrl) + BlockAPI.EXPECT().FreeImportedBlockNotifierChannel(gomock.Any()) + BlockAPI.EXPECT().FreeFinalisedNotifierChannel(gomock.Any()) wsconn.BlockAPI = BlockAPI - ctrl := gomock.NewController(t) TxStateAPI := NewMockTransactionStateAPI(ctrl) TxStateAPI.EXPECT().FreeStatusNotifierChannel(gomock.Any()).AnyTimes() TxStateAPI.EXPECT().GetStatusNotifierChannel(gomock.Any()).Return(make(chan transaction.Status)).AnyTimes() @@ -241,6 +245,8 @@ func TestExtrinsicSubmitListener_Listen(t *testing.T) { func TestGrandpaJustification_Listen(t *testing.T) { t.Run("When justification doesnt returns error", func(t *testing.T) { + ctrl := gomock.NewController(t) + wsconn, ws, cancel := setupWSConn(t) defer cancel() @@ -255,9 +261,9 @@ func TestGrandpaJustification_Listen(t *testing.T) { mockedJustBytes, err := scale.Marshal(mockedJust) require.NoError(t, err) - blockStateMock := mocks.NewBlockAPI(t) - blockStateMock.On("GetJustification", mock.AnythingOfType("common.Hash")).Return(mockedJustBytes, nil) - blockStateMock.On("FreeFinalisedNotifierChannel", mock.AnythingOfType("chan *types.FinalisationInfo")) + blockStateMock := mocks.NewMockBlockAPI(ctrl) + blockStateMock.EXPECT().GetJustification(gomock.Any()).Return(mockedJustBytes, nil) + blockStateMock.EXPECT().FreeFinalisedNotifierChannel(gomock.Any()) wsconn.BlockAPI = blockStateMock finchannel := make(chan *types.FinalisationInfo) @@ -290,13 +296,15 @@ func TestGrandpaJustification_Listen(t *testing.T) { } func TestRuntimeChannelListener_Listen(t *testing.T) { + ctrl := gomock.NewController(t) + notifyChan := make(chan runtime.Version) mockConnection := &mockWSConnAPI{} rvl := RuntimeVersionListener{ wsconn: mockConnection, subID: 0, runtimeUpdate: notifyChan, - coreAPI: modules.NewMockCoreAPI(t), + coreAPI: modules.NewMockAnyAPI(ctrl), } expectedInitialVersion := modules.StateRuntimeVersionResponse{ diff --git a/dot/rpc/subscription/websocket_integration_test.go b/dot/rpc/subscription/websocket_integration_test.go index de981469cc..a427da8859 100644 --- a/dot/rpc/subscription/websocket_integration_test.go +++ b/dot/rpc/subscription/websocket_integration_test.go @@ -21,11 +21,12 @@ import ( "github.com/golang/mock/gomock" "github.com/gorilla/websocket" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) func TestWSConn_HandleConn(t *testing.T) { + ctrl := gomock.NewController(t) + wsconn, c, cancel := setupWSConn(t) wsconn.Subscriptions = make(map[uint32]Listener) defer cancel() @@ -44,7 +45,7 @@ func TestWSConn_HandleConn(t *testing.T) { `"error":{"code":null,"message":"error StorageAPI not set"},`+ `"id":1}`+"\n"), msg) - wsconn.StorageAPI = modules.NewMockeryStorageAPI(t) + wsconn.StorageAPI = modules.NewMockAnyStorageAPI(ctrl) res, err = wsconn.initStorageChangeListener(1, nil) require.Nil(t, res) @@ -170,7 +171,7 @@ func TestWSConn_HandleConn(t *testing.T) { require.NoError(t, err) require.Equal(t, []byte(`{"jsonrpc":"2.0","error":{"code":null,"message":"error BlockAPI not set"},"id":1}`+"\n"), msg) - wsconn.BlockAPI = modules.NewMockeryBlockAPI(t) + wsconn.BlockAPI = modules.NewMockAnyBlockAPI(ctrl) res, err = wsconn.initBlockListener(1, nil) require.NoError(t, err) @@ -200,7 +201,7 @@ func TestWSConn_HandleConn(t *testing.T) { require.NoError(t, err) require.Equal(t, []byte(`{"jsonrpc":"2.0","error":{"code":null,"message":"error BlockAPI not set"},"id":1}`+"\n"), msg) - wsconn.BlockAPI = modules.NewMockeryBlockAPI(t) + wsconn.BlockAPI = modules.NewMockAnyBlockAPI(ctrl) res, err = wsconn.initBlockFinalizedListener(1, nil) require.NoError(t, err) @@ -211,10 +212,9 @@ func TestWSConn_HandleConn(t *testing.T) { require.Equal(t, []byte(`{"jsonrpc":"2.0","result":7,"id":1}`+"\n"), msg) // test initExtrinsicWatch - wsconn.CoreAPI = modules.NewMockCoreAPI(t) + wsconn.CoreAPI = modules.NewMockAnyAPI(ctrl) wsconn.BlockAPI = nil - ctrl := gomock.NewController(t) transactionStateAPI := NewMockTransactionStateAPI(ctrl) transactionStateAPI.EXPECT().FreeStatusNotifierChannel(gomock.Any()).AnyTimes() transactionStateAPI.EXPECT().GetStatusNotifierChannel(gomock.Any()).Return(make(chan transaction.Status)).AnyTimes() @@ -228,7 +228,7 @@ func TestWSConn_HandleConn(t *testing.T) { require.EqualError(t, err, "error BlockAPI not set") require.Nil(t, listner) - wsconn.BlockAPI = modules.NewMockeryBlockAPI(t) + wsconn.BlockAPI = modules.NewMockAnyBlockAPI(ctrl) listner, err = wsconn.initExtrinsicWatch(0, []interface{}{"0x26aa"}) require.NoError(t, err) require.NotNil(t, listner) @@ -244,8 +244,8 @@ func TestWSConn_HandleConn(t *testing.T) { require.NoError(t, err) require.NoError(t, err) - coreAPI := mocks.NewCoreAPI(t) - coreAPI.On("HandleSubmittedExtrinsic", mock.AnythingOfType("types.Extrinsic")). + coreAPI := mocks.NewMockCoreAPI(ctrl) + coreAPI.EXPECT().HandleSubmittedExtrinsic(gomock.Any()). Return(invalidTransaction) wsconn.CoreAPI = coreAPI listner, err = wsconn.initExtrinsicWatch(0, @@ -269,14 +269,14 @@ func TestWSConn_HandleConn(t *testing.T) { mockedJustBytes, err := scale.Marshal(mockedJust) require.NoError(t, err) - wsconn.CoreAPI = modules.NewMockCoreAPI(t) - BlockAPI := mocks.NewBlockAPI(t) + wsconn.CoreAPI = modules.NewMockAnyAPI(ctrl) + BlockAPI := mocks.NewMockBlockAPI(ctrl) fCh := make(chan *types.FinalisationInfo, 5) - BlockAPI.On("GetFinalisedNotifierChannel").Return(fCh) + BlockAPI.EXPECT().GetFinalisedNotifierChannel().Return(fCh) - BlockAPI.On("GetJustification", mock.AnythingOfType("common.Hash")).Return(mockedJustBytes, nil) - BlockAPI.On("FreeFinalisedNotifierChannel", mock.AnythingOfType("chan *types.FinalisationInfo")) + BlockAPI.EXPECT().GetJustification(gomock.Any()).Return(mockedJustBytes, nil) + BlockAPI.EXPECT().FreeFinalisedNotifierChannel(gomock.Any()) wsconn.BlockAPI = BlockAPI listener, err := wsconn.initGrandpaJustificationListener(0, nil) @@ -309,6 +309,8 @@ func TestWSConn_HandleConn(t *testing.T) { } func TestSubscribeAllHeads(t *testing.T) { + ctrl := gomock.NewController(t) + wsconn, c, cancel := setupWSConn(t) wsconn.Subscriptions = make(map[uint32]Listener) defer cancel() @@ -322,15 +324,15 @@ func TestSubscribeAllHeads(t *testing.T) { require.NoError(t, err) require.Equal(t, []byte(`{"jsonrpc":"2.0","error":{"code":null,"message":"error BlockAPI not set"},"id":1}`+"\n"), msg) - mockBlockAPI := mocks.NewBlockAPI(t) + mockBlockAPI := mocks.NewMockBlockAPI(ctrl) wsconn.BlockAPI = mockBlockAPI iCh := make(chan *types.Block) - mockBlockAPI.On("GetImportedBlockNotifierChannel").Return(iCh).Once() + mockBlockAPI.EXPECT().GetImportedBlockNotifierChannel().Return(iCh) fCh := make(chan *types.FinalisationInfo) - mockBlockAPI.On("GetFinalisedNotifierChannel").Return(fCh).Once() + mockBlockAPI.EXPECT().GetFinalisedNotifierChannel().Return(fCh) l, err := wsconn.initAllBlocksListerner(1, nil) require.NoError(t, err) @@ -385,11 +387,10 @@ func TestSubscribeAllHeads(t *testing.T) { require.NoError(t, err) require.Equal(t, []byte(expected+"\n"), msg) - mockBlockAPI.On("FreeImportedBlockNotifierChannel", mock.AnythingOfType("chan *types.Block")) - mockBlockAPI.On("FreeFinalisedNotifierChannel", mock.AnythingOfType("chan *types.FinalisationInfo")) + mockBlockAPI.EXPECT().FreeImportedBlockNotifierChannel(gomock.Any()) + mockBlockAPI.EXPECT().FreeFinalisedNotifierChannel(gomock.Any()) require.NoError(t, l.Stop()) - mockBlockAPI.On("FreeImportedBlockNotifierChannel", mock.AnythingOfType("chan *types.Block")) } func TestWSConn_CheckWebsocketInvalidData(t *testing.T) { diff --git a/dot/rpc/websocket_integration_test.go b/dot/rpc/websocket_integration_test.go index 46399b8f36..3844ac79ba 100644 --- a/dot/rpc/websocket_integration_test.go +++ b/dot/rpc/websocket_integration_test.go @@ -65,15 +65,16 @@ var testCalls = []struct { } func TestHTTPServer_ServeHTTP(t *testing.T) { + ctrl := gomock.NewController(t) + coreAPI := newCoreServiceTest(t) si := &types.SystemInfo{ SystemName: "gossamer", } sysAPI := system.NewService(si, nil) - bAPI := modules.NewMockeryBlockAPI(t) - sAPI := modules.NewMockeryStorageAPI(t) + bAPI := modules.NewMockAnyBlockAPI(ctrl) + sAPI := modules.NewMockAnyStorageAPI(ctrl) - ctrl := gomock.NewController(t) TxStateAPI := NewMockTransactionStateAPI(ctrl) TxStateAPI.EXPECT().FreeStatusNotifierChannel(gomock.Any()).AnyTimes() TxStateAPI.EXPECT().GetStatusNotifierChannel(gomock.Any()).Return(make(chan transaction.Status)).AnyTimes() diff --git a/dot/state/mock_observer.go b/dot/state/mock_observer.go deleted file mode 100644 index e24536562e..0000000000 --- a/dot/state/mock_observer.go +++ /dev/null @@ -1,60 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package state - -import mock "github.com/stretchr/testify/mock" - -// MockObserver is an autogenerated mock type for the Observer type -type MockObserver struct { - mock.Mock -} - -// GetFilter provides a mock function with given fields: -func (_m *MockObserver) GetFilter() map[string][]byte { - ret := _m.Called() - - var r0 map[string][]byte - if rf, ok := ret.Get(0).(func() map[string][]byte); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(map[string][]byte) - } - } - - return r0 -} - -// GetID provides a mock function with given fields: -func (_m *MockObserver) GetID() uint { - ret := _m.Called() - - var r0 uint - if rf, ok := ret.Get(0).(func() uint); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(uint) - } - - return r0 -} - -// Update provides a mock function with given fields: result -func (_m *MockObserver) Update(result *SubscriptionResult) { - _m.Called(result) -} - -type mockConstructorTestingTNewMockObserver interface { - mock.TestingT - Cleanup(func()) -} - -// NewMockObserver creates a new instance of MockObserver. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewMockObserver(t mockConstructorTestingTNewMockObserver) *MockObserver { - mock := &MockObserver{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/dot/state/mocks_generate_test.go b/dot/state/mocks_generate_test.go index 8ca7caa79a..77bd4179c3 100644 --- a/dot/state/mocks_generate_test.go +++ b/dot/state/mocks_generate_test.go @@ -3,4 +3,4 @@ package state -//go:generate mockgen -destination=mocks_test.go -package $GOPACKAGE . Telemetry,BlockStateDatabase +//go:generate mockgen -destination=mocks_test.go -package $GOPACKAGE . Telemetry,BlockStateDatabase,Observer diff --git a/dot/state/mocks_test.go b/dot/state/mocks_test.go index 56511b238d..1ca5e3196a 100644 --- a/dot/state/mocks_test.go +++ b/dot/state/mocks_test.go @@ -1,5 +1,5 @@ // Code generated by MockGen. DO NOT EDIT. -// Source: github.com/ChainSafe/gossamer/dot/state (interfaces: Telemetry,BlockStateDatabase) +// Source: github.com/ChainSafe/gossamer/dot/state (interfaces: Telemetry,BlockStateDatabase,Observer) // Package state is a generated GoMock package. package state @@ -141,3 +141,66 @@ func (mr *MockBlockStateDatabaseMockRecorder) Put(arg0, arg1 interface{}) *gomoc mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Put", reflect.TypeOf((*MockBlockStateDatabase)(nil).Put), arg0, arg1) } + +// MockObserver is a mock of Observer interface. +type MockObserver struct { + ctrl *gomock.Controller + recorder *MockObserverMockRecorder +} + +// MockObserverMockRecorder is the mock recorder for MockObserver. +type MockObserverMockRecorder struct { + mock *MockObserver +} + +// NewMockObserver creates a new mock instance. +func NewMockObserver(ctrl *gomock.Controller) *MockObserver { + mock := &MockObserver{ctrl: ctrl} + mock.recorder = &MockObserverMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockObserver) EXPECT() *MockObserverMockRecorder { + return m.recorder +} + +// GetFilter mocks base method. +func (m *MockObserver) GetFilter() map[string][]byte { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetFilter") + ret0, _ := ret[0].(map[string][]byte) + return ret0 +} + +// GetFilter indicates an expected call of GetFilter. +func (mr *MockObserverMockRecorder) GetFilter() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetFilter", reflect.TypeOf((*MockObserver)(nil).GetFilter)) +} + +// GetID mocks base method. +func (m *MockObserver) GetID() uint { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetID") + ret0, _ := ret[0].(uint) + return ret0 +} + +// GetID indicates an expected call of GetID. +func (mr *MockObserverMockRecorder) GetID() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetID", reflect.TypeOf((*MockObserver)(nil).GetID)) +} + +// Update mocks base method. +func (m *MockObserver) Update(arg0 *SubscriptionResult) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Update", arg0) +} + +// Update indicates an expected call of Update. +func (mr *MockObserverMockRecorder) Update(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockObserver)(nil).Update), arg0) +} diff --git a/dot/state/storage_notify.go b/dot/state/storage_notify.go index 46077393c4..45cf810f06 100644 --- a/dot/state/storage_notify.go +++ b/dot/state/storage_notify.go @@ -37,8 +37,6 @@ func (s SubscriptionResult) String() string { return "[" + strings.Join(changes, ", ") + "]" } -//go:generate mockery --name Observer --structname MockObserver --case underscore --inpackage - // Observer interface defines functions needed for observers, Observer Design Pattern type Observer interface { Update(result *SubscriptionResult) diff --git a/dot/state/storage_notify_test.go b/dot/state/storage_notify_test.go index 8ffe88bce2..018d31faf1 100644 --- a/dot/state/storage_notify_test.go +++ b/dot/state/storage_notify_test.go @@ -13,35 +13,37 @@ import ( "github.com/ChainSafe/chaindb" "github.com/ChainSafe/gossamer/lib/common" - "github.com/stretchr/testify/mock" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/require" ) func TestStorageState_RegisterStorageObserver(t *testing.T) { + ctrl := gomock.NewController(t) + ss := newTestStorageState(t) ts, err := ss.TrieState(nil) require.NoError(t, err) mockfilter := map[string][]byte{} - mockobs := NewMockObserver(t) + mockobs := NewMockObserver(ctrl) - mockobs.On("GetID").Return(uint(10)).Times(2) + mockobs.EXPECT().GetID().Return(uint(10)).Times(2) var fireAndForgetMockCallsWG sync.WaitGroup fireAndForgetMockCallsWG.Add(2) - getFilterCall := mockobs.On("GetFilter").Return(mockfilter).Times(2) - getFilterCall.RunFn = func(args mock.Arguments) { - fireAndForgetMockCallsWG.Done() - } + mockobs.EXPECT().GetFilter().DoAndReturn(func() map[string][]byte { + defer fireAndForgetMockCallsWG.Done() + return mockfilter + }).Times(2) fireAndForgetMockCallsWG.Add(1) - lastMockCall := mockobs.On("Update", mock.AnythingOfType("*state.SubscriptionResult")). - Return(map[string][]byte{}).Once() - lastMockCall.RunFn = func(args mock.Arguments) { - fireAndForgetMockCallsWG.Done() - } + mockobs.EXPECT().Update(gomock.Any()). + DoAndReturn(func(r *SubscriptionResult) map[string][]byte { + defer fireAndForgetMockCallsWG.Done() + return map[string][]byte{} + }) ss.RegisterStorageObserver(mockobs) defer ss.UnregisterStorageObserver(mockobs) @@ -57,6 +59,8 @@ func TestStorageState_RegisterStorageObserver(t *testing.T) { } func TestStorageState_RegisterStorageObserver_Multi(t *testing.T) { + ctrl := gomock.NewController(t) + ss := newTestStorageState(t) ts, err := ss.TrieState(nil) require.NoError(t, err) @@ -67,11 +71,11 @@ func TestStorageState_RegisterStorageObserver_Multi(t *testing.T) { for i := 0; i < num; i++ { mockfilter := map[string][]byte{} - mockobs := NewMockObserver(t) + mockobs := NewMockObserver(ctrl) - mockobs.On("Update", mock.AnythingOfType("*state.SubscriptionResult")) - mockobs.On("GetID").Return(uint(10)) - mockobs.On("GetFilter").Return(mockfilter).Times(2) + mockobs.EXPECT().Update(gomock.Any()) + mockobs.EXPECT().GetID().Return(uint(10)).Times(2) + mockobs.EXPECT().GetFilter().Return(mockfilter).Times(2) mocks = append(mocks, mockobs) ss.RegisterStorageObserver(mockobs) @@ -94,7 +98,9 @@ func TestStorageState_RegisterStorageObserver_Multi(t *testing.T) { } func TestStorageState_RegisterStorageObserver_Multi_Filter(t *testing.T) { - t.Skip() // this seems to fail often on CI + t.Skip() // this seems to fail often on CI\ + + ctrl := gomock.NewController(t) ss := newTestStorageState(t) ts, err := ss.TrieState(nil) require.NoError(t, err) @@ -109,10 +115,10 @@ func TestStorageState_RegisterStorageObserver_Multi_Filter(t *testing.T) { } for i := 0; i < num; i++ { - mockobs := NewMockObserver(t) - mockobs.On("Update", mock.AnythingOfType("*state.SubscriptionResult")) - mockobs.On("GetID").Return(uint(i)) - mockobs.On("GetFilter").Return(filter).Times(len(filter) + 3) + mockobs := NewMockObserver(ctrl) + mockobs.EXPECT().Update(gomock.Any()) + mockobs.EXPECT().GetID().Return(uint(i)) + mockobs.EXPECT().GetFilter().Return(filter).Times(len(filter) + 3) mocks = append(mocks, mockobs) ss.RegisterStorageObserver(mockobs) diff --git a/go.mod b/go.mod index 3c27ed5a87..976977c5c7 100644 --- a/go.mod +++ b/go.mod @@ -159,7 +159,6 @@ require ( github.com/shirou/gopsutil v3.21.4-0.20210419000835-c7a38de76ee5+incompatible // indirect github.com/spacemonkeygo/spacelog v0.0.0-20180420211403-2296661a0572 // indirect github.com/spaolacci/murmur3 v1.1.0 // indirect - github.com/stretchr/objx v0.5.0 // indirect github.com/tklauser/go-sysconf v0.3.5 // indirect github.com/tklauser/numcpus v0.2.2 // indirect github.com/tomasen/realip v0.0.0-20180522021738-f0c99a92ddce // indirect diff --git a/go.sum b/go.sum index 114305a628..260bddcc6e 100644 --- a/go.sum +++ b/go.sum @@ -756,7 +756,6 @@ github.com/src-d/envconfig v1.0.0/go.mod h1:Q9YQZ7BKITldTBnoxsE5gOeB5y66RyPXeue/ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= -github.com/stretchr/objx v0.5.0 h1:1zr/of2m5FGMsad5YfcqgdqdWrIhu+EBEJRhR1U7z/c= github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= diff --git a/lib/babe/babe_integration_test.go b/lib/babe/babe_integration_test.go index 3a188b773a..8ebd8e381e 100644 --- a/lib/babe/babe_integration_test.go +++ b/lib/babe/babe_integration_test.go @@ -23,7 +23,6 @@ import ( "github.com/ChainSafe/gossamer/pkg/scale" "github.com/golang/mock/gomock" - "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) @@ -202,11 +201,11 @@ func TestService_SlotDuration(t *testing.T) { } func TestService_ProducesBlocks(t *testing.T) { - blockImportHandler := mocks.NewBlockImportHandler(t) - blockImportHandler. - On("HandleBlockProduced", - mock.AnythingOfType("*types.Block"), mock.AnythingOfType("*storage.TrieState")). - Return(nil) + ctrl := gomock.NewController(t) + + blockImportHandler := NewMockBlockImportHandler(ctrl) + blockImportHandler.EXPECT().HandleBlockProduced(gomock.Any(), gomock.Any()). + Return(nil).MinTimes(2) cfg := ServiceConfig{ Authority: true, Lead: true, @@ -216,15 +215,9 @@ func TestService_ProducesBlocks(t *testing.T) { err := babeService.Start() require.NoError(t, err) - defer func() { - _ = babeService.Stop() - }() - time.Sleep(babeService.constants.slotDuration * 2) - babeService.blockImportHandler.(*mocks.BlockImportHandler). - AssertCalled(t, "HandleBlockProduced", - mock.AnythingOfType("*types.Block"), - mock.AnythingOfType("*storage.TrieState")) + err = babeService.Stop() + require.NoError(t, err) } func TestService_GetAuthorityIndex(t *testing.T) { diff --git a/lib/babe/mocks/block_import_handler.go b/lib/babe/mocks/block_import_handler.go deleted file mode 100644 index c546ace666..0000000000 --- a/lib/babe/mocks/block_import_handler.go +++ /dev/null @@ -1,43 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - types "github.com/ChainSafe/gossamer/dot/types" - storage "github.com/ChainSafe/gossamer/lib/runtime/storage" - mock "github.com/stretchr/testify/mock" -) - -// BlockImportHandler is an autogenerated mock type for the BlockImportHandler type -type BlockImportHandler struct { - mock.Mock -} - -// HandleBlockProduced provides a mock function with given fields: block, state -func (_m *BlockImportHandler) HandleBlockProduced(block *types.Block, state *storage.TrieState) error { - ret := _m.Called(block, state) - - var r0 error - if rf, ok := ret.Get(0).(func(*types.Block, *storage.TrieState) error); ok { - r0 = rf(block, state) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewBlockImportHandler interface { - mock.TestingT - Cleanup(func()) -} - -// NewBlockImportHandler creates a new instance of BlockImportHandler. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewBlockImportHandler(t mockConstructorTestingTNewBlockImportHandler) *BlockImportHandler { - mock := &BlockImportHandler{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/lib/babe/mocks_generate_test.go b/lib/babe/mocks_generate_test.go index 7a216cac08..55cabd4f0b 100644 --- a/lib/babe/mocks_generate_test.go +++ b/lib/babe/mocks_generate_test.go @@ -3,5 +3,4 @@ package babe -//go:generate mockgen -destination=mocks/network.go -package=mocks github.com/ChainSafe/gossamer/dot/core Network //go:generate mockgen -destination=mock_telemetry_test.go -package $GOPACKAGE . Telemetry diff --git a/lib/babe/state.go b/lib/babe/state.go index 0bdd05efba..a5bf89d324 100644 --- a/lib/babe/state.go +++ b/lib/babe/state.go @@ -70,8 +70,6 @@ type EpochState interface { SkipVerify(*types.Header) (bool, error) } -//go:generate mockery --name BlockImportHandler --structname BlockImportHandler --case underscore --keeptree - // BlockImportHandler is the interface for the handler of new blocks type BlockImportHandler interface { HandleBlockProduced(block *types.Block, state *rtstorage.TrieState) error diff --git a/lib/grandpa/mocks/network.go b/lib/grandpa/mocks/network.go deleted file mode 100644 index 5d92323d34..0000000000 --- a/lib/grandpa/mocks/network.go +++ /dev/null @@ -1,65 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - network "github.com/ChainSafe/gossamer/dot/network" - mock "github.com/stretchr/testify/mock" - - peer "github.com/libp2p/go-libp2p/core/peer" - - protocol "github.com/libp2p/go-libp2p/core/protocol" -) - -// Network is an autogenerated mock type for the Network type -type Network struct { - mock.Mock -} - -// GossipMessage provides a mock function with given fields: msg -func (_m *Network) GossipMessage(msg network.NotificationsMessage) { - _m.Called(msg) -} - -// RegisterNotificationsProtocol provides a mock function with given fields: sub, messageID, handshakeGetter, handshakeDecoder, handshakeValidator, messageDecoder, messageHandler, batchHandler, maxSize -func (_m *Network) RegisterNotificationsProtocol(sub protocol.ID, messageID byte, handshakeGetter func() (network.Handshake, error), handshakeDecoder func([]byte) (network.Handshake, error), handshakeValidator func(peer.ID, network.Handshake) error, messageDecoder func([]byte) (network.NotificationsMessage, error), messageHandler func(peer.ID, network.NotificationsMessage) (bool, error), batchHandler func(peer.ID, network.NotificationsMessage), maxSize uint64) error { - ret := _m.Called(sub, messageID, handshakeGetter, handshakeDecoder, handshakeValidator, messageDecoder, messageHandler, batchHandler, maxSize) - - var r0 error - if rf, ok := ret.Get(0).(func(protocol.ID, byte, func() (network.Handshake, error), func([]byte) (network.Handshake, error), func(peer.ID, network.Handshake) error, func([]byte) (network.NotificationsMessage, error), func(peer.ID, network.NotificationsMessage) (bool, error), func(peer.ID, network.NotificationsMessage), uint64) error); ok { - r0 = rf(sub, messageID, handshakeGetter, handshakeDecoder, handshakeValidator, messageDecoder, messageHandler, batchHandler, maxSize) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// SendMessage provides a mock function with given fields: to, msg -func (_m *Network) SendMessage(to peer.ID, msg network.NotificationsMessage) error { - ret := _m.Called(to, msg) - - var r0 error - if rf, ok := ret.Get(0).(func(peer.ID, network.NotificationsMessage) error); ok { - r0 = rf(to, msg) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewNetwork interface { - mock.TestingT - Cleanup(func()) -} - -// NewNetwork creates a new instance of Network. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewNetwork(t mockConstructorTestingTNewNetwork) *Network { - mock := &Network{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/lib/grandpa/mocks_generate_test.go b/lib/grandpa/mocks_generate_test.go index 7c65dd0e5d..0f01cb2d9e 100644 --- a/lib/grandpa/mocks_generate_test.go +++ b/lib/grandpa/mocks_generate_test.go @@ -5,5 +5,4 @@ package grandpa //go:generate mockgen -destination=mocks_test.go -package $GOPACKAGE . BlockState,GrandpaState,Network //go:generate mockgen -source=finalisation.go -destination=mock_ephemeral_service_test.go -package $GOPACKAGE . ephemeralService -//go:generate mockery --name Network --structname Network --case underscore --keeptree //go:generate mockgen -destination=mock_telemetry_test.go -package $GOPACKAGE . Telemetry diff --git a/lib/runtime/allocator_test.go b/lib/runtime/allocator_test.go index a775f23e28..1867e828cb 100644 --- a/lib/runtime/allocator_test.go +++ b/lib/runtime/allocator_test.go @@ -9,7 +9,7 @@ import ( "reflect" "testing" - "github.com/stretchr/testify/mock" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/require" ) @@ -257,16 +257,17 @@ var allTests = []testHolder{ // iterates allTests and runs tests on them based on data contained in // test holder func TestAllocator(t *testing.T) { + ctrl := gomock.NewController(t) + for _, test := range allTests { - memmock := newMockMemory(t) + memmock := NewMockMemory(ctrl) const size = 1 << 16 testobj := make([]byte, size) - memmock.On("Data").Return(testobj) - lengthcall := memmock.On("Length") - lengthcall.RunFn = func(args mock.Arguments) { - lengthcall.ReturnArguments = mock.Arguments{uint32(len(testobj))} - } + memmock.EXPECT().Data().Return(testobj).AnyTimes() + memmock.EXPECT().Length().DoAndReturn(func() uint32 { + return uint32(len(testobj)) + }).Times(2) allocator := NewAllocator(memmock, test.offset) @@ -313,20 +314,19 @@ func compareState(allocator FreeingBumpHeapAllocator, state allocatorState, // test that allocator should grow memory if the allocation request is larger than current size func TestShouldGrowMemory(t *testing.T) { - mem := newMockMemory(t) + ctrl := gomock.NewController(t) + + mem := NewMockMemory(ctrl) const size = 1 << 16 testobj := make([]byte, size) - mem.On("Data").Return(testobj) - lengthcall := mem.On("Length") - lengthcall.RunFn = func(args mock.Arguments) { - lengthcall.ReturnArguments = mock.Arguments{uint32(len(testobj))} - } - growcall := mem.On("Grow", mock.Anything) - growcall.RunFn = func(args mock.Arguments) { - arg := args[0].(uint32) + mem.EXPECT().Data().Return(testobj).Times(9) + mem.EXPECT().Length().DoAndReturn(func() uint32 { + return uint32(len(testobj)) + }).Times(5) + mem.EXPECT().Grow(gomock.Any()).DoAndReturn(func(arg uint32) error { testobj = append(testobj, make([]byte, PageSize*arg)...) - growcall.ReturnArguments = mock.Arguments{nil} - } + return nil + }) currentSize := mem.Length() @@ -340,20 +340,19 @@ func TestShouldGrowMemory(t *testing.T) { // test that the allocator should grow memory if it's already full func TestShouldGrowMemoryIfFull(t *testing.T) { - mem := newMockMemory(t) + ctrl := gomock.NewController(t) + + mem := NewMockMemory(ctrl) const size = 1 << 16 testobj := make([]byte, size) - mem.On("Data").Return(testobj) - lengthcall := mem.On("Length").Return(uint32(size)) - lengthcall.RunFn = func(args mock.Arguments) { - lengthcall.ReturnArguments = mock.Arguments{uint32(len(testobj))} - } - growcall := mem.On("Grow", mock.Anything) - growcall.RunFn = func(args mock.Arguments) { - arg := args[0].(uint32) + mem.EXPECT().Data().Return(testobj).Times(18) + mem.EXPECT().Length().DoAndReturn(func() uint32 { + return uint32(len(testobj)) + }).Times(5) + mem.EXPECT().Grow(gomock.Any()).DoAndReturn(func(arg uint32) error { testobj = append(testobj, make([]byte, PageSize*arg)...) - growcall.ReturnArguments = mock.Arguments{nil} - } + return nil + }) currentSize := mem.Length() fbha := NewAllocator(mem, 0) @@ -373,14 +372,16 @@ func TestShouldGrowMemoryIfFull(t *testing.T) { // test to confirm that allocator can allocate the MaxPossibleAllocation func TestShouldAllocateMaxPossibleAllocationSize(t *testing.T) { + ctrl := gomock.NewController(t) + // given, grow heap memory so that we have at least MaxPossibleAllocation available const initialSize = 1 << 16 const pagesNeeded = (MaxPossibleAllocation / PageSize) - (initialSize / PageSize) + 1 - mem := newMockMemory(t) + mem := NewMockMemory(ctrl) const size = initialSize + pagesNeeded*65*1024 testobj := make([]byte, size) - mem.On("Data").Return(testobj) - mem.On("Length").Return(uint32(size)) + mem.EXPECT().Data().Return(testobj).Times(9) + mem.EXPECT().Length().Return(uint32(size)).Times(2) fbha := NewAllocator(mem, 0) @@ -396,8 +397,10 @@ func TestShouldAllocateMaxPossibleAllocationSize(t *testing.T) { // test that allocator should not allocate memory if request is too large func TestShouldNotAllocateIfRequestSizeTooLarge(t *testing.T) { - memory := newMockMemory(t) - memory.On("Length").Return(uint32(1 << 16)) + ctrl := gomock.NewController(t) + + memory := NewMockMemory(ctrl) + memory.EXPECT().Length().Return(uint32(1 << 16)).Times(2) fbha := NewAllocator(memory, 0) diff --git a/lib/runtime/interfaces.go b/lib/runtime/interfaces.go index f429a17729..a69f444689 100644 --- a/lib/runtime/interfaces.go +++ b/lib/runtime/interfaces.go @@ -46,8 +46,6 @@ type BasicStorage interface { Del(key []byte) error } -//go:generate mockery --name TransactionState --structname TransactionState --case underscore --keeptree - // TransactionState interface for adding transactions to pool type TransactionState interface { AddToPool(vt *transaction.ValidTransaction) common.Hash diff --git a/lib/runtime/memory.go b/lib/runtime/memory.go index 42b20999f9..1ad59ff279 100644 --- a/lib/runtime/memory.go +++ b/lib/runtime/memory.go @@ -6,8 +6,6 @@ package runtime // PageSize is 65kb const PageSize = 65536 -//go:generate mockery --name Memory --structname mockMemory --case underscore --inpackage --filename mock_memory_test.go - // Memory is a raw memory interface type Memory interface { Data() []byte diff --git a/lib/runtime/mock_memory_test.go b/lib/runtime/mock_memory_test.go deleted file mode 100644 index c3cb0c9a37..0000000000 --- a/lib/runtime/mock_memory_test.go +++ /dev/null @@ -1,69 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package runtime - -import mock "github.com/stretchr/testify/mock" - -// mockMemory is an autogenerated mock type for the Memory type -type mockMemory struct { - mock.Mock -} - -// Data provides a mock function with given fields: -func (_m *mockMemory) Data() []byte { - ret := _m.Called() - - var r0 []byte - if rf, ok := ret.Get(0).(func() []byte); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - return r0 -} - -// Grow provides a mock function with given fields: _a0 -func (_m *mockMemory) Grow(_a0 uint32) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func(uint32) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// Length provides a mock function with given fields: -func (_m *mockMemory) Length() uint32 { - ret := _m.Called() - - var r0 uint32 - if rf, ok := ret.Get(0).(func() uint32); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(uint32) - } - - return r0 -} - -type mockConstructorTestingTnewMockMemory interface { - mock.TestingT - Cleanup(func()) -} - -// newMockMemory creates a new instance of mockMemory. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func newMockMemory(t mockConstructorTestingTnewMockMemory) *mockMemory { - mock := &mockMemory{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/lib/runtime/mocks/instance.go b/lib/runtime/mocks/instance.go deleted file mode 100644 index d4b10e4902..0000000000 --- a/lib/runtime/mocks/instance.go +++ /dev/null @@ -1,437 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - common "github.com/ChainSafe/gossamer/lib/common" - keystore "github.com/ChainSafe/gossamer/lib/keystore" - - mock "github.com/stretchr/testify/mock" - - runtime "github.com/ChainSafe/gossamer/lib/runtime" - - transaction "github.com/ChainSafe/gossamer/lib/transaction" - - types "github.com/ChainSafe/gossamer/dot/types" -) - -// Instance is an autogenerated mock type for the Instance type -type Instance struct { - mock.Mock -} - -// ApplyExtrinsic provides a mock function with given fields: data -func (_m *Instance) ApplyExtrinsic(data types.Extrinsic) ([]byte, error) { - ret := _m.Called(data) - - var r0 []byte - if rf, ok := ret.Get(0).(func(types.Extrinsic) []byte); ok { - r0 = rf(data) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(types.Extrinsic) error); ok { - r1 = rf(data) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// BabeConfiguration provides a mock function with given fields: -func (_m *Instance) BabeConfiguration() (*types.BabeConfiguration, error) { - ret := _m.Called() - - var r0 *types.BabeConfiguration - if rf, ok := ret.Get(0).(func() *types.BabeConfiguration); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*types.BabeConfiguration) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// CheckInherents provides a mock function with given fields: -func (_m *Instance) CheckInherents() { - _m.Called() -} - -// DecodeSessionKeys provides a mock function with given fields: enc -func (_m *Instance) DecodeSessionKeys(enc []byte) ([]byte, error) { - ret := _m.Called(enc) - - var r0 []byte - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(enc) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(enc) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// Exec provides a mock function with given fields: function, data -func (_m *Instance) Exec(function string, data []byte) ([]byte, error) { - ret := _m.Called(function, data) - - var r0 []byte - if rf, ok := ret.Get(0).(func(string, []byte) []byte); ok { - r0 = rf(function, data) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(string, []byte) error); ok { - r1 = rf(function, data) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// ExecuteBlock provides a mock function with given fields: block -func (_m *Instance) ExecuteBlock(block *types.Block) ([]byte, error) { - ret := _m.Called(block) - - var r0 []byte - if rf, ok := ret.Get(0).(func(*types.Block) []byte); ok { - r0 = rf(block) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(*types.Block) error); ok { - r1 = rf(block) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// FinalizeBlock provides a mock function with given fields: -func (_m *Instance) FinalizeBlock() (*types.Header, error) { - ret := _m.Called() - - var r0 *types.Header - if rf, ok := ret.Get(0).(func() *types.Header); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*types.Header) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// GenerateSessionKeys provides a mock function with given fields: -func (_m *Instance) GenerateSessionKeys() { - _m.Called() -} - -// GetCodeHash provides a mock function with given fields: -func (_m *Instance) GetCodeHash() common.Hash { - ret := _m.Called() - - var r0 common.Hash - if rf, ok := ret.Get(0).(func() common.Hash); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - return r0 -} - -// GrandpaAuthorities provides a mock function with given fields: -func (_m *Instance) GrandpaAuthorities() ([]types.Authority, error) { - ret := _m.Called() - - var r0 []types.Authority - if rf, ok := ret.Get(0).(func() []types.Authority); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]types.Authority) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// InherentExtrinsics provides a mock function with given fields: data -func (_m *Instance) InherentExtrinsics(data []byte) ([]byte, error) { - ret := _m.Called(data) - - var r0 []byte - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(data) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(data) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// InitializeBlock provides a mock function with given fields: header -func (_m *Instance) InitializeBlock(header *types.Header) error { - ret := _m.Called(header) - - var r0 error - if rf, ok := ret.Get(0).(func(*types.Header) error); ok { - r0 = rf(header) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// Keystore provides a mock function with given fields: -func (_m *Instance) Keystore() *keystore.GlobalKeystore { - ret := _m.Called() - - var r0 *keystore.GlobalKeystore - if rf, ok := ret.Get(0).(func() *keystore.GlobalKeystore); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*keystore.GlobalKeystore) - } - } - - return r0 -} - -// Metadata provides a mock function with given fields: -func (_m *Instance) Metadata() ([]byte, error) { - ret := _m.Called() - - var r0 []byte - if rf, ok := ret.Get(0).(func() []byte); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).([]byte) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// NetworkService provides a mock function with given fields: -func (_m *Instance) NetworkService() runtime.BasicNetwork { - ret := _m.Called() - - var r0 runtime.BasicNetwork - if rf, ok := ret.Get(0).(func() runtime.BasicNetwork); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(runtime.BasicNetwork) - } - } - - return r0 -} - -// NodeStorage provides a mock function with given fields: -func (_m *Instance) NodeStorage() runtime.NodeStorage { - ret := _m.Called() - - var r0 runtime.NodeStorage - if rf, ok := ret.Get(0).(func() runtime.NodeStorage); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(runtime.NodeStorage) - } - - return r0 -} - -// OffchainWorker provides a mock function with given fields: -func (_m *Instance) OffchainWorker() { - _m.Called() -} - -// PaymentQueryInfo provides a mock function with given fields: ext -func (_m *Instance) PaymentQueryInfo(ext []byte) (*types.RuntimeDispatchInfo, error) { - ret := _m.Called(ext) - - var r0 *types.RuntimeDispatchInfo - if rf, ok := ret.Get(0).(func([]byte) *types.RuntimeDispatchInfo); ok { - r0 = rf(ext) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*types.RuntimeDispatchInfo) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(ext) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// RandomSeed provides a mock function with given fields: -func (_m *Instance) RandomSeed() { - _m.Called() -} - -// SetContextStorage provides a mock function with given fields: s -func (_m *Instance) SetContextStorage(s runtime.Storage) { - _m.Called(s) -} - -// Stop provides a mock function with given fields: -func (_m *Instance) Stop() { - _m.Called() -} - -// UpdateRuntimeCode provides a mock function with given fields: _a0 -func (_m *Instance) UpdateRuntimeCode(_a0 []byte) error { - ret := _m.Called(_a0) - - var r0 error - if rf, ok := ret.Get(0).(func([]byte) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// ValidateTransaction provides a mock function with given fields: e -func (_m *Instance) ValidateTransaction(e types.Extrinsic) (*transaction.Validity, error) { - ret := _m.Called(e) - - var r0 *transaction.Validity - if rf, ok := ret.Get(0).(func(types.Extrinsic) *transaction.Validity); ok { - r0 = rf(e) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*transaction.Validity) - } - } - - var r1 error - if rf, ok := ret.Get(1).(func(types.Extrinsic) error); ok { - r1 = rf(e) - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// Validator provides a mock function with given fields: -func (_m *Instance) Validator() bool { - ret := _m.Called() - - var r0 bool - if rf, ok := ret.Get(0).(func() bool); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(bool) - } - - return r0 -} - -// Version provides a mock function with given fields: -func (_m *Instance) Version() runtime.Version { - ret := _m.Called() - - var r0 runtime.Version - if rf, ok := ret.Get(0).(func() runtime.Version); ok { - r0 = rf() - } else { - r0 = ret.Get(0).(runtime.Version) - } - - return r0 -} - -type mockConstructorTestingTNewInstance interface { - mock.TestingT - Cleanup(func()) -} - -// NewInstance creates a new instance of Instance. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewInstance(t mockConstructorTestingTNewInstance) *Instance { - mock := &Instance{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/lib/runtime/mocks/mocks.go b/lib/runtime/mocks/mocks.go new file mode 100644 index 0000000000..a3a2b752af --- /dev/null +++ b/lib/runtime/mocks/mocks.go @@ -0,0 +1,425 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/ChainSafe/gossamer/lib/runtime (interfaces: Instance,TransactionState) + +// Package mocks is a generated GoMock package. +package mocks + +import ( + reflect "reflect" + + types "github.com/ChainSafe/gossamer/dot/types" + common "github.com/ChainSafe/gossamer/lib/common" + keystore "github.com/ChainSafe/gossamer/lib/keystore" + runtime "github.com/ChainSafe/gossamer/lib/runtime" + transaction "github.com/ChainSafe/gossamer/lib/transaction" + gomock "github.com/golang/mock/gomock" +) + +// MockInstance is a mock of Instance interface. +type MockInstance struct { + ctrl *gomock.Controller + recorder *MockInstanceMockRecorder +} + +// MockInstanceMockRecorder is the mock recorder for MockInstance. +type MockInstanceMockRecorder struct { + mock *MockInstance +} + +// NewMockInstance creates a new mock instance. +func NewMockInstance(ctrl *gomock.Controller) *MockInstance { + mock := &MockInstance{ctrl: ctrl} + mock.recorder = &MockInstanceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockInstance) EXPECT() *MockInstanceMockRecorder { + return m.recorder +} + +// ApplyExtrinsic mocks base method. +func (m *MockInstance) ApplyExtrinsic(arg0 types.Extrinsic) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ApplyExtrinsic", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ApplyExtrinsic indicates an expected call of ApplyExtrinsic. +func (mr *MockInstanceMockRecorder) ApplyExtrinsic(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplyExtrinsic", reflect.TypeOf((*MockInstance)(nil).ApplyExtrinsic), arg0) +} + +// BabeConfiguration mocks base method. +func (m *MockInstance) BabeConfiguration() (*types.BabeConfiguration, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "BabeConfiguration") + ret0, _ := ret[0].(*types.BabeConfiguration) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// BabeConfiguration indicates an expected call of BabeConfiguration. +func (mr *MockInstanceMockRecorder) BabeConfiguration() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BabeConfiguration", reflect.TypeOf((*MockInstance)(nil).BabeConfiguration)) +} + +// CheckInherents mocks base method. +func (m *MockInstance) CheckInherents() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "CheckInherents") +} + +// CheckInherents indicates an expected call of CheckInherents. +func (mr *MockInstanceMockRecorder) CheckInherents() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckInherents", reflect.TypeOf((*MockInstance)(nil).CheckInherents)) +} + +// DecodeSessionKeys mocks base method. +func (m *MockInstance) DecodeSessionKeys(arg0 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "DecodeSessionKeys", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// DecodeSessionKeys indicates an expected call of DecodeSessionKeys. +func (mr *MockInstanceMockRecorder) DecodeSessionKeys(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DecodeSessionKeys", reflect.TypeOf((*MockInstance)(nil).DecodeSessionKeys), arg0) +} + +// Exec mocks base method. +func (m *MockInstance) Exec(arg0 string, arg1 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Exec", arg0, arg1) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Exec indicates an expected call of Exec. +func (mr *MockInstanceMockRecorder) Exec(arg0, arg1 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Exec", reflect.TypeOf((*MockInstance)(nil).Exec), arg0, arg1) +} + +// ExecuteBlock mocks base method. +func (m *MockInstance) ExecuteBlock(arg0 *types.Block) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ExecuteBlock", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ExecuteBlock indicates an expected call of ExecuteBlock. +func (mr *MockInstanceMockRecorder) ExecuteBlock(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecuteBlock", reflect.TypeOf((*MockInstance)(nil).ExecuteBlock), arg0) +} + +// FinalizeBlock mocks base method. +func (m *MockInstance) FinalizeBlock() (*types.Header, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "FinalizeBlock") + ret0, _ := ret[0].(*types.Header) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// FinalizeBlock indicates an expected call of FinalizeBlock. +func (mr *MockInstanceMockRecorder) FinalizeBlock() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FinalizeBlock", reflect.TypeOf((*MockInstance)(nil).FinalizeBlock)) +} + +// GenerateSessionKeys mocks base method. +func (m *MockInstance) GenerateSessionKeys() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "GenerateSessionKeys") +} + +// GenerateSessionKeys indicates an expected call of GenerateSessionKeys. +func (mr *MockInstanceMockRecorder) GenerateSessionKeys() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GenerateSessionKeys", reflect.TypeOf((*MockInstance)(nil).GenerateSessionKeys)) +} + +// GetCodeHash mocks base method. +func (m *MockInstance) GetCodeHash() common.Hash { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GetCodeHash") + ret0, _ := ret[0].(common.Hash) + return ret0 +} + +// GetCodeHash indicates an expected call of GetCodeHash. +func (mr *MockInstanceMockRecorder) GetCodeHash() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCodeHash", reflect.TypeOf((*MockInstance)(nil).GetCodeHash)) +} + +// GrandpaAuthorities mocks base method. +func (m *MockInstance) GrandpaAuthorities() ([]types.Authority, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "GrandpaAuthorities") + ret0, _ := ret[0].([]types.Authority) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// GrandpaAuthorities indicates an expected call of GrandpaAuthorities. +func (mr *MockInstanceMockRecorder) GrandpaAuthorities() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GrandpaAuthorities", reflect.TypeOf((*MockInstance)(nil).GrandpaAuthorities)) +} + +// InherentExtrinsics mocks base method. +func (m *MockInstance) InherentExtrinsics(arg0 []byte) ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "InherentExtrinsics", arg0) + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// InherentExtrinsics indicates an expected call of InherentExtrinsics. +func (mr *MockInstanceMockRecorder) InherentExtrinsics(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InherentExtrinsics", reflect.TypeOf((*MockInstance)(nil).InherentExtrinsics), arg0) +} + +// InitializeBlock mocks base method. +func (m *MockInstance) InitializeBlock(arg0 *types.Header) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "InitializeBlock", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// InitializeBlock indicates an expected call of InitializeBlock. +func (mr *MockInstanceMockRecorder) InitializeBlock(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InitializeBlock", reflect.TypeOf((*MockInstance)(nil).InitializeBlock), arg0) +} + +// Keystore mocks base method. +func (m *MockInstance) Keystore() *keystore.GlobalKeystore { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Keystore") + ret0, _ := ret[0].(*keystore.GlobalKeystore) + return ret0 +} + +// Keystore indicates an expected call of Keystore. +func (mr *MockInstanceMockRecorder) Keystore() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Keystore", reflect.TypeOf((*MockInstance)(nil).Keystore)) +} + +// Metadata mocks base method. +func (m *MockInstance) Metadata() ([]byte, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Metadata") + ret0, _ := ret[0].([]byte) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// Metadata indicates an expected call of Metadata. +func (mr *MockInstanceMockRecorder) Metadata() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Metadata", reflect.TypeOf((*MockInstance)(nil).Metadata)) +} + +// NetworkService mocks base method. +func (m *MockInstance) NetworkService() runtime.BasicNetwork { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "NetworkService") + ret0, _ := ret[0].(runtime.BasicNetwork) + return ret0 +} + +// NetworkService indicates an expected call of NetworkService. +func (mr *MockInstanceMockRecorder) NetworkService() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NetworkService", reflect.TypeOf((*MockInstance)(nil).NetworkService)) +} + +// NodeStorage mocks base method. +func (m *MockInstance) NodeStorage() runtime.NodeStorage { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "NodeStorage") + ret0, _ := ret[0].(runtime.NodeStorage) + return ret0 +} + +// NodeStorage indicates an expected call of NodeStorage. +func (mr *MockInstanceMockRecorder) NodeStorage() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NodeStorage", reflect.TypeOf((*MockInstance)(nil).NodeStorage)) +} + +// OffchainWorker mocks base method. +func (m *MockInstance) OffchainWorker() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "OffchainWorker") +} + +// OffchainWorker indicates an expected call of OffchainWorker. +func (mr *MockInstanceMockRecorder) OffchainWorker() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OffchainWorker", reflect.TypeOf((*MockInstance)(nil).OffchainWorker)) +} + +// PaymentQueryInfo mocks base method. +func (m *MockInstance) PaymentQueryInfo(arg0 []byte) (*types.RuntimeDispatchInfo, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "PaymentQueryInfo", arg0) + ret0, _ := ret[0].(*types.RuntimeDispatchInfo) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// PaymentQueryInfo indicates an expected call of PaymentQueryInfo. +func (mr *MockInstanceMockRecorder) PaymentQueryInfo(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PaymentQueryInfo", reflect.TypeOf((*MockInstance)(nil).PaymentQueryInfo), arg0) +} + +// RandomSeed mocks base method. +func (m *MockInstance) RandomSeed() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "RandomSeed") +} + +// RandomSeed indicates an expected call of RandomSeed. +func (mr *MockInstanceMockRecorder) RandomSeed() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RandomSeed", reflect.TypeOf((*MockInstance)(nil).RandomSeed)) +} + +// SetContextStorage mocks base method. +func (m *MockInstance) SetContextStorage(arg0 runtime.Storage) { + m.ctrl.T.Helper() + m.ctrl.Call(m, "SetContextStorage", arg0) +} + +// SetContextStorage indicates an expected call of SetContextStorage. +func (mr *MockInstanceMockRecorder) SetContextStorage(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetContextStorage", reflect.TypeOf((*MockInstance)(nil).SetContextStorage), arg0) +} + +// Stop mocks base method. +func (m *MockInstance) Stop() { + m.ctrl.T.Helper() + m.ctrl.Call(m, "Stop") +} + +// Stop indicates an expected call of Stop. +func (mr *MockInstanceMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockInstance)(nil).Stop)) +} + +// UpdateRuntimeCode mocks base method. +func (m *MockInstance) UpdateRuntimeCode(arg0 []byte) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "UpdateRuntimeCode", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// UpdateRuntimeCode indicates an expected call of UpdateRuntimeCode. +func (mr *MockInstanceMockRecorder) UpdateRuntimeCode(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateRuntimeCode", reflect.TypeOf((*MockInstance)(nil).UpdateRuntimeCode), arg0) +} + +// ValidateTransaction mocks base method. +func (m *MockInstance) ValidateTransaction(arg0 types.Extrinsic) (*transaction.Validity, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "ValidateTransaction", arg0) + ret0, _ := ret[0].(*transaction.Validity) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// ValidateTransaction indicates an expected call of ValidateTransaction. +func (mr *MockInstanceMockRecorder) ValidateTransaction(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ValidateTransaction", reflect.TypeOf((*MockInstance)(nil).ValidateTransaction), arg0) +} + +// Validator mocks base method. +func (m *MockInstance) Validator() bool { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Validator") + ret0, _ := ret[0].(bool) + return ret0 +} + +// Validator indicates an expected call of Validator. +func (mr *MockInstanceMockRecorder) Validator() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validator", reflect.TypeOf((*MockInstance)(nil).Validator)) +} + +// Version mocks base method. +func (m *MockInstance) Version() runtime.Version { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Version") + ret0, _ := ret[0].(runtime.Version) + return ret0 +} + +// Version indicates an expected call of Version. +func (mr *MockInstanceMockRecorder) Version() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Version", reflect.TypeOf((*MockInstance)(nil).Version)) +} + +// MockTransactionState is a mock of TransactionState interface. +type MockTransactionState struct { + ctrl *gomock.Controller + recorder *MockTransactionStateMockRecorder +} + +// MockTransactionStateMockRecorder is the mock recorder for MockTransactionState. +type MockTransactionStateMockRecorder struct { + mock *MockTransactionState +} + +// NewMockTransactionState creates a new mock instance. +func NewMockTransactionState(ctrl *gomock.Controller) *MockTransactionState { + mock := &MockTransactionState{ctrl: ctrl} + mock.recorder = &MockTransactionStateMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockTransactionState) EXPECT() *MockTransactionStateMockRecorder { + return m.recorder +} + +// AddToPool mocks base method. +func (m *MockTransactionState) AddToPool(arg0 *transaction.ValidTransaction) common.Hash { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "AddToPool", arg0) + ret0, _ := ret[0].(common.Hash) + return ret0 +} + +// AddToPool indicates an expected call of AddToPool. +func (mr *MockTransactionStateMockRecorder) AddToPool(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddToPool", reflect.TypeOf((*MockTransactionState)(nil).AddToPool), arg0) +} diff --git a/lib/runtime/mocks/transaction_state.go b/lib/runtime/mocks/transaction_state.go deleted file mode 100644 index 277b98c2ee..0000000000 --- a/lib/runtime/mocks/transaction_state.go +++ /dev/null @@ -1,46 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import ( - common "github.com/ChainSafe/gossamer/lib/common" - mock "github.com/stretchr/testify/mock" - - transaction "github.com/ChainSafe/gossamer/lib/transaction" -) - -// TransactionState is an autogenerated mock type for the TransactionState type -type TransactionState struct { - mock.Mock -} - -// AddToPool provides a mock function with given fields: vt -func (_m *TransactionState) AddToPool(vt *transaction.ValidTransaction) common.Hash { - ret := _m.Called(vt) - - var r0 common.Hash - if rf, ok := ret.Get(0).(func(*transaction.ValidTransaction) common.Hash); ok { - r0 = rf(vt) - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(common.Hash) - } - } - - return r0 -} - -type mockConstructorTestingTNewTransactionState interface { - mock.TestingT - Cleanup(func()) -} - -// NewTransactionState creates a new instance of TransactionState. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewTransactionState(t mockConstructorTestingTNewTransactionState) *TransactionState { - mock := &TransactionState{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/lib/runtime/mocks_generate_test.go b/lib/runtime/mocks_generate_test.go new file mode 100644 index 0000000000..9388b76453 --- /dev/null +++ b/lib/runtime/mocks_generate_test.go @@ -0,0 +1,7 @@ +// Copyright 2023 ChainSafe Systems (ON) +// SPDX-License-Identifier: LGPL-3.0-only + +package runtime + +//go:generate mockgen -destination=mocks/mocks.go -package mocks . Instance,TransactionState +//go:generate mockgen -destination=mocks_test.go -package $GOPACKAGE . Memory diff --git a/lib/runtime/mocks_test.go b/lib/runtime/mocks_test.go new file mode 100644 index 0000000000..a5626539e8 --- /dev/null +++ b/lib/runtime/mocks_test.go @@ -0,0 +1,76 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/ChainSafe/gossamer/lib/runtime (interfaces: Memory) + +// Package runtime is a generated GoMock package. +package runtime + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockMemory is a mock of Memory interface. +type MockMemory struct { + ctrl *gomock.Controller + recorder *MockMemoryMockRecorder +} + +// MockMemoryMockRecorder is the mock recorder for MockMemory. +type MockMemoryMockRecorder struct { + mock *MockMemory +} + +// NewMockMemory creates a new mock instance. +func NewMockMemory(ctrl *gomock.Controller) *MockMemory { + mock := &MockMemory{ctrl: ctrl} + mock.recorder = &MockMemoryMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockMemory) EXPECT() *MockMemoryMockRecorder { + return m.recorder +} + +// Data mocks base method. +func (m *MockMemory) Data() []byte { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Data") + ret0, _ := ret[0].([]byte) + return ret0 +} + +// Data indicates an expected call of Data. +func (mr *MockMemoryMockRecorder) Data() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Data", reflect.TypeOf((*MockMemory)(nil).Data)) +} + +// Grow mocks base method. +func (m *MockMemory) Grow(arg0 uint32) error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Grow", arg0) + ret0, _ := ret[0].(error) + return ret0 +} + +// Grow indicates an expected call of Grow. +func (mr *MockMemoryMockRecorder) Grow(arg0 interface{}) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Grow", reflect.TypeOf((*MockMemory)(nil).Grow), arg0) +} + +// Length mocks base method. +func (m *MockMemory) Length() uint32 { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Length") + ret0, _ := ret[0].(uint32) + return ret0 +} + +// Length indicates an expected call of Length. +func (mr *MockMemoryMockRecorder) Length() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Length", reflect.TypeOf((*MockMemory)(nil).Length)) +} diff --git a/lib/runtime/test_helpers.go b/lib/runtime/test_helpers.go index 13fc426f2c..7a98fd4022 100644 --- a/lib/runtime/test_helpers.go +++ b/lib/runtime/test_helpers.go @@ -263,8 +263,6 @@ func NewTestExtrinsic(t *testing.T, rt MetadataVersioner, genHash, blockHash com return extEnc } -//go:generate mockery --name Instance --structname Instance --case underscore --keeptree - // Instance is the interface to interact with the runtime. type Instance interface { UpdateRuntimeCode([]byte) error diff --git a/lib/runtime/wasmer/test_helpers.go b/lib/runtime/wasmer/test_helpers.go index b32887d969..e01a519562 100644 --- a/lib/runtime/wasmer/test_helpers.go +++ b/lib/runtime/wasmer/test_helpers.go @@ -14,6 +14,7 @@ import ( "github.com/ChainSafe/gossamer/lib/runtime/mocks" "github.com/ChainSafe/gossamer/lib/runtime/storage" "github.com/ChainSafe/gossamer/lib/trie" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/require" ) @@ -32,7 +33,9 @@ func NewTestInstance(t *testing.T, targetRuntime string) *Instance { func NewTestInstanceWithTrie(t *testing.T, targetRuntime string, tt *trie.Trie) *Instance { t.Helper() - cfg := setupConfig(t, tt, DefaultTestLogLvl, common.NoNetworkRole, targetRuntime) + ctrl := gomock.NewController(t) + + cfg := setupConfig(t, ctrl, tt, DefaultTestLogLvl, common.NoNetworkRole, targetRuntime) targetRuntime, err := runtime.GetRuntime(context.Background(), targetRuntime) require.NoError(t, err) @@ -42,7 +45,7 @@ func NewTestInstanceWithTrie(t *testing.T, targetRuntime string, tt *trie.Trie) return r } -func setupConfig(t *testing.T, tt *trie.Trie, lvl log.Level, +func setupConfig(t *testing.T, ctrl *gomock.Controller, tt *trie.Trie, lvl log.Level, role common.Roles, targetRuntime string) Config { t.Helper() @@ -68,7 +71,7 @@ func setupConfig(t *testing.T, tt *trie.Trie, lvl log.Level, LogLvl: lvl, NodeStorage: ns, Network: new(runtime.TestRuntimeNetwork), - Transaction: mocks.NewTransactionState(t), + Transaction: mocks.NewMockTransactionState(ctrl), Role: role, testVersion: version, } diff --git a/lib/services/mocks/service.go b/lib/services/mocks/service.go deleted file mode 100644 index 4b8e7c862a..0000000000 --- a/lib/services/mocks/service.go +++ /dev/null @@ -1,53 +0,0 @@ -// Code generated by mockery v2.14.1. DO NOT EDIT. - -package mocks - -import mock "github.com/stretchr/testify/mock" - -// Service is an autogenerated mock type for the Service type -type Service struct { - mock.Mock -} - -// Start provides a mock function with given fields: -func (_m *Service) Start() error { - ret := _m.Called() - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// Stop provides a mock function with given fields: -func (_m *Service) Stop() error { - ret := _m.Called() - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -type mockConstructorTestingTNewService interface { - mock.TestingT - Cleanup(func()) -} - -// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -func NewService(t mockConstructorTestingTNewService) *Service { - mock := &Service{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/lib/services/mocks_test.go b/lib/services/mocks_test.go new file mode 100644 index 0000000000..ab09a2cb8c --- /dev/null +++ b/lib/services/mocks_test.go @@ -0,0 +1,62 @@ +// Code generated by MockGen. DO NOT EDIT. +// Source: github.com/ChainSafe/gossamer/lib/services (interfaces: Service) + +// Package services is a generated GoMock package. +package services + +import ( + reflect "reflect" + + gomock "github.com/golang/mock/gomock" +) + +// MockService is a mock of Service interface. +type MockService struct { + ctrl *gomock.Controller + recorder *MockServiceMockRecorder +} + +// MockServiceMockRecorder is the mock recorder for MockService. +type MockServiceMockRecorder struct { + mock *MockService +} + +// NewMockService creates a new mock instance. +func NewMockService(ctrl *gomock.Controller) *MockService { + mock := &MockService{ctrl: ctrl} + mock.recorder = &MockServiceMockRecorder{mock} + return mock +} + +// EXPECT returns an object that allows the caller to indicate expected use. +func (m *MockService) EXPECT() *MockServiceMockRecorder { + return m.recorder +} + +// Start mocks base method. +func (m *MockService) Start() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Start") + ret0, _ := ret[0].(error) + return ret0 +} + +// Start indicates an expected call of Start. +func (mr *MockServiceMockRecorder) Start() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockService)(nil).Start)) +} + +// Stop mocks base method. +func (m *MockService) Stop() error { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "Stop") + ret0, _ := ret[0].(error) + return ret0 +} + +// Stop indicates an expected call of Stop. +func (mr *MockServiceMockRecorder) Stop() *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockService)(nil).Stop)) +} diff --git a/lib/services/services.go b/lib/services/services.go index b1103eb089..876202f169 100644 --- a/lib/services/services.go +++ b/lib/services/services.go @@ -7,7 +7,7 @@ import ( "reflect" ) -//go:generate mockery --name Service --structname Service --case underscore --keeptree +//go:generate mockgen -destination=mocks_test.go -package $GOPACKAGE . Service // Service must be implemented by all services type Service interface { diff --git a/lib/services/services_test.go b/lib/services/services_test.go index ff3bf254fe..c7ed5b0aac 100644 --- a/lib/services/services_test.go +++ b/lib/services/services_test.go @@ -8,25 +8,25 @@ import ( "testing" "github.com/ChainSafe/gossamer/internal/log" - "github.com/ChainSafe/gossamer/lib/services/mocks" + "github.com/golang/mock/gomock" "github.com/stretchr/testify/require" ) func TestServiceRegistry_RegisterService(t *testing.T) { r := NewServiceRegistry(log.New(log.SetWriter(io.Discard))) - r.RegisterService(&mocks.Service{}) - r.RegisterService(&mocks.Service{}) + r.RegisterService(&MockService{}) + r.RegisterService(&MockService{}) require.Len(t, r.services, 1) } func TestServiceRegistry_StartStopAll(t *testing.T) { r := NewServiceRegistry(log.New(log.SetWriter(io.Discard))) - - m := mocks.NewService(t) - m.On("Start").Return(nil) - m.On("Stop").Return(nil) + ctrl := gomock.NewController(t) + m := NewMockService(ctrl) + m.EXPECT().Start().Return(nil) + m.EXPECT().Stop().Return(nil) r.RegisterService(m) @@ -37,7 +37,7 @@ func TestServiceRegistry_StartStopAll(t *testing.T) { func TestServiceRegistry_Get_Err(t *testing.T) { r := NewServiceRegistry(log.New(log.SetWriter(io.Discard))) - a := mocks.NewService(t) + a := NewMockService(nil) r.RegisterService(a) require.NotNil(t, r.Get(a))