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  }