// Code generated by mockery v1.0.0. DO NOT EDIT.

package mocks

import (
	context "context"

	kms "github.com/aws/aws-sdk-go/service/kms"
	mock "github.com/stretchr/testify/mock"
)

// EncryptionAtRestManager is an autogenerated mock type for the EncryptionAtRestManager type
type EncryptionAtRestManager struct {
	mock.Mock
}

// AllGrants provides a mock function with given fields: ctx
func (_m *EncryptionAtRestManager) AllGrants(ctx context.Context) ([]*kms.GrantListEntry, error) {
	ret := _m.Called(ctx)

	var r0 []*kms.GrantListEntry
	if rf, ok := ret.Get(0).(func(context.Context) []*kms.GrantListEntry); ok {
		r0 = rf(ctx)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).([]*kms.GrantListEntry)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context) error); ok {
		r1 = rf(ctx)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// FindByGrantID provides a mock function with given fields: ctx, grantID
func (_m *EncryptionAtRestManager) FindByGrantID(ctx context.Context, grantID string) (*kms.GrantListEntry, error) {
	ret := _m.Called(ctx, grantID)

	var r0 *kms.GrantListEntry
	if rf, ok := ret.Get(0).(func(context.Context, string) *kms.GrantListEntry); ok {
		r0 = rf(ctx, grantID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*kms.GrantListEntry)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
		r1 = rf(ctx, grantID)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// GrantEncryptionAtRest provides a mock function with given fields: ctx, iamRoleARN
func (_m *EncryptionAtRestManager) GrantEncryptionAtRest(ctx context.Context, iamRoleARN string) (string, error) {
	ret := _m.Called(ctx, iamRoleARN)

	var r0 string
	if rf, ok := ret.Get(0).(func(context.Context, string) string); ok {
		r0 = rf(ctx, iamRoleARN)
	} else {
		r0 = ret.Get(0).(string)
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
		r1 = rf(ctx, iamRoleARN)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}
