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,121 @@
|
||||
package proto
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"github.com/gotd/neo"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/testutil"
|
||||
)
|
||||
|
||||
func TestMessageID(t *testing.T) {
|
||||
now := time.Date(2018, 10, 10, 23, 42, 6, 13600, time.UTC)
|
||||
id := MessageID(newMessageID(now.UnixNano(), 0))
|
||||
if id.Type() != MessageFromClient {
|
||||
t.Fatal("invalid type")
|
||||
}
|
||||
if id != 0x5bbe8e4e00003520 {
|
||||
t.Error("mismatch")
|
||||
}
|
||||
delta := id.Time().Sub(now)
|
||||
if delta < 0 {
|
||||
delta *= -1
|
||||
}
|
||||
if delta > time.Second {
|
||||
t.Fatal("unexpected time drift")
|
||||
}
|
||||
t.Run("NewMessageID", func(t *testing.T) {
|
||||
if NewMessageID(now, MessageFromServer).Type() != MessageFromServer {
|
||||
t.Error("Mismatch")
|
||||
}
|
||||
if NewMessageID(now, 100).Type() != MessageFromClient {
|
||||
t.Error("Mismatch")
|
||||
}
|
||||
})
|
||||
t.Run("String", func(t *testing.T) {
|
||||
require.Equal(t, "5bbe8e4e00003520 (FromClient, 2018-10-10T23:42:06Z)", id.String())
|
||||
})
|
||||
}
|
||||
|
||||
func BenchmarkNewMessageID(b *testing.B) {
|
||||
// Note that most overhead will be from time.Now() calls.
|
||||
// Just ensuring that NewMessageID itself is reasonably fast.
|
||||
now := testutil.Date()
|
||||
for i := 0; i < b.N; i++ {
|
||||
if NewMessageID(now, MessageFromServer).Type() != MessageFromServer {
|
||||
b.Fatal("Mismatch")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestMessageIDGen(t *testing.T) {
|
||||
date := testutil.Date()
|
||||
clock := neo.NewTime(date)
|
||||
|
||||
gen := NewMessageIDGen(clock.Now)
|
||||
met := make(map[int64]bool)
|
||||
|
||||
for i := 0; i < 1000; i++ {
|
||||
if i%10 == 0 {
|
||||
clock.Travel(time.Millisecond * 100)
|
||||
}
|
||||
|
||||
id := gen.New(MessageFromClient)
|
||||
if met[id] {
|
||||
t.Fatal("met")
|
||||
}
|
||||
|
||||
met[id] = true
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkMsgIDGen_New(b *testing.B) {
|
||||
b.ReportAllocs()
|
||||
|
||||
date := testutil.Date()
|
||||
var dateCalls int
|
||||
now := func() time.Time {
|
||||
if dateCalls%100 == 0 {
|
||||
date = date.Add(time.Millisecond)
|
||||
}
|
||||
return date
|
||||
}
|
||||
|
||||
gen := NewMessageIDGen(now)
|
||||
|
||||
for i := 0; i < b.N; i++ {
|
||||
_ = gen.New(MessageFromServer)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewMessageIDBuf(t *testing.T) {
|
||||
t.Run("Zero", func(t *testing.T) {
|
||||
buf := NewMessageIDBuf(10)
|
||||
|
||||
assert.False(t, buf.Consume(0))
|
||||
})
|
||||
t.Run("Ok", func(t *testing.T) {
|
||||
buf := NewMessageIDBuf(10)
|
||||
|
||||
assert.True(t, buf.Consume(1))
|
||||
assert.False(t, buf.Consume(1))
|
||||
|
||||
t.Run("Sequence", func(t *testing.T) {
|
||||
for i := 2; i <= 20; i++ {
|
||||
assert.True(t, buf.Consume(int64(i)))
|
||||
}
|
||||
assert.False(t, buf.Consume(-1))
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
func BenchmarkMessageIDBuf(b *testing.B) {
|
||||
buf := NewMessageIDBuf(100)
|
||||
for i := 0; i < b.N; i++ {
|
||||
buf.Consume(int64(i))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user