add more test

e2e/tests/permission_test.go

@@ -7,6 +7,7 @@ import (
 	"github.com/bnb-chain/greenfield/e2e/core"
 	storageutil "github.com/bnb-chain/greenfield/testutil/storage"
 	types2 "github.com/bnb-chain/greenfield/types"
+	"github.com/bnb-chain/greenfield/types/resource"
 	"github.com/bnb-chain/greenfield/x/permission/types"
 	storagetypes "github.com/bnb-chain/greenfield/x/storage/types"
 )
@@ -81,7 +82,18 @@ func (s *StorageTestSuite) TestDeleteBucketPermission() {
 	s.T().Logf("resp: %s, rep %s", verifyPermReq.String(), verifyPermResp.String())
 	s.Require().NoError(err)
 	s.Require().Equal(verifyPermResp.Effect, types.EFFECT_ALLOW)
+
+	// Query bucket proxy
+	grn := types2.NewBucketGRN(bucketName)
+	queryPolicyForAccountReq := storagetypes.QueryPolicyForAccountRequest{Resource: grn.String(),
+		PrincipalAddress: user[1].GetAddr().String()}
+	queryPolicyForAccountResp, err := s.Client.QueryPolicyForAccount(ctx, &queryPolicyForAccountReq)
+	s.Require().NoError(err)
+	s.Require().Equal(queryPolicyForAccountResp.Policy.ResourceType, resource.RESOURCE_TYPE_BUCKET)
+	s.Require().Equal(queryPolicyForAccountResp.Policy.ResourceId, queryHeadBucketResponse.BucketInfo.Id)
+
 	// DeleteBucket
 	msgDeleteBucket := storagetypes.NewMsgDeleteBucket(user[1].GetAddr(), bucketName)
 	s.SendTxBlock(msgDeleteBucket, user[1])
+
 }

x/storage/keeper/permission.go

@@ -141,8 +141,8 @@ func (k Keeper) GetPolicy(ctx sdk.Context, grn *types2.GRN, principal *permtypes
 		return nil, permtypes.ErrInvalidPrincipal
 	}
 
-	if found {
-		return nil, types.ErrNoSuchPolicy
+	if !found {
+		return nil, types.ErrNoSuchPolicy.Wrapf("GRN: %s, principal:%s", grn.String(), principal.String())
 	}
 	return policy, nil
 }

x/storage/keeper/query.go

@@ -260,7 +260,7 @@ func (k Keeper) QueryPolicyForAccount(goCtx context.Context, req *types.QueryPol
 
 	policy, err := k.GetPolicy(ctx, &grn, permtypes.NewPrincipalWithAccount(principalAcc))
 	if err != nil {
-		return nil, types.ErrNoSuchPolicy
+		return nil, err
 	}
 
 	return &types.QueryPolicyForAccountResponse{Policy: policy}, nil
@@ -287,7 +287,7 @@ func (k Keeper) QueryPolicyForGroup(goCtx context.Context, req *types.QueryPolic
 
 	policy, err := k.GetPolicy(ctx, &grn, permtypes.NewPrincipalWithGroup(id))
 	if err != nil {
-		return nil, types.ErrNoSuchPolicy
+		return nil, err
 	}
 	return &types.QueryPolicyForGroupResponse{Policy: policy}, nil
 }