package discord

import (
	"github.com/stretchr/testify/mock"
)

type GuildServiceMock struct {
	mock.Mock
}

func (gs *GuildServiceMock) GetGuild(guildID string) (IGuild, error) {
	args := gs.Called(guildID)

	if args.Get(0) == nil {
		return nil, args.Error(1)
	}

	return args.Get(0).(IGuild), args.Error(1)
}