Add SecondLevelDomain.GetRole()

x/registry/keeper/query_role.go

@@ -32,7 +32,7 @@ func (k Keeper) Role(goCtx context.Context, req *types.QueryRoleRequest) (*types
 		if !found {
 			return nil, errorsmod.Wrapf(sdkerrors.ErrNotFound, "domain not found")
 		}
-		role = sld.AccessControl[req.Address]
+		role = sld.GetRole(req.Address)
 	default:
 		return nil, errorsmod.Wrapf(sdkerrors.ErrInvalidRequest, "invalid request: domain name")
 	}

x/registry/types/secend_level_domain.go

@@ -116,3 +116,8 @@ func (secondLevelDomain SecondLevelDomain) IsRecordEditable(sender string) (isEd
 	isEditable = secondLevelDomain.AccessControl[sender] == DomainRole_EDITOR || secondLevelDomain.AccessControl[sender] == DomainRole_OWNER
 	return isEditable, err
 }
+
+func (secondLevelDomain *SecondLevelDomain) GetRole(address string) (role DomainRole) {
+	role = secondLevelDomain.AccessControl[address]
+	return role
+}

x/registry/types/second_level_domain_test.go

@@ -1,11 +1,14 @@
 package types
 
 import (
+	fmt "fmt"
 	"testing"
 
 	errorsmod "cosmossdk.io/errors"
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/stretchr/testify/require"
+
+	"github.com/mycel-domain/mycel/testutil"
 )
 
 type DomainTest struct {
@@ -197,3 +200,59 @@ func TestDomainUpdateDnsRecord(t *testing.T) {
 		}
 	}
 }
+
+func TestGetRoleSLD(t *testing.T) {
+	testCases := []struct {
+		domain SecondLevelDomain
+		req    string
+		exp    DomainRole
+	}{
+		// Valid domains
+		{
+			domain: SecondLevelDomain{
+				Name:          "myc",
+				AccessControl: map[string]DomainRole{testutil.Alice: DomainRole_NO_ROLE},
+			},
+			req: testutil.Alice,
+			exp: DomainRole_NO_ROLE,
+		},
+		{
+			domain: SecondLevelDomain{
+				Name:          "myc",
+				AccessControl: map[string]DomainRole{testutil.Alice: DomainRole_OWNER},
+			},
+			req: testutil.Alice,
+			exp: DomainRole_OWNER,
+		},
+		{
+			domain: SecondLevelDomain{
+				Name:          "myc",
+				AccessControl: map[string]DomainRole{testutil.Alice: DomainRole_EDITOR},
+			},
+			req: testutil.Alice,
+			exp: DomainRole_EDITOR,
+		},
+		{
+			domain: SecondLevelDomain{
+				Name:          "myc",
+				AccessControl: map[string]DomainRole{testutil.Alice: DomainRole_OWNER},
+			},
+			req: testutil.Bob,
+			exp: DomainRole_NO_ROLE,
+		},
+		{
+			domain: SecondLevelDomain{
+				Name: "myc",
+			},
+			req: testutil.Alice,
+			exp: DomainRole_NO_ROLE,
+		},
+	}
+
+	for i, tc := range testCases {
+		t.Run(fmt.Sprintf("Case %d", i), func(t *testing.T) {
+			r := tc.domain.GetRole(tc.req)
+			require.Equal(t, tc.exp, r)
+		})
+	}
+}

x/registry/types/top_level_domain_test.go

@@ -76,7 +76,7 @@ func TestExtendExpirationDate(t *testing.T) {
 	}
 }
 
-func TestGetRole(t *testing.T) {
+func TestGetRoleTLD(t *testing.T) {
 	testCases := []struct {
 		domain TopLevelDomain
 		req    string