package permissions import ( "context" "encoding/json" "fmt" "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("", "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) }