103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package permissions
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/status-im/status-go/appdatabase"
|
|
"github.com/status-im/status-go/sqlite"
|
|
)
|
|
|
|
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", sqlite.ReducedKDFIterationsNumber)
|
|
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[0].Name, rst[0].Name)
|
|
require.Equal(t, expected[0].Permissions, rst[0].Permissions)
|
|
|
|
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.Name, rst[0].Name)
|
|
require.Equal(t, perms.Permissions, rst[0].Permissions)
|
|
|
|
}
|
|
|
|
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)
|
|
}
|