53 lines
1.5 KiB
Go
53 lines
1.5 KiB
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)
|
|
}
|