package mailservers import ( "context" "testing" "github.com/stretchr/testify/require" "github.com/status-im/status-go/appdatabase" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/protocol/sqlite" "github.com/status-im/status-go/protocol/transport" "github.com/status-im/status-go/t/helpers" "github.com/status-im/status-go/wakuv2" ) func setupTestDB(t *testing.T) (*Database, func()) { db, cleanup, err := helpers.SetupTestSQLDB(appdatabase.DbInitializer{}, "maliservers-tests-") require.NoError(t, err) err = sqlite.Migrate(db) // migrate default require.NoError(t, err) return NewDB(db), func() { require.NoError(t, cleanup()) } } func TestAddGetDeleteMailserver(t *testing.T) { db, close := setupTestDB(t) defer close() api := &API{db: db} testMailserver := Mailserver{ ID: "mailserver001", Name: "My Mailserver", Addr: MustDecodeMultiaddress("/dns4/node-01.do-ams3.waku.test.status.im/tcp/30303/p2p/16Uiu2HAkykgaECHswi3YKJ5dMLbq2kPVCo89fcyTd38UcQD6ej5W"), Custom: true, Fleet: "prod", } testMailserverWithPassword := testMailserver testMailserverWithPassword.ID = "mailserver002" testMailserverWithPassword.Password = "test-pass" err := api.AddMailserver(context.Background(), testMailserver) require.NoError(t, err) err = api.AddMailserver(context.Background(), testMailserverWithPassword) require.NoError(t, err) mailservers, err := api.GetMailservers(context.Background()) require.NoError(t, err) require.EqualValues(t, []Mailserver{testMailserver, testMailserverWithPassword}, mailservers) err = api.DeleteMailserver(context.Background(), testMailserver.ID) require.NoError(t, err) // Verify they was deleted. mailservers, err = api.GetMailservers(context.Background()) require.NoError(t, err) require.EqualValues(t, []Mailserver{testMailserverWithPassword}, mailservers) // Delete non-existing mailserver. err = api.DeleteMailserver(context.Background(), "other-id") require.NoError(t, err) } func TestTopic(t *testing.T) { db, close := setupTestDB(t) defer close() topicA := "0x61000000" topicD := "0x64000000" topic1 := MailserverTopic{PubsubTopic: wakuv2.DefaultShardPubsubTopic(), ContentTopic: topicA, LastRequest: 1} topic2 := MailserverTopic{PubsubTopic: wakuv2.DefaultShardPubsubTopic(), ContentTopic: "0x6200000", LastRequest: 2} topic3 := MailserverTopic{PubsubTopic: wakuv2.DefaultShardPubsubTopic(), ContentTopic: "0x6300000", LastRequest: 3} require.NoError(t, db.AddTopic(topic1)) require.NoError(t, db.AddTopic(topic2)) require.NoError(t, db.AddTopic(topic3)) topics, err := db.Topics() require.NoError(t, err) require.Len(t, topics, 3) filters := []*transport.Filter{ // Existing topic, is not updated { PubsubTopic: wakuv2.DefaultShardPubsubTopic(), ContentTopic: types.BytesToTopic([]byte{0x61}), }, // Non existing topic is not inserted { Discovery: true, Negotiated: true, PubsubTopic: wakuv2.DefaultShardPubsubTopic(), ContentTopic: types.BytesToTopic([]byte{0x64}), }, } require.NoError(t, db.SetTopics(filters)) topics, err = db.Topics() require.NoError(t, err) require.Len(t, topics, 2) require.Equal(t, topics[0].ContentTopic, topicA) require.Equal(t, topics[0].LastRequest, 1) require.Equal(t, topics[0].ContentTopic, topicA) require.Equal(t, topics[0].LastRequest, 1) require.Equal(t, topics[1].ContentTopic, topicD) require.NotEmpty(t, topics[1].LastRequest) require.True(t, topics[1].Negotiated) require.True(t, topics[1].Discovery) } func TestAddGetDeleteMailserverRequestGap(t *testing.T) { db, close := setupTestDB(t) defer close() chatID1 := "chat-id-1" chatID2 := "chat-id-2" api := &API{db: db} gap1 := MailserverRequestGap{ID: "1", ChatID: chatID1, From: 1, To: 2} gap2 := MailserverRequestGap{ID: "2", ChatID: chatID2, From: 1, To: 2} gap3 := MailserverRequestGap{ID: "3", ChatID: chatID2, From: 1, To: 2} gaps := []MailserverRequestGap{ gap1, gap2, gap3, } err := api.AddMailserverRequestGaps(context.Background(), gaps) require.NoError(t, err) actualGaps, err := api.GetMailserverRequestGaps(context.Background(), chatID1) require.NoError(t, err) require.EqualValues(t, []MailserverRequestGap{gap1}, actualGaps) actualGaps, err = api.GetMailserverRequestGaps(context.Background(), chatID2) require.NoError(t, err) require.EqualValues(t, []MailserverRequestGap{gap2, gap3}, actualGaps) err = api.DeleteMailserverRequestGaps(context.Background(), []string{gap1.ID, gap2.ID}) require.NoError(t, err) // Verify it was deleted. actualGaps, err = api.GetMailserverRequestGaps(context.Background(), chatID1) require.NoError(t, err) require.Len(t, actualGaps, 0) actualGaps, err = api.GetMailserverRequestGaps(context.Background(), chatID2) require.NoError(t, err) require.Len(t, actualGaps, 1) err = api.DeleteMailserverRequestGapsByChatID(context.Background(), chatID2) require.NoError(t, err) // Verify it was deleted. actualGaps, err = api.GetMailserverRequestGaps(context.Background(), chatID2) require.NoError(t, err) require.Len(t, actualGaps, 0) } func TestAddGetDeleteMailserverTopics(t *testing.T) { db, close := setupTestDB(t) defer close() api := &API{db: db} testTopic := MailserverTopic{ PubsubTopic: wakuv2.DefaultShardPubsubTopic(), ContentTopic: "topic-001", ChatIDs: []string{"chatID01", "chatID02"}, LastRequest: 10, } err := api.AddMailserverTopic(context.Background(), testTopic) require.NoError(t, err) // Verify topics were added. topics, err := api.GetMailserverTopics(context.Background()) require.NoError(t, err) require.EqualValues(t, []MailserverTopic{testTopic}, topics) err = api.DeleteMailserverTopic(context.Background(), wakuv2.DefaultShardPubsubTopic(), testTopic.ContentTopic) require.NoError(t, err) topics, err = api.GetMailserverTopics(context.Background()) require.NoError(t, err) require.EqualValues(t, ([]MailserverTopic)(nil), topics) // Delete non-existing topic. err = api.DeleteMailserverTopic(context.Background(), wakuv2.DefaultShardPubsubTopic(), "non-existing-topic") require.NoError(t, err) } func TestAddGetDeleteChatRequestRanges(t *testing.T) { db, close := setupTestDB(t) defer close() api := &API{db: db} chatRequestRange1 := ChatRequestRange{ ChatID: "chat-id-001", LowestRequestFrom: 123, HighestRequestTo: 456, } chatRequestRange2 := chatRequestRange1 chatRequestRange2.ChatID = "chat-id-002" err := api.AddChatRequestRange(context.Background(), chatRequestRange1) require.NoError(t, err) err = api.AddChatRequestRange(context.Background(), chatRequestRange2) require.NoError(t, err) // Verify topics were added. ranges, err := api.GetChatRequestRanges(context.Background()) require.NoError(t, err) require.EqualValues(t, []ChatRequestRange{chatRequestRange1, chatRequestRange2}, ranges) err = api.DeleteChatRequestRange(context.Background(), chatRequestRange1.ChatID) require.NoError(t, err) ranges, err = api.GetChatRequestRanges(context.Background()) require.NoError(t, err) require.EqualValues(t, []ChatRequestRange{chatRequestRange2}, ranges) // Delete non-existing topic. err = api.DeleteChatRequestRange(context.Background(), "non-existing-chat-id") require.NoError(t, err) }