|
|
@@ -14,7 +14,7 @@ import (
|
|
|
"go.uber.org/mock/gomock"
|
|
|
)
|
|
|
|
|
|
-var dbErr = errors.New("db error")
|
|
|
+var errDB = errors.New("db error")
|
|
|
|
|
|
// TC-0232: isSuperAdmin=true, deptId=0, FindAllCodesByProductCode返回err
|
|
|
func TestGetUserPerms_Mock_SuperAdmin_FindAllCodesFail(t *testing.T) {
|
|
|
@@ -22,14 +22,14 @@ func TestGetUserPerms_Mock_SuperAdmin_FindAllCodesFail(t *testing.T) {
|
|
|
defer ctrl.Finish()
|
|
|
|
|
|
mockPerm := mocks.NewMockSysPermModel(ctrl)
|
|
|
- mockPerm.EXPECT().FindAllCodesByProductCode(gomock.Any(), "pc").Return(nil, dbErr)
|
|
|
+ mockPerm.EXPECT().FindAllCodesByProductCode(gomock.Any(), "pc").Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
Perm: mockPerm,
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", true)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Empty(t, memberType)
|
|
|
}
|
|
|
@@ -40,14 +40,14 @@ func TestGetUserPerms_Mock_MemberQueryDBError(t *testing.T) {
|
|
|
defer ctrl.Finish()
|
|
|
|
|
|
mockPM := mocks.NewMockSysProductMemberModel(ctrl)
|
|
|
- mockPM.EXPECT().FindOneByProductCodeUserId(gomock.Any(), "pc", int64(1)).Return(nil, dbErr)
|
|
|
+ mockPM.EXPECT().FindOneByProductCodeUserId(gomock.Any(), "pc", int64(1)).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Empty(t, memberType)
|
|
|
}
|
|
|
@@ -62,7 +62,7 @@ func TestGetUserPerms_Mock_Developer_FindAllCodesFail(t *testing.T) {
|
|
|
Return(&productmember.SysProductMember{MemberType: consts.MemberTypeDeveloper}, nil)
|
|
|
|
|
|
mockPerm := mocks.NewMockSysPermModel(ctrl)
|
|
|
- mockPerm.EXPECT().FindAllCodesByProductCode(gomock.Any(), "pc").Return(nil, dbErr)
|
|
|
+ mockPerm.EXPECT().FindAllCodesByProductCode(gomock.Any(), "pc").Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -70,7 +70,7 @@ func TestGetUserPerms_Mock_Developer_FindAllCodesFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeDeveloper, memberType)
|
|
|
}
|
|
|
@@ -85,7 +85,7 @@ func TestGetUserPerms_Mock_Member_FindRoleIdsFail(t *testing.T) {
|
|
|
Return(&productmember.SysProductMember{MemberType: consts.MemberTypeMember}, nil)
|
|
|
|
|
|
mockUR := mocks.NewMockSysUserRoleModel(ctrl)
|
|
|
- mockUR.EXPECT().FindRoleIdsByUserId(gomock.Any(), int64(1)).Return(nil, dbErr)
|
|
|
+ mockUR.EXPECT().FindRoleIdsByUserId(gomock.Any(), int64(1)).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -93,7 +93,7 @@ func TestGetUserPerms_Mock_Member_FindRoleIdsFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|
|
|
@@ -111,7 +111,7 @@ func TestGetUserPerms_Mock_Member_FindRoleByIdsFail(t *testing.T) {
|
|
|
mockUR.EXPECT().FindRoleIdsByUserId(gomock.Any(), int64(1)).Return([]int64{10, 20}, nil)
|
|
|
|
|
|
mockRole := mocks.NewMockSysRoleModel(ctrl)
|
|
|
- mockRole.EXPECT().FindByIds(gomock.Any(), []int64{10, 20}).Return(nil, dbErr)
|
|
|
+ mockRole.EXPECT().FindByIds(gomock.Any(), []int64{10, 20}).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -120,7 +120,7 @@ func TestGetUserPerms_Mock_Member_FindRoleByIdsFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|
|
|
@@ -138,7 +138,7 @@ func TestGetUserPerms_Mock_Member_FindPermIdsByRoleIdsFail(t *testing.T) {
|
|
|
mockUR.EXPECT().FindRoleIdsByUserId(gomock.Any(), int64(1)).Return([]int64{}, nil)
|
|
|
|
|
|
mockRP := mocks.NewMockSysRolePermModel(ctrl)
|
|
|
- mockRP.EXPECT().FindPermIdsByRoleIds(gomock.Any(), gomock.Any()).Return(nil, dbErr)
|
|
|
+ mockRP.EXPECT().FindPermIdsByRoleIds(gomock.Any(), gomock.Any()).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -147,7 +147,7 @@ func TestGetUserPerms_Mock_Member_FindPermIdsByRoleIdsFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|
|
|
@@ -168,7 +168,7 @@ func TestGetUserPerms_Mock_Member_FindAllowPermIdsFail(t *testing.T) {
|
|
|
mockRP.EXPECT().FindPermIdsByRoleIds(gomock.Any(), gomock.Any()).Return([]int64{}, nil)
|
|
|
|
|
|
mockUP := mocks.NewMockSysUserPermModel(ctrl)
|
|
|
- mockUP.EXPECT().FindPermIdsByUserIdAndEffect(gomock.Any(), int64(1), consts.PermEffectAllow).Return(nil, dbErr)
|
|
|
+ mockUP.EXPECT().FindPermIdsByUserIdAndEffect(gomock.Any(), int64(1), consts.PermEffectAllow).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -178,7 +178,7 @@ func TestGetUserPerms_Mock_Member_FindAllowPermIdsFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|
|
|
@@ -200,7 +200,7 @@ func TestGetUserPerms_Mock_Member_FindDenyPermIdsFail(t *testing.T) {
|
|
|
|
|
|
mockUP := mocks.NewMockSysUserPermModel(ctrl)
|
|
|
mockUP.EXPECT().FindPermIdsByUserIdAndEffect(gomock.Any(), int64(1), consts.PermEffectAllow).Return([]int64{}, nil)
|
|
|
- mockUP.EXPECT().FindPermIdsByUserIdAndEffect(gomock.Any(), int64(1), consts.PermEffectDeny).Return(nil, dbErr)
|
|
|
+ mockUP.EXPECT().FindPermIdsByUserIdAndEffect(gomock.Any(), int64(1), consts.PermEffectDeny).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -210,7 +210,7 @@ func TestGetUserPerms_Mock_Member_FindDenyPermIdsFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|
|
|
@@ -235,7 +235,7 @@ func TestGetUserPerms_Mock_Member_FindPermByIdsFail(t *testing.T) {
|
|
|
mockUP.EXPECT().FindPermIdsByUserIdAndEffect(gomock.Any(), int64(1), consts.PermEffectDeny).Return([]int64{}, nil)
|
|
|
|
|
|
mockPerm := mocks.NewMockSysPermModel(ctrl)
|
|
|
- mockPerm.EXPECT().FindByIds(gomock.Any(), gomock.Any()).Return(nil, dbErr)
|
|
|
+ mockPerm.EXPECT().FindByIds(gomock.Any(), gomock.Any()).Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -246,7 +246,7 @@ func TestGetUserPerms_Mock_Member_FindPermByIdsFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 0, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|
|
|
@@ -293,7 +293,7 @@ func TestGetUserPerms_Mock_Member_DevDeptFindAllCodesFail(t *testing.T) {
|
|
|
Return(&dept.SysDept{Id: 100, DeptType: consts.DeptTypeDev}, nil)
|
|
|
|
|
|
mockPerm := mocks.NewMockSysPermModel(ctrl)
|
|
|
- mockPerm.EXPECT().FindAllCodesByProductCode(gomock.Any(), "pc").Return(nil, dbErr)
|
|
|
+ mockPerm.EXPECT().FindAllCodesByProductCode(gomock.Any(), "pc").Return(nil, errDB)
|
|
|
|
|
|
svcCtx := mocks.NewMockServiceContext(mocks.MockModels{
|
|
|
ProductMember: mockPM,
|
|
|
@@ -302,7 +302,7 @@ func TestGetUserPerms_Mock_Member_DevDeptFindAllCodesFail(t *testing.T) {
|
|
|
})
|
|
|
|
|
|
perms, memberType, err := GetUserPerms(context.Background(), svcCtx, 1, 100, "pc", false)
|
|
|
- assert.ErrorIs(t, err, dbErr)
|
|
|
+ assert.ErrorIs(t, err, errDB)
|
|
|
assert.Nil(t, perms)
|
|
|
assert.Equal(t, consts.MemberTypeMember, memberType)
|
|
|
}
|