71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
// Code generated by mockery v2.9.4. DO NOT EDIT.
|
|
|
|
package mocks
|
|
|
|
import (
|
|
context "context"
|
|
io "io"
|
|
|
|
mock "github.com/stretchr/testify/mock"
|
|
|
|
youtube "github.com/kkdai/youtube/v2"
|
|
)
|
|
|
|
// YoutubeClient is an autogenerated mock type for the YoutubeClient type
|
|
type YoutubeClient struct {
|
|
mock.Mock
|
|
}
|
|
|
|
// GetStreamContext provides a mock function with given fields: ctx, video, format
|
|
func (_m *YoutubeClient) GetStreamContext(ctx context.Context, video *youtube.Video, format *youtube.Format) (io.ReadCloser, int64, error) {
|
|
ret := _m.Called(ctx, video, format)
|
|
|
|
var r0 io.ReadCloser
|
|
if rf, ok := ret.Get(0).(func(context.Context, *youtube.Video, *youtube.Format) io.ReadCloser); ok {
|
|
r0 = rf(ctx, video, format)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(io.ReadCloser)
|
|
}
|
|
}
|
|
|
|
var r1 int64
|
|
if rf, ok := ret.Get(1).(func(context.Context, *youtube.Video, *youtube.Format) int64); ok {
|
|
r1 = rf(ctx, video, format)
|
|
} else {
|
|
r1 = ret.Get(1).(int64)
|
|
}
|
|
|
|
var r2 error
|
|
if rf, ok := ret.Get(2).(func(context.Context, *youtube.Video, *youtube.Format) error); ok {
|
|
r2 = rf(ctx, video, format)
|
|
} else {
|
|
r2 = ret.Error(2)
|
|
}
|
|
|
|
return r0, r1, r2
|
|
}
|
|
|
|
// GetVideoContext provides a mock function with given fields: ctx, id
|
|
func (_m *YoutubeClient) GetVideoContext(ctx context.Context, id string) (*youtube.Video, error) {
|
|
ret := _m.Called(ctx, id)
|
|
|
|
var r0 *youtube.Video
|
|
if rf, ok := ret.Get(0).(func(context.Context, string) *youtube.Video); ok {
|
|
r0 = rf(ctx, id)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*youtube.Video)
|
|
}
|
|
}
|
|
|
|
var r1 error
|
|
if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
|
|
r1 = rf(ctx, id)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|