84 lines
2.5 KiB
Go
84 lines
2.5 KiB
Go
|
package common
|
||
|
|
||
|
import (
|
||
|
"crypto/ecdsa"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
|
||
|
"github.com/status-im/status-go/appdatabase"
|
||
|
"github.com/status-im/status-go/eth-node/crypto"
|
||
|
"github.com/status-im/status-go/protocol/sqlite"
|
||
|
"github.com/status-im/status-go/t/helpers"
|
||
|
)
|
||
|
|
||
|
func TestSaveRawMessage(t *testing.T) {
|
||
|
db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
|
||
|
require.NoError(t, err)
|
||
|
require.NoError(t, sqlite.Migrate(db))
|
||
|
p := NewRawMessagesPersistence(db)
|
||
|
|
||
|
pk, err := crypto.GenerateKey()
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
err = p.SaveRawMessage(&RawMessage{
|
||
|
ID: "1",
|
||
|
ResendType: ResendTypeRawMessage,
|
||
|
LocalChatID: "",
|
||
|
CommunityID: []byte("c1"),
|
||
|
CommunityKeyExMsgType: KeyExMsgRekey,
|
||
|
Sender: pk,
|
||
|
ResendMethod: ResendMethodSendPrivate,
|
||
|
Recipients: []*ecdsa.PublicKey{pk.Public().(*ecdsa.PublicKey)},
|
||
|
})
|
||
|
require.NoError(t, err)
|
||
|
m, err := p.RawMessageByID("1")
|
||
|
require.NoError(t, err)
|
||
|
require.Equal(t, "1", m.ID)
|
||
|
require.Equal(t, ResendTypeRawMessage, m.ResendType)
|
||
|
require.Equal(t, KeyExMsgRekey, m.CommunityKeyExMsgType)
|
||
|
require.Equal(t, "c1", string(m.CommunityID))
|
||
|
require.Equal(t, pk, m.Sender)
|
||
|
require.Equal(t, ResendMethodSendPrivate, m.ResendMethod)
|
||
|
require.Equal(t, 1, len(m.Recipients))
|
||
|
}
|
||
|
|
||
|
func TestUpdateRawMessageSent(t *testing.T) {
|
||
|
db, err := helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
|
||
|
require.NoError(t, err)
|
||
|
require.NoError(t, sqlite.Migrate(db))
|
||
|
p := NewRawMessagesPersistence(db)
|
||
|
|
||
|
pk, err := crypto.GenerateKey()
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
rawMessageID := "1"
|
||
|
err = p.SaveRawMessage(&RawMessage{
|
||
|
ID: rawMessageID,
|
||
|
ResendType: ResendTypeRawMessage,
|
||
|
LocalChatID: "",
|
||
|
CommunityID: []byte("c1"),
|
||
|
CommunityKeyExMsgType: KeyExMsgRekey,
|
||
|
Sender: pk,
|
||
|
ResendMethod: ResendMethodSendPrivate,
|
||
|
Recipients: []*ecdsa.PublicKey{pk.Public().(*ecdsa.PublicKey)},
|
||
|
Sent: true,
|
||
|
LastSent: uint64(time.Now().UnixNano() / int64(time.Millisecond)),
|
||
|
})
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
rawMessage, err := p.RawMessageByID(rawMessageID)
|
||
|
require.NoError(t, err)
|
||
|
require.True(t, rawMessage.Sent)
|
||
|
require.Greater(t, rawMessage.LastSent, uint64(0))
|
||
|
|
||
|
err = p.UpdateRawMessageSent(rawMessageID, false, 0)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
m, err := p.RawMessageByID(rawMessageID)
|
||
|
require.NoError(t, err)
|
||
|
require.False(t, m.Sent)
|
||
|
require.Equal(t, m.LastSent, uint64(0))
|
||
|
}
|