Adam Van Ymeren
7a04f298d2
- 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
150 lines
2.6 KiB
Go
150 lines
2.6 KiB
Go
package peers
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
func TestMemoryStorage(t *testing.T) {
|
|
ctx := context.Background()
|
|
a := require.New(t)
|
|
k := Key{
|
|
Prefix: usersPrefix,
|
|
ID: 1,
|
|
}
|
|
v := Value{
|
|
AccessHash: 10,
|
|
}
|
|
phone := "phone"
|
|
|
|
var m InmemoryStorage
|
|
_, found, err := m.Find(ctx, k)
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.Save(ctx, k, v))
|
|
|
|
v2, found, err := m.Find(ctx, k)
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(v, v2)
|
|
|
|
_, _, found, err = m.FindPhone(ctx, phone)
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SavePhone(ctx, phone, k))
|
|
|
|
k2, v2, found, err := m.FindPhone(ctx, phone)
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(k, k2)
|
|
a.Equal(v, v2)
|
|
|
|
hash, err := m.GetContactsHash(ctx)
|
|
a.NoError(err)
|
|
a.Zero(hash)
|
|
|
|
a.NoError(m.SaveContactsHash(ctx, 1))
|
|
|
|
hash, err = m.GetContactsHash(ctx)
|
|
a.NoError(err)
|
|
a.Equal(int64(1), hash)
|
|
}
|
|
|
|
func TestInmemoryCache(t *testing.T) {
|
|
ctx := context.Background()
|
|
a := require.New(t)
|
|
|
|
var m InmemoryCache
|
|
|
|
{
|
|
value := &tg.User{ID: 10}
|
|
|
|
_, found, err := m.FindUser(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SaveUsers(ctx, value))
|
|
|
|
r, found, err := m.FindUser(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(value, r)
|
|
}
|
|
{
|
|
value := &tg.UserFull{ID: 10}
|
|
|
|
_, found, err := m.FindUserFull(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SaveUserFulls(ctx, value))
|
|
|
|
r, found, err := m.FindUserFull(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(value, r)
|
|
}
|
|
{
|
|
value := &tg.Chat{ID: 10}
|
|
|
|
_, found, err := m.FindChat(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SaveChats(ctx, value))
|
|
|
|
r, found, err := m.FindChat(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(value, r)
|
|
}
|
|
{
|
|
value := &tg.ChatFull{ID: 10}
|
|
|
|
_, found, err := m.FindChatFull(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SaveChatFulls(ctx, value))
|
|
|
|
r, found, err := m.FindChatFull(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(value, r)
|
|
}
|
|
{
|
|
value := &tg.Channel{ID: 10}
|
|
|
|
_, found, err := m.FindChannel(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SaveChannels(ctx, value))
|
|
|
|
r, found, err := m.FindChannel(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(value, r)
|
|
}
|
|
{
|
|
value := &tg.ChannelFull{ID: 10}
|
|
|
|
_, found, err := m.FindChannelFull(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.False(found)
|
|
|
|
a.NoError(m.SaveChannelFulls(ctx, value))
|
|
|
|
r, found, err := m.FindChannelFull(ctx, value.GetID())
|
|
a.NoError(err)
|
|
a.True(found)
|
|
a.Equal(value, r)
|
|
}
|
|
}
|