status-go/services/wallet/community/database_test.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)
}