github.com/jxgolibs/go-oauth2-server@v1.0.1/oauth/mocks/ServiceInterface.go (about) 1 package mocks 2 3 import "github.com/RichardKnop/go-oauth2-server/oauth" 4 import "github.com/stretchr/testify/mock" 5 6 import "github.com/RichardKnop/go-oauth2-server/config" 7 import "github.com/RichardKnop/go-oauth2-server/models" 8 import "github.com/RichardKnop/go-oauth2-server/util/routes" 9 import "github.com/gorilla/mux" 10 import "github.com/jinzhu/gorm" 11 12 type ServiceInterface struct { 13 mock.Mock 14 } 15 16 func (_m *ServiceInterface) GetConfig() *config.Config { 17 ret := _m.Called() 18 19 var r0 *config.Config 20 if rf, ok := ret.Get(0).(func() *config.Config); ok { 21 r0 = rf() 22 } else { 23 if ret.Get(0) != nil { 24 r0 = ret.Get(0).(*config.Config) 25 } 26 } 27 28 return r0 29 } 30 func (_m *ServiceInterface) RestrictToRoles(allowedRoles ...string) { 31 _m.Called(allowedRoles) 32 } 33 func (_m *ServiceInterface) IsRoleAllowed(role string) bool { 34 ret := _m.Called(role) 35 36 var r0 bool 37 if rf, ok := ret.Get(0).(func(string) bool); ok { 38 r0 = rf(role) 39 } else { 40 r0 = ret.Get(0).(bool) 41 } 42 43 return r0 44 } 45 func (_m *ServiceInterface) GetRoutes() []routes.Route { 46 ret := _m.Called() 47 48 var r0 []routes.Route 49 if rf, ok := ret.Get(0).(func() []routes.Route); ok { 50 r0 = rf() 51 } else { 52 if ret.Get(0) != nil { 53 r0 = ret.Get(0).([]routes.Route) 54 } 55 } 56 57 return r0 58 } 59 func (_m *ServiceInterface) RegisterRoutes(router *mux.Router, prefix string) { 60 _m.Called(router, prefix) 61 } 62 func (_m *ServiceInterface) ClientExists(clientID string) bool { 63 ret := _m.Called(clientID) 64 65 var r0 bool 66 if rf, ok := ret.Get(0).(func(string) bool); ok { 67 r0 = rf(clientID) 68 } else { 69 r0 = ret.Get(0).(bool) 70 } 71 72 return r0 73 } 74 func (_m *ServiceInterface) FindClientByClientID(clientID string) (*models.OauthClient, error) { 75 ret := _m.Called(clientID) 76 77 var r0 *models.OauthClient 78 if rf, ok := ret.Get(0).(func(string) *models.OauthClient); ok { 79 r0 = rf(clientID) 80 } else { 81 if ret.Get(0) != nil { 82 r0 = ret.Get(0).(*models.OauthClient) 83 } 84 } 85 86 var r1 error 87 if rf, ok := ret.Get(1).(func(string) error); ok { 88 r1 = rf(clientID) 89 } else { 90 r1 = ret.Error(1) 91 } 92 93 return r0, r1 94 } 95 func (_m *ServiceInterface) CreateClient(clientID string, secret string, redirectURI string) (*models.OauthClient, error) { 96 ret := _m.Called(clientID, secret, redirectURI) 97 98 var r0 *models.OauthClient 99 if rf, ok := ret.Get(0).(func(string, string, string) *models.OauthClient); ok { 100 r0 = rf(clientID, secret, redirectURI) 101 } else { 102 if ret.Get(0) != nil { 103 r0 = ret.Get(0).(*models.OauthClient) 104 } 105 } 106 107 var r1 error 108 if rf, ok := ret.Get(1).(func(string, string, string) error); ok { 109 r1 = rf(clientID, secret, redirectURI) 110 } else { 111 r1 = ret.Error(1) 112 } 113 114 return r0, r1 115 } 116 func (_m *ServiceInterface) CreateClientTx(tx *gorm.DB, clientID string, secret string, redirectURI string) (*models.OauthClient, error) { 117 ret := _m.Called(tx, clientID, secret, redirectURI) 118 119 var r0 *models.OauthClient 120 if rf, ok := ret.Get(0).(func(*gorm.DB, string, string, string) *models.OauthClient); ok { 121 r0 = rf(tx, clientID, secret, redirectURI) 122 } else { 123 if ret.Get(0) != nil { 124 r0 = ret.Get(0).(*models.OauthClient) 125 } 126 } 127 128 var r1 error 129 if rf, ok := ret.Get(1).(func(*gorm.DB, string, string, string) error); ok { 130 r1 = rf(tx, clientID, secret, redirectURI) 131 } else { 132 r1 = ret.Error(1) 133 } 134 135 return r0, r1 136 } 137 func (_m *ServiceInterface) AuthClient(clientID string, secret string) (*models.OauthClient, error) { 138 ret := _m.Called(clientID, secret) 139 140 var r0 *models.OauthClient 141 if rf, ok := ret.Get(0).(func(string, string) *models.OauthClient); ok { 142 r0 = rf(clientID, secret) 143 } else { 144 if ret.Get(0) != nil { 145 r0 = ret.Get(0).(*models.OauthClient) 146 } 147 } 148 149 var r1 error 150 if rf, ok := ret.Get(1).(func(string, string) error); ok { 151 r1 = rf(clientID, secret) 152 } else { 153 r1 = ret.Error(1) 154 } 155 156 return r0, r1 157 } 158 func (_m *ServiceInterface) UserExists(username string) bool { 159 ret := _m.Called(username) 160 161 var r0 bool 162 if rf, ok := ret.Get(0).(func(string) bool); ok { 163 r0 = rf(username) 164 } else { 165 r0 = ret.Get(0).(bool) 166 } 167 168 return r0 169 } 170 func (_m *ServiceInterface) FindUserByUsername(username string) (*models.OauthUser, error) { 171 ret := _m.Called(username) 172 173 var r0 *models.OauthUser 174 if rf, ok := ret.Get(0).(func(string) *models.OauthUser); ok { 175 r0 = rf(username) 176 } else { 177 if ret.Get(0) != nil { 178 r0 = ret.Get(0).(*models.OauthUser) 179 } 180 } 181 182 var r1 error 183 if rf, ok := ret.Get(1).(func(string) error); ok { 184 r1 = rf(username) 185 } else { 186 r1 = ret.Error(1) 187 } 188 189 return r0, r1 190 } 191 func (_m *ServiceInterface) CreateUser(roleID string, username string, password string) (*models.OauthUser, error) { 192 ret := _m.Called(roleID, username, password) 193 194 var r0 *models.OauthUser 195 if rf, ok := ret.Get(0).(func(string, string, string) *models.OauthUser); ok { 196 r0 = rf(roleID, username, password) 197 } else { 198 if ret.Get(0) != nil { 199 r0 = ret.Get(0).(*models.OauthUser) 200 } 201 } 202 203 var r1 error 204 if rf, ok := ret.Get(1).(func(string, string, string) error); ok { 205 r1 = rf(roleID, username, password) 206 } else { 207 r1 = ret.Error(1) 208 } 209 210 return r0, r1 211 } 212 func (_m *ServiceInterface) CreateUserTx(tx *gorm.DB, roleID string, username string, password string) (*models.OauthUser, error) { 213 ret := _m.Called(tx, roleID, username, password) 214 215 var r0 *models.OauthUser 216 if rf, ok := ret.Get(0).(func(*gorm.DB, string, string, string) *models.OauthUser); ok { 217 r0 = rf(tx, roleID, username, password) 218 } else { 219 if ret.Get(0) != nil { 220 r0 = ret.Get(0).(*models.OauthUser) 221 } 222 } 223 224 var r1 error 225 if rf, ok := ret.Get(1).(func(*gorm.DB, string, string, string) error); ok { 226 r1 = rf(tx, roleID, username, password) 227 } else { 228 r1 = ret.Error(1) 229 } 230 231 return r0, r1 232 } 233 func (_m *ServiceInterface) SetPassword(user *models.OauthUser, password string) error { 234 ret := _m.Called(user, password) 235 236 var r0 error 237 if rf, ok := ret.Get(0).(func(*models.OauthUser, string) error); ok { 238 r0 = rf(user, password) 239 } else { 240 r0 = ret.Error(0) 241 } 242 243 return r0 244 } 245 func (_m *ServiceInterface) SetPasswordTx(tx *gorm.DB, user *models.OauthUser, password string) error { 246 ret := _m.Called(tx, user, password) 247 248 var r0 error 249 if rf, ok := ret.Get(0).(func(*gorm.DB, *models.OauthUser, string) error); ok { 250 r0 = rf(tx, user, password) 251 } else { 252 r0 = ret.Error(0) 253 } 254 255 return r0 256 } 257 func (_m *ServiceInterface) UpdateUsername(user *models.OauthUser, username string) error { 258 ret := _m.Called(user, username) 259 260 var r0 error 261 if rf, ok := ret.Get(0).(func(*models.OauthUser, string) error); ok { 262 r0 = rf(user, username) 263 } else { 264 r0 = ret.Error(0) 265 } 266 267 return r0 268 } 269 func (_m *ServiceInterface) UpdateUsernameTx(db *gorm.DB, user *models.OauthUser, username string) error { 270 ret := _m.Called(db, user, username) 271 272 var r0 error 273 if rf, ok := ret.Get(0).(func(*gorm.DB, *models.OauthUser, string) error); ok { 274 r0 = rf(db, user, username) 275 } else { 276 r0 = ret.Error(0) 277 } 278 279 return r0 280 } 281 func (_m *ServiceInterface) AuthUser(username string, thePassword string) (*models.OauthUser, error) { 282 ret := _m.Called(username, thePassword) 283 284 var r0 *models.OauthUser 285 if rf, ok := ret.Get(0).(func(string, string) *models.OauthUser); ok { 286 r0 = rf(username, thePassword) 287 } else { 288 if ret.Get(0) != nil { 289 r0 = ret.Get(0).(*models.OauthUser) 290 } 291 } 292 293 var r1 error 294 if rf, ok := ret.Get(1).(func(string, string) error); ok { 295 r1 = rf(username, thePassword) 296 } else { 297 r1 = ret.Error(1) 298 } 299 300 return r0, r1 301 } 302 func (_m *ServiceInterface) GetScope(requestedScope string) (string, error) { 303 ret := _m.Called(requestedScope) 304 305 var r0 string 306 if rf, ok := ret.Get(0).(func(string) string); ok { 307 r0 = rf(requestedScope) 308 } else { 309 r0 = ret.Get(0).(string) 310 } 311 312 var r1 error 313 if rf, ok := ret.Get(1).(func(string) error); ok { 314 r1 = rf(requestedScope) 315 } else { 316 r1 = ret.Error(1) 317 } 318 319 return r0, r1 320 } 321 func (_m *ServiceInterface) Login(client *models.OauthClient, user *models.OauthUser, scope string) (*models.OauthAccessToken, *models.OauthRefreshToken, error) { 322 ret := _m.Called(client, user, scope) 323 324 var r0 *models.OauthAccessToken 325 if rf, ok := ret.Get(0).(func(*models.OauthClient, *models.OauthUser, string) *models.OauthAccessToken); ok { 326 r0 = rf(client, user, scope) 327 } else { 328 if ret.Get(0) != nil { 329 r0 = ret.Get(0).(*models.OauthAccessToken) 330 } 331 } 332 333 var r1 *models.OauthRefreshToken 334 if rf, ok := ret.Get(1).(func(*models.OauthClient, *models.OauthUser, string) *models.OauthRefreshToken); ok { 335 r1 = rf(client, user, scope) 336 } else { 337 if ret.Get(1) != nil { 338 r1 = ret.Get(1).(*models.OauthRefreshToken) 339 } 340 } 341 342 var r2 error 343 if rf, ok := ret.Get(2).(func(*models.OauthClient, *models.OauthUser, string) error); ok { 344 r2 = rf(client, user, scope) 345 } else { 346 r2 = ret.Error(2) 347 } 348 349 return r0, r1, r2 350 } 351 func (_m *ServiceInterface) GrantAuthorizationCode(client *models.OauthClient, user *models.OauthUser, expiresIn int, redirectURI string, scope string) (*models.OauthAuthorizationCode, error) { 352 ret := _m.Called(client, user, expiresIn, redirectURI, scope) 353 354 var r0 *models.OauthAuthorizationCode 355 if rf, ok := ret.Get(0).(func(*models.OauthClient, *models.OauthUser, int, string, string) *models.OauthAuthorizationCode); ok { 356 r0 = rf(client, user, expiresIn, redirectURI, scope) 357 } else { 358 if ret.Get(0) != nil { 359 r0 = ret.Get(0).(*models.OauthAuthorizationCode) 360 } 361 } 362 363 var r1 error 364 if rf, ok := ret.Get(1).(func(*models.OauthClient, *models.OauthUser, int, string, string) error); ok { 365 r1 = rf(client, user, expiresIn, redirectURI, scope) 366 } else { 367 r1 = ret.Error(1) 368 } 369 370 return r0, r1 371 } 372 func (_m *ServiceInterface) GrantAccessToken(client *models.OauthClient, user *models.OauthUser, expiresIn int, scope string) (*models.OauthAccessToken, error) { 373 ret := _m.Called(client, user, expiresIn, scope) 374 375 var r0 *models.OauthAccessToken 376 if rf, ok := ret.Get(0).(func(*models.OauthClient, *models.OauthUser, int, string) *models.OauthAccessToken); ok { 377 r0 = rf(client, user, expiresIn, scope) 378 } else { 379 if ret.Get(0) != nil { 380 r0 = ret.Get(0).(*models.OauthAccessToken) 381 } 382 } 383 384 var r1 error 385 if rf, ok := ret.Get(1).(func(*models.OauthClient, *models.OauthUser, int, string) error); ok { 386 r1 = rf(client, user, expiresIn, scope) 387 } else { 388 r1 = ret.Error(1) 389 } 390 391 return r0, r1 392 } 393 func (_m *ServiceInterface) GetOrCreateRefreshToken(client *models.OauthClient, user *models.OauthUser, expiresIn int, scope string) (*models.OauthRefreshToken, error) { 394 ret := _m.Called(client, user, expiresIn, scope) 395 396 var r0 *models.OauthRefreshToken 397 if rf, ok := ret.Get(0).(func(*models.OauthClient, *models.OauthUser, int, string) *models.OauthRefreshToken); ok { 398 r0 = rf(client, user, expiresIn, scope) 399 } else { 400 if ret.Get(0) != nil { 401 r0 = ret.Get(0).(*models.OauthRefreshToken) 402 } 403 } 404 405 var r1 error 406 if rf, ok := ret.Get(1).(func(*models.OauthClient, *models.OauthUser, int, string) error); ok { 407 r1 = rf(client, user, expiresIn, scope) 408 } else { 409 r1 = ret.Error(1) 410 } 411 412 return r0, r1 413 } 414 func (_m *ServiceInterface) GetValidRefreshToken(token string, client *models.OauthClient) (*models.OauthRefreshToken, error) { 415 ret := _m.Called(token, client) 416 417 var r0 *models.OauthRefreshToken 418 if rf, ok := ret.Get(0).(func(string, *models.OauthClient) *models.OauthRefreshToken); ok { 419 r0 = rf(token, client) 420 } else { 421 if ret.Get(0) != nil { 422 r0 = ret.Get(0).(*models.OauthRefreshToken) 423 } 424 } 425 426 var r1 error 427 if rf, ok := ret.Get(1).(func(string, *models.OauthClient) error); ok { 428 r1 = rf(token, client) 429 } else { 430 r1 = ret.Error(1) 431 } 432 433 return r0, r1 434 } 435 func (_m *ServiceInterface) Authenticate(token string) (*models.OauthAccessToken, error) { 436 ret := _m.Called(token) 437 438 var r0 *models.OauthAccessToken 439 if rf, ok := ret.Get(0).(func(string) *models.OauthAccessToken); ok { 440 r0 = rf(token) 441 } else { 442 if ret.Get(0) != nil { 443 r0 = ret.Get(0).(*models.OauthAccessToken) 444 } 445 } 446 447 var r1 error 448 if rf, ok := ret.Get(1).(func(string) error); ok { 449 r1 = rf(token) 450 } else { 451 r1 = ret.Error(1) 452 } 453 454 return r0, r1 455 } 456 func (_m *ServiceInterface) NewIntrospectResponseFromAccessToken(accessToken *models.OauthAccessToken) (*oauth.IntrospectResponse, error) { 457 ret := _m.Called(accessToken) 458 459 var r0 *oauth.IntrospectResponse 460 if rf, ok := ret.Get(0).(func(*models.OauthAccessToken) *oauth.IntrospectResponse); ok { 461 r0 = rf(accessToken) 462 } else { 463 if ret.Get(0) != nil { 464 r0 = ret.Get(0).(*oauth.IntrospectResponse) 465 } 466 } 467 468 var r1 error 469 if rf, ok := ret.Get(1).(func(*models.OauthAccessToken) error); ok { 470 r1 = rf(accessToken) 471 } else { 472 r1 = ret.Error(1) 473 } 474 475 return r0, r1 476 } 477 func (_m *ServiceInterface) NewIntrospectResponseFromRefreshToken(refreshToken *models.OauthRefreshToken) (*oauth.IntrospectResponse, error) { 478 ret := _m.Called(refreshToken) 479 480 var r0 *oauth.IntrospectResponse 481 if rf, ok := ret.Get(0).(func(*models.OauthRefreshToken) *oauth.IntrospectResponse); ok { 482 r0 = rf(refreshToken) 483 } else { 484 if ret.Get(0) != nil { 485 r0 = ret.Get(0).(*oauth.IntrospectResponse) 486 } 487 } 488 489 var r1 error 490 if rf, ok := ret.Get(1).(func(*models.OauthRefreshToken) error); ok { 491 r1 = rf(refreshToken) 492 } else { 493 r1 = ret.Error(1) 494 } 495 496 return r0, r1 497 }