github.com/xmidt-org/webpa-common@v1.11.9/secure/mocks_test.go (about)

     1  package secure
     2  
     3  import (
     4  	"github.com/SermoDigital/jose"
     5  	"github.com/SermoDigital/jose/crypto"
     6  	"github.com/SermoDigital/jose/jws"
     7  	"github.com/SermoDigital/jose/jwt"
     8  	"github.com/stretchr/testify/mock"
     9  )
    10  
    11  type mockJWSParser struct {
    12  	mock.Mock
    13  }
    14  
    15  func (parser *mockJWSParser) ParseJWS(token *Token) (jws.JWS, error) {
    16  	arguments := parser.Called(token)
    17  	jwsToken, _ := arguments.Get(0).(jws.JWS)
    18  	return jwsToken, arguments.Error(1)
    19  }
    20  
    21  type mockJWS struct {
    22  	mock.Mock
    23  }
    24  
    25  var _ jwt.JWT = (*mockJWS)(nil)
    26  var _ jws.JWS = (*mockJWS)(nil)
    27  
    28  func (j *mockJWS) Claims() jwt.Claims {
    29  	arguments := j.Called()
    30  	return arguments.Get(0).(jwt.Claims)
    31  }
    32  
    33  func (j *mockJWS) Validate(key interface{}, method crypto.SigningMethod, v ...*jwt.Validator) error {
    34  	arguments := j.Called(key, method, v)
    35  	return arguments.Error(0)
    36  }
    37  
    38  func (j *mockJWS) Serialize(key interface{}) ([]byte, error) {
    39  	arguments := j.Called(key)
    40  	return arguments.Get(0).([]byte), arguments.Error(1)
    41  }
    42  
    43  func (j *mockJWS) Payload() interface{} {
    44  	arguments := j.Called()
    45  	return arguments.Get(0)
    46  }
    47  
    48  func (j *mockJWS) SetPayload(p interface{}) {
    49  	j.Called(p)
    50  }
    51  
    52  func (j *mockJWS) Protected() jose.Protected {
    53  	arguments := j.Called()
    54  	protected, _ := arguments.Get(0).(jose.Protected)
    55  	return protected
    56  }
    57  
    58  func (j *mockJWS) ProtectedAt(i int) jose.Protected {
    59  	arguments := j.Called(i)
    60  	return arguments.Get(0).(jose.Protected)
    61  }
    62  
    63  func (j *mockJWS) Header() jose.Header {
    64  	arguments := j.Called()
    65  	return arguments.Get(0).(jose.Header)
    66  }
    67  
    68  func (j *mockJWS) HeaderAt(i int) jose.Header {
    69  	arguments := j.Called(i)
    70  	return arguments.Get(0).(jose.Header)
    71  }
    72  
    73  func (j *mockJWS) Verify(key interface{}, method crypto.SigningMethod) error {
    74  	arguments := j.Called(key, method)
    75  	return arguments.Error(0)
    76  }
    77  
    78  func (j *mockJWS) VerifyMulti(keys []interface{}, methods []crypto.SigningMethod, o *jws.SigningOpts) error {
    79  	arguments := j.Called(keys, methods, o)
    80  	return arguments.Error(0)
    81  }
    82  
    83  func (j *mockJWS) VerifyCallback(fn jws.VerifyCallback, methods []crypto.SigningMethod, o *jws.SigningOpts) error {
    84  	arguments := j.Called(fn, methods, o)
    85  	return arguments.Error(0)
    86  }
    87  
    88  func (j *mockJWS) General(keys ...interface{}) ([]byte, error) {
    89  	arguments := j.Called(keys)
    90  	return arguments.Get(0).([]byte), arguments.Error(1)
    91  }
    92  
    93  func (j *mockJWS) Flat(key interface{}) ([]byte, error) {
    94  	arguments := j.Called(key)
    95  	return arguments.Get(0).([]byte), arguments.Error(1)
    96  }
    97  
    98  func (j *mockJWS) Compact(key interface{}) ([]byte, error) {
    99  	arguments := j.Called(key)
   100  	return arguments.Get(0).([]byte), arguments.Error(1)
   101  }
   102  
   103  func (j *mockJWS) IsJWT() bool {
   104  	arguments := j.Called()
   105  	return arguments.Bool(0)
   106  }