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,114 @@
|
||||
package tgmock
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
|
||||
)
|
||||
|
||||
// TestingT is simplified *testing.T interface.
|
||||
type TestingT interface {
|
||||
require.TestingT
|
||||
assert.TestingT
|
||||
Helper()
|
||||
Cleanup(cb func())
|
||||
}
|
||||
|
||||
// Mock is a mock for tg.Invoker with testify/require support.
|
||||
type Mock struct {
|
||||
calls []Handler
|
||||
assert assertions
|
||||
}
|
||||
|
||||
// Option configures Mock.
|
||||
type Option interface {
|
||||
apply(t TestingT, m *Mock)
|
||||
}
|
||||
|
||||
type assertions interface {
|
||||
Truef(value bool, msg string, args ...interface{})
|
||||
Equal(expected interface{}, actual interface{}, msgAndArgs ...interface{})
|
||||
}
|
||||
|
||||
type assertAssertions struct {
|
||||
assert *assert.Assertions
|
||||
}
|
||||
|
||||
func (a assertAssertions) Equal(expected, actual interface{}, msgAndArgs ...interface{}) {
|
||||
a.assert.Equal(expected, actual, msgAndArgs...)
|
||||
}
|
||||
|
||||
func (a assertAssertions) Truef(value bool, msg string, args ...interface{}) {
|
||||
a.assert.Truef(value, msg, args...)
|
||||
}
|
||||
|
||||
type optionFunc func(t TestingT, m *Mock)
|
||||
|
||||
func (o optionFunc) apply(t TestingT, m *Mock) { o(t, m) }
|
||||
|
||||
// WithRequire configures mock to use "require" assertions.
|
||||
func WithRequire() Option {
|
||||
return optionFunc(func(t TestingT, m *Mock) {
|
||||
m.assert = require.New(t)
|
||||
})
|
||||
}
|
||||
|
||||
// NewRequire creates new Mock with "require" assertions.
|
||||
func NewRequire(t TestingT) *Mock {
|
||||
return New(t, WithRequire())
|
||||
}
|
||||
|
||||
// New creates new Mock.
|
||||
func New(t TestingT, options ...Option) *Mock {
|
||||
m := &Mock{
|
||||
assert: &assertAssertions{
|
||||
assert: assert.New(t),
|
||||
},
|
||||
}
|
||||
for _, o := range options {
|
||||
o.apply(t, m)
|
||||
}
|
||||
|
||||
t.Cleanup(func() {
|
||||
m.assert.Truef(
|
||||
m.AllWereMet(),
|
||||
"not all expected calls happen (expected yet %d)",
|
||||
len(m.calls),
|
||||
)
|
||||
})
|
||||
return m
|
||||
}
|
||||
|
||||
// AllWereMet returns true if all expected calls happened.
|
||||
func (i *Mock) AllWereMet() bool {
|
||||
return len(i.calls) == 0
|
||||
}
|
||||
|
||||
func (i *Mock) add(h Handler) *Mock {
|
||||
i.calls = append(i.calls, h)
|
||||
return i
|
||||
}
|
||||
|
||||
func (i *Mock) pop() (Handler, bool) {
|
||||
if len(i.calls) < 1 {
|
||||
return nil, false
|
||||
}
|
||||
|
||||
h := i.calls[0]
|
||||
// Delete from SliceTricks.
|
||||
copy(i.calls, i.calls[1:])
|
||||
i.calls[len(i.calls)-1] = nil
|
||||
i.calls = i.calls[:len(i.calls)-1]
|
||||
return h, true
|
||||
}
|
||||
|
||||
// Handler returns HandlerFunc of Mock.
|
||||
func (i *Mock) Handler() HandlerFunc {
|
||||
return func(id int64, body bin.Encoder) (bin.Encoder, error) {
|
||||
h, ok := i.pop()
|
||||
i.assert.Truef(ok, "unexpected call")
|
||||
|
||||
return h.Handle(id, body)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user