status-go/services/stickers/recent_test.go

package stickers

import (
	"context"
	"encoding/json"
	"math/big"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"go.uber.org/mock/gomock"

	"github.com/status-im/status-go/multiaccounts/accounts"
	mock_settings "github.com/status-im/status-go/multiaccounts/settings/mock"
	"github.com/status-im/status-go/services/wallet/bigint"
)

func SetupAPI(t *testing.T) (*API, *mock_settings.MockDatabaseSettingsManager) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	mockDB := mock_settings.NewMockDatabaseSettingsManager(ctrl)

	require.NotNil(t, mockDB)

	accountDatabase := &accounts.Database{
		DatabaseSettingsManager: mockDB,
	}

	api := NewAPI(context.Background(), accountDatabase, nil, nil, nil, "test-store-dir", nil, nil)
	require.NotNil(t, api)

	return api, mockDB
}

func Test_WHEN_EmptyRecentStickers_And_EmptyStickerPacks_THEN_EmptyStickers_Returned(t *testing.T) {

	api, mockDB := SetupAPI(t)

	mockDB.EXPECT().GetInstalledStickerPacks().Return(nil, nil)

	actual, err := api.recentStickers()

	assert.NoError(t, err)
	assert.Equal(t, []Sticker{}, actual)
}

func Test_WHEN_EmptyStickerPacks_THEN_EmptyStickers_Returned(t *testing.T) {
	expectedStickers := []Sticker{
		{
			PackID: &bigint.BigInt{Int: big.NewInt(1)},
			URL:    "sticker1",
			Hash:   "0x1",
		},
		{
			PackID: &bigint.BigInt{Int: big.NewInt(2)},
			URL:    "sticker2",
			Hash:   "0x2",
		},
	}

	expectedStickerPacks := map[string]StickerPack{}

	api, mockDB := SetupAPI(t)

	expectedStickersJSON, err := json.Marshal(expectedStickers)
	require.NoError(t, err)

	expectedStickerPacksJSON, err := json.Marshal(expectedStickerPacks)
	require.NoError(t, err)

	mockDB.EXPECT().GetInstalledStickerPacks().Return((*json.RawMessage)(&expectedStickerPacksJSON), nil)
	mockDB.EXPECT().GetRecentStickers().Return((*json.RawMessage)(&expectedStickersJSON), nil)

	actual, err := api.recentStickers()

	require.NoError(t, err)
	require.Equal(t, 0, len(actual))
	require.Equal(t, []Sticker{}, actual)
}

func Test_WHEN_Stickers_In_Existing_SingleStickerPack_THEN_Stickers_Returned(t *testing.T) {
	expectedStickers := []Sticker{
		{
			PackID: &bigint.BigInt{Int: big.NewInt(1)},
			URL:    "sticker1",
			Hash:   "0x1",
		},
		{
			PackID: &bigint.BigInt{Int: big.NewInt(2)},
			URL:    "sticker2",
			Hash:   "0x2",
		},
	}

	expectedStickerPacks := map[string]StickerPack{
		"1": {
			ID:        &bigint.BigInt{Int: big.NewInt(1)},
			Name:      "test",
			Author:    "test",
			Owner:     [20]byte{},
			Price:     &bigint.BigInt{Int: big.NewInt(10)},
			Preview:   "",
			Thumbnail: "",
			Stickers:  expectedStickers,
			Status:    1,
		},
		"2": {
			ID:        &bigint.BigInt{Int: big.NewInt(2)},
			Name:      "test",
			Author:    "test",
			Owner:     [20]byte{},
			Price:     &bigint.BigInt{Int: big.NewInt(10)},
			Preview:   "",
			Thumbnail: "",
			Stickers:  expectedStickers,
			Status:    1,
		},
	}

	api, mockDB := SetupAPI(t)

	expectedStickersJSON, err := json.Marshal(expectedStickers)
	require.NoError(t, err)

	expectedStickerPacksJSON, err := json.Marshal(expectedStickerPacks)
	require.NoError(t, err)

	mockDB.EXPECT().GetInstalledStickerPacks().Return((*json.RawMessage)(&expectedStickerPacksJSON), nil)
	mockDB.EXPECT().GetRecentStickers().Return((*json.RawMessage)(&expectedStickersJSON), nil)

	actual, err := api.recentStickers()

	require.NoError(t, err)
	require.Equal(t, 2, len(actual))
	require.Equal(t, expectedStickers, actual)
}

func Test_WHEN_Stickers_In_Existing_In_MultipleStickerPacks_THEN_Stickers_Returned(t *testing.T) {
	expectedStickers := []Sticker{
		{
			PackID: &bigint.BigInt{Int: big.NewInt(1)},
			URL:    "sticker1",
			Hash:   "0x1",
		},
		{
			PackID: &bigint.BigInt{Int: big.NewInt(2)},
			URL:    "sticker2",
			Hash:   "0x2",
		},
	}

	expectedStickerPacks := map[string]StickerPack{
		"1": {
			ID:        &bigint.BigInt{Int: big.NewInt(1)},
			Name:      "test",
			Author:    "test",
			Owner:     [20]byte{},
			Price:     &bigint.BigInt{Int: big.NewInt(10)},
			Preview:   "",
			Thumbnail: "",
			Stickers:  expectedStickers,
			Status:    1,
		},
		"2": {
			ID:        &bigint.BigInt{Int: big.NewInt(2)},
			Name:      "test",
			Author:    "test",
			Owner:     [20]byte{},
			Price:     &bigint.BigInt{Int: big.NewInt(10)},
			Preview:   "",
			Thumbnail: "",
			Stickers:  expectedStickers,
			Status:    1,
		},
		"3": {
			ID:        &bigint.BigInt{Int: big.NewInt(3)},
			Name:      "test",
			Author:    "test",
			Owner:     [20]byte{},
			Price:     &bigint.BigInt{Int: big.NewInt(10)},
			Preview:   "",
			Thumbnail: "",
			Stickers:  expectedStickers,
			Status:    1,
		},
	}

	api, mockDB := SetupAPI(t)

	expectedStickersJSON, err := json.Marshal(expectedStickers)
	require.NoError(t, err)

	expectedStickerPacksJSON, err := json.Marshal(expectedStickerPacks)
	require.NoError(t, err)

	mockDB.EXPECT().GetInstalledStickerPacks().Return((*json.RawMessage)(&expectedStickerPacksJSON), nil)
	mockDB.EXPECT().GetRecentStickers().Return((*json.RawMessage)(&expectedStickersJSON), nil)

	actual, err := api.recentStickers()

	require.NoError(t, err)
	require.Equal(t, 2, len(actual))
	require.Equal(t, expectedStickers, actual)
}