status-go/services/permissions/api_test.go

package permissions

import (
	"context"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
	"sort"
	"testing"

	"github.com/status-im/status-go/appdatabase"
	"github.com/stretchr/testify/require"
)

func setupTestDB(t *testing.T) (*Database, func()) {
	tmpfile, err := ioutil.TempFile("", "perm-tests-")
	require.NoError(t, err)
	db, err := appdatabase.InitializeDB(tmpfile.Name(), "perm-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 TestDappPermissionsStored(t *testing.T) {
	api, cancel := setupTestAPI(t)
	defer cancel()

	expected := []DappPermissions{
		{
			Name:        "first",
			Permissions: []string{"r", "w"},
		},
		{
			Name:        "second",
			Permissions: []string{"r", "x"},
		},
		{
			Name: "third",
		},
	}
	for _, perms := range expected {
		require.NoError(t, api.AddDappPermissions(context.TODO(), perms))
	}
	rst, err := api.GetDappPermissions(context.TODO())
	require.NoError(t, err)
	// sort in lexicographic order by name
	sort.Slice(rst, func(i, j int) bool {
		return rst[i].Name < rst[j].Name
	})
	require.Equal(t, expected, rst)

	data, err := json.Marshal(rst)
	require.NoError(t, err)
	fmt.Println(string(data))
}

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

	perms := DappPermissions{
		Name:        "first",
		Permissions: []string{"r", "w"},
	}
	require.NoError(t, api.AddDappPermissions(context.TODO(), perms))
	perms.Permissions = append(perms.Permissions, "x")
	require.NoError(t, api.AddDappPermissions(context.TODO(), perms))
	rst, err := api.GetDappPermissions(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 1)
	require.Equal(t, perms, rst[0])
}

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

	perms := DappPermissions{
		Name: "first",
	}
	require.NoError(t, api.AddDappPermissions(context.TODO(), perms))
	rst, err := api.GetDappPermissions(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 1)
	require.NoError(t, api.DeleteDappPermissions(context.TODO(), perms.Name))
	rst, err = api.GetDappPermissions(context.TODO())
	require.NoError(t, err)
	require.Len(t, rst, 0)
}