/
mocktokenaccessor.go
159 lines (134 loc) · 7.33 KB
/
mocktokenaccessor.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
// Code generated by MockGen. DO NOT EDIT.
// Source: controller/internal/enforcer/nfqdatapath/tokenaccessor/interfaces.go
// Package mocktokenaccessor is a generated GoMock package.
package mocktokenaccessor
import (
ecdsa "crypto/ecdsa"
reflect "reflect"
time "time"
gomock "github.com/golang/mock/gomock"
ephemeralkeys "go.aporeto.io/enforcerd/trireme-lib/controller/internal/enforcer/utils/ephemeralkeys"
claimsheader "go.aporeto.io/enforcerd/trireme-lib/controller/pkg/claimsheader"
pkiverifier "go.aporeto.io/enforcerd/trireme-lib/controller/pkg/pkiverifier"
secrets "go.aporeto.io/enforcerd/trireme-lib/controller/pkg/secrets"
tokens "go.aporeto.io/enforcerd/trireme-lib/controller/pkg/tokens"
)
// MockTokenAccessor is a mock of TokenAccessor interface
type MockTokenAccessor struct {
ctrl *gomock.Controller
recorder *MockTokenAccessorMockRecorder
}
// MockTokenAccessorMockRecorder is the mock recorder for MockTokenAccessor
type MockTokenAccessorMockRecorder struct {
mock *MockTokenAccessor
}
// NewMockTokenAccessor creates a new mock instance
func NewMockTokenAccessor(ctrl *gomock.Controller) *MockTokenAccessor {
mock := &MockTokenAccessor{ctrl: ctrl}
mock.recorder = &MockTokenAccessorMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockTokenAccessor) EXPECT() *MockTokenAccessorMockRecorder {
return m.recorder
}
// GetTokenValidity mocks base method
func (m *MockTokenAccessor) GetTokenValidity() time.Duration {
ret := m.ctrl.Call(m, "GetTokenValidity")
ret0, _ := ret[0].(time.Duration)
return ret0
}
// GetTokenValidity indicates an expected call of GetTokenValidity
func (mr *MockTokenAccessorMockRecorder) GetTokenValidity() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTokenValidity", reflect.TypeOf((*MockTokenAccessor)(nil).GetTokenValidity))
}
// GetTokenServerID mocks base method
func (m *MockTokenAccessor) GetTokenServerID() string {
ret := m.ctrl.Call(m, "GetTokenServerID")
ret0, _ := ret[0].(string)
return ret0
}
// GetTokenServerID indicates an expected call of GetTokenServerID
func (mr *MockTokenAccessorMockRecorder) GetTokenServerID() *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTokenServerID", reflect.TypeOf((*MockTokenAccessor)(nil).GetTokenServerID))
}
// CreateAckPacketToken mocks base method
func (m *MockTokenAccessor) CreateAckPacketToken(proto314 bool, secretKey []byte, claims *tokens.ConnectionClaims, encodedBuf []byte) ([]byte, error) {
ret := m.ctrl.Call(m, "CreateAckPacketToken", proto314, secretKey, claims, encodedBuf)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateAckPacketToken indicates an expected call of CreateAckPacketToken
func (mr *MockTokenAccessorMockRecorder) CreateAckPacketToken(proto314, secretKey, claims, encodedBuf interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateAckPacketToken", reflect.TypeOf((*MockTokenAccessor)(nil).CreateAckPacketToken), proto314, secretKey, claims, encodedBuf)
}
// CreateSynPacketToken mocks base method
func (m *MockTokenAccessor) CreateSynPacketToken(claims *tokens.ConnectionClaims, encodedBuf, nonce []byte, claimsHeader *claimsheader.ClaimsHeader, secrets secrets.Secrets) ([]byte, error) {
ret := m.ctrl.Call(m, "CreateSynPacketToken", claims, encodedBuf, nonce, claimsHeader, secrets)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateSynPacketToken indicates an expected call of CreateSynPacketToken
func (mr *MockTokenAccessorMockRecorder) CreateSynPacketToken(claims, encodedBuf, nonce, claimsHeader, secrets interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSynPacketToken", reflect.TypeOf((*MockTokenAccessor)(nil).CreateSynPacketToken), claims, encodedBuf, nonce, claimsHeader, secrets)
}
// CreateSynAckPacketToken mocks base method
func (m *MockTokenAccessor) CreateSynAckPacketToken(proto314 bool, claims *tokens.ConnectionClaims, encodedBuf, nonce []byte, claimsHeader *claimsheader.ClaimsHeader, secrets secrets.Secrets, secretKey []byte) ([]byte, error) {
ret := m.ctrl.Call(m, "CreateSynAckPacketToken", proto314, claims, encodedBuf, nonce, claimsHeader, secrets, secretKey)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// CreateSynAckPacketToken indicates an expected call of CreateSynAckPacketToken
func (mr *MockTokenAccessorMockRecorder) CreateSynAckPacketToken(proto314, claims, encodedBuf, nonce, claimsHeader, secrets, secretKey interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateSynAckPacketToken", reflect.TypeOf((*MockTokenAccessor)(nil).CreateSynAckPacketToken), proto314, claims, encodedBuf, nonce, claimsHeader, secrets, secretKey)
}
// ParsePacketToken mocks base method
func (m *MockTokenAccessor) ParsePacketToken(privateKey *ephemeralkeys.PrivateKey, data []byte, secrets secrets.Secrets, c *tokens.ConnectionClaims, b bool) ([]byte, *claimsheader.ClaimsHeader, *pkiverifier.PKIControllerInfo, []byte, string, bool, error) {
ret := m.ctrl.Call(m, "ParsePacketToken", privateKey, data, secrets, c, b)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(*claimsheader.ClaimsHeader)
ret2, _ := ret[2].(*pkiverifier.PKIControllerInfo)
ret3, _ := ret[3].([]byte)
ret4, _ := ret[4].(string)
ret5, _ := ret[5].(bool)
ret6, _ := ret[6].(error)
return ret0, ret1, ret2, ret3, ret4, ret5, ret6
}
// ParsePacketToken indicates an expected call of ParsePacketToken
func (mr *MockTokenAccessorMockRecorder) ParsePacketToken(privateKey, data, secrets, c, b interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ParsePacketToken", reflect.TypeOf((*MockTokenAccessor)(nil).ParsePacketToken), privateKey, data, secrets, c, b)
}
// ParseAckToken mocks base method
func (m *MockTokenAccessor) ParseAckToken(proto314 bool, secretKey, nonce, data []byte, connClaims *tokens.ConnectionClaims) error {
ret := m.ctrl.Call(m, "ParseAckToken", proto314, secretKey, nonce, data, connClaims)
ret0, _ := ret[0].(error)
return ret0
}
// ParseAckToken indicates an expected call of ParseAckToken
func (mr *MockTokenAccessorMockRecorder) ParseAckToken(proto314, secretKey, nonce, data, connClaims interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ParseAckToken", reflect.TypeOf((*MockTokenAccessor)(nil).ParseAckToken), proto314, secretKey, nonce, data, connClaims)
}
// Randomize mocks base method
func (m *MockTokenAccessor) Randomize(arg0, arg1 []byte) error {
ret := m.ctrl.Call(m, "Randomize", arg0, arg1)
ret0, _ := ret[0].(error)
return ret0
}
// Randomize indicates an expected call of Randomize
func (mr *MockTokenAccessorMockRecorder) Randomize(arg0, arg1 interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Randomize", reflect.TypeOf((*MockTokenAccessor)(nil).Randomize), arg0, arg1)
}
// Sign mocks base method
func (m *MockTokenAccessor) Sign(arg0 []byte, arg1 *ecdsa.PrivateKey) ([]byte, error) {
ret := m.ctrl.Call(m, "Sign", arg0, arg1)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// Sign indicates an expected call of Sign
func (mr *MockTokenAccessorMockRecorder) Sign(arg0, arg1 interface{}) *gomock.Call {
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Sign", reflect.TypeOf((*MockTokenAccessor)(nil).Sign), arg0, arg1)
}