status-im/status-go

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/common/shard"
    "github.com/status-im/status-go/protocol/sqlite"
    "github.com/status-im/status-go/protocol/transport"
    "github.com/status-im/status-go/t/helpers"
)

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: shard.DefaultShardPubsubTopic(), ContentTopic: topicA, LastRequest: 1}
    topic2 := MailserverTopic{PubsubTopic: shard.DefaultShardPubsubTopic(), ContentTopic: "0x6200000", LastRequest: 2}
    topic3 := MailserverTopic{PubsubTopic: shard.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:  shard.DefaultShardPubsubTopic(),
            ContentTopic: types.BytesToTopic([]byte{0x61}),
        },
        // Non existing topic is not inserted
        {
            Discovery:    true,
            Negotiated:   true,
            PubsubTopic:  shard.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:  shard.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(), shard.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(), shard.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)
}