Skip to content

Commit 9d7a82c

Browse files
committed
fix mock
1 parent 8c2036f commit 9d7a82c

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

api/user/mock.go

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,3 +70,19 @@ func (m *MockService) IsEmailExists(email string) (bool, error) {
7070
args := m.Called(email)
7171
return args.Bool(0), args.Error(1)
7272
}
73+
74+
func (m *MockService) CreateRole(code model.RoleCode) (*model.Role, error) {
75+
args := m.Called(code)
76+
if args.Get(0) == nil {
77+
return nil, args.Error(1)
78+
}
79+
return args.Get(0).(*model.Role), args.Error(1)
80+
}
81+
82+
func (m *MockService) DeleteRole(role *model.Role) (bool, error) {
83+
args := m.Called(role)
84+
if args.Get(0) == nil {
85+
return false, args.Error(1)
86+
}
87+
return args.Bool(0), args.Error(1)
88+
}

0 commit comments

Comments
 (0)