status-im/status-go

package community

import (
    "testing"

    "github.com/status-im/status-go/services/wallet/thirdparty"
    "github.com/status-im/status-go/t/helpers"
    "github.com/status-im/status-go/walletdatabase"

    "github.com/stretchr/testify/require"
)

func setupCommunityDataDBTest(t *testing.T) (*DataDB, func()) {
    db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
    require.NoError(t, err)
    return NewDataDB(db), func() {
        require.NoError(t, db.Close())
    }
}

func TestUpdateCommunityInfo(t *testing.T) {
    db, cleanup := setupCommunityDataDBTest(t)
    defer cleanup()

    communityData := thirdparty.GenerateTestCommunityInfo(10)
    extraCommunityID := "extra-community-id"

    for communityID, communityInfo := range communityData {
        communityInfo := communityInfo // Prevent lint warning G601: Implicit memory aliasing in for loop.
        err := db.SetCommunityInfo(communityID, &communityInfo)
        require.NoError(t, err)
    }
    err := db.SetCommunityInfo(extraCommunityID, nil)
    require.NoError(t, err)

    for communityID, communityInfo := range communityData {
        info, state, err := db.GetCommunityInfo(communityID)
        require.NoError(t, err)
        require.Equal(t, communityInfo, *info)
        require.True(t, state.LastUpdateSuccesful)
    }
    info, state, err := db.GetCommunityInfo(extraCommunityID)
    require.NoError(t, err)
    require.Empty(t, info)
    require.False(t, state.LastUpdateSuccesful)

    randomCommunityID := "random-community-id"
    info, state, err = db.GetCommunityInfo(randomCommunityID)
    require.NoError(t, err)
    require.Empty(t, info)
    require.Empty(t, state)
}