status-go/services/browsers/api_test.go

package browsers

import (
	"context"
	"io/ioutil"
	"os"
	"sort"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/status-im/status-go/appdatabase"
)

func setupTestDB(t *testing.T) (*Database, func()) {
	tmpfile, err := ioutil.TempFile("", "browsers-tests-")
	require.NoError(t, err)
	db, err := appdatabase.InitializeDB(tmpfile.Name(), "browsers-tests")
	require.NoError(t, err)
	return NewDB(db), func() {
		require.NoError(t, db.Close())
		require.NoError(t, os.Remove(tmpfile.Name()))
	}
}

func setupTestAPI(t *testing.T) (*API, func()) {
	db, cancel := setupTestDB(t)
	return &API{db: db}, cancel
}

func TestBrowsersOrderedNewestFirst(t *testing.T) {
	api, cancel := setupTestAPI(t)
	defer cancel()

	browsers := []*Browser{
		{
			ID:        "1",
			Name:      "first",
			Dapp:      true,
			Timestamp: 10,
		},
		{
			ID:        "2",
			Name:      "second",
			Dapp:      true,
			Timestamp: 50,
		},
		{
			ID:           "3",
			Name:         "third",
			Dapp:         true,
			Timestamp:    100,
			HistoryIndex: 0,
			History:      []string{"zero"},
		},
	}
	for i := 0; i < len(browsers); i++ {
		require.NoError(t, api.AddBrowser(context.TODO(), *browsers[i]))
	}

	sort.Slice(browsers, func(i, j int) bool {
		return browsers[i].Timestamp > browsers[j].Timestamp
	})

	rst, err := api.GetBrowsers(context.TODO())
	require.NoError(t, err)
	require.Equal(t, browsers, rst)
}

func TestBrowsersHistoryIncluded(t *testing.T) {
	api, cancel := setupTestAPI(t)
	defer cancel()

	browser := &Browser{
		ID:           "1",
		Name:         "first",
		Dapp:         true,
		Timestamp:    10,
		HistoryIndex: 1,
		History:      []string{"one", "two"},
	}
	require.NoError(t, api.AddBrowser(context.TODO(), *browser))
	rst, err := api.GetBrowsers(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 1)
	require.Equal(t, browser, rst[0])
}

func TestBrowsersReplaceOnUpdate(t *testing.T) {
	api, cancel := setupTestAPI(t)
	defer cancel()

	browser := &Browser{
		ID:        "1",
		Name:      "first",
		Dapp:      true,
		Timestamp: 10,
		History:   []string{"one", "two"},
	}
	require.NoError(t, api.AddBrowser(context.TODO(), *browser))
	browser.Dapp = false
	browser.History = []string{"one", "three"}
	browser.Timestamp = 107
	require.NoError(t, api.AddBrowser(context.TODO(), *browser))
	rst, err := api.GetBrowsers(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 1)
	require.Equal(t, browser, rst[0])
}

func TestDeleteBrowser(t *testing.T) {
	api, cancel := setupTestAPI(t)
	defer cancel()

	browser := &Browser{
		ID:        "1",
		Name:      "first",
		Dapp:      true,
		Timestamp: 10,
		History:   []string{"one", "two"},
	}

	require.NoError(t, api.AddBrowser(context.TODO(), *browser))
	rst, err := api.GetBrowsers(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 1)

	require.NoError(t, api.DeleteBrowser(context.TODO(), browser.ID))
	rst, err = api.GetBrowsers(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 0)
}