24 lines
428 B
Go
24 lines
428 B
Go
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)
|
|
}
|
|
|
|
func (gs *GuildServiceMock) Client() IDiscordClient {
|
|
args := gs.Called()
|
|
return args.Get(0).(IDiscordClient)
|
|
}
|