move gotd fork into repo. (#111)
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
This commit is contained in:
@@ -0,0 +1,116 @@
|
||||
package tgmock
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tgerr"
|
||||
)
|
||||
|
||||
func (i *Mock) request(fn func(b bin.Encoder)) *RequestBuilder {
|
||||
return &RequestBuilder{
|
||||
mock: i,
|
||||
expectFn: fn,
|
||||
times: 1,
|
||||
}
|
||||
}
|
||||
|
||||
// Expect creates builder of new expected call.
|
||||
func (i *Mock) Expect() *RequestBuilder {
|
||||
return i.request(nil)
|
||||
}
|
||||
|
||||
// ExpectFunc creates builder of new expected call with given input checker.
|
||||
func (i *Mock) ExpectFunc(fn func(b bin.Encoder)) *RequestBuilder {
|
||||
return i.request(fn)
|
||||
}
|
||||
|
||||
// ExpectCall creates builder of new expected call with given input.
|
||||
func (i *Mock) ExpectCall(body bin.Encoder) *RequestBuilder {
|
||||
return i.request(func(got bin.Encoder) {
|
||||
i.assert.Equal(body, got)
|
||||
})
|
||||
}
|
||||
|
||||
// RequestBuilder is builder of expected RPC request.
|
||||
type RequestBuilder struct {
|
||||
mock *Mock
|
||||
expectFn func(b bin.Encoder)
|
||||
times int
|
||||
}
|
||||
|
||||
// N sets count of expected calls.
|
||||
func (b *RequestBuilder) N(times int) *RequestBuilder {
|
||||
b.times = times
|
||||
return b
|
||||
}
|
||||
|
||||
// ThenResult adds call result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenResult(body bin.Encoder) *Mock {
|
||||
return b.result(body, nil)
|
||||
}
|
||||
|
||||
// ThenTrue adds call tg.BoolTrue result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenTrue() *Mock {
|
||||
return b.result(&tg.BoolTrue{}, nil)
|
||||
}
|
||||
|
||||
// ThenFalse adds call tg.BoolFalse result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenFalse() *Mock {
|
||||
return b.result(&tg.BoolFalse{}, nil)
|
||||
}
|
||||
|
||||
// ThenErr adds call result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenErr(err error) *Mock {
|
||||
return b.result(nil, err)
|
||||
}
|
||||
|
||||
// ThenRPCErr adds call result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenRPCErr(err *tgerr.Error) *Mock {
|
||||
return b.ThenErr(err)
|
||||
}
|
||||
|
||||
// ThenMigrate adds call result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenMigrate(typ string, arg int) *Mock {
|
||||
t := strings.ToUpper(typ) + "_MIGRATE"
|
||||
return b.ThenRPCErr(&tgerr.Error{
|
||||
Code: 303,
|
||||
Message: t + "_" + strconv.Itoa(arg),
|
||||
Type: t,
|
||||
Argument: arg,
|
||||
})
|
||||
}
|
||||
|
||||
// ThenFlood adds call result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenFlood(arg int) *Mock {
|
||||
return b.ThenRPCErr(&tgerr.Error{
|
||||
Code: 420,
|
||||
Message: "FLOOD_WAIT_" + strconv.Itoa(arg),
|
||||
Type: "FLOOD_WAIT",
|
||||
Argument: arg,
|
||||
})
|
||||
}
|
||||
|
||||
// ThenUnregistered adds call result to the end of call stack.
|
||||
func (b *RequestBuilder) ThenUnregistered() *Mock {
|
||||
return b.ThenRPCErr(&tgerr.Error{
|
||||
Code: 401,
|
||||
Message: "AUTH_KEY_UNREGISTERED",
|
||||
Type: "AUTH_KEY_UNREGISTERED",
|
||||
})
|
||||
}
|
||||
|
||||
func (b *RequestBuilder) result(r bin.Encoder, err error) *Mock {
|
||||
for i := 0; i < b.times; i++ {
|
||||
b.mock.add(HandlerFunc(func(id int64, body bin.Encoder) (bin.Encoder, error) {
|
||||
if b.expectFn != nil {
|
||||
b.expectFn(body)
|
||||
}
|
||||
return r, err
|
||||
}))
|
||||
}
|
||||
|
||||
return b.mock
|
||||
}
|
||||
Reference in New Issue
Block a user