Skip to content

Commit 85f3014

Browse files
committed
Add more delegation tests
1 parent 196de66 commit 85f3014

File tree

3 files changed

+370
-435
lines changed

3 files changed

+370
-435
lines changed

test/FlexVotingClient.t.sol

Lines changed: 56 additions & 140 deletions
Original file line numberDiff line numberDiff line change
@@ -24,167 +24,125 @@ contract BlockNumberClock_Deployment is Deployment {
2424
function _timestampClock() internal pure override returns (bool) {
2525
return false;
2626
}
27-
function _deployClient(address _governor) internal override {
28-
client = address(new MockFlexVotingClient(_governor));
29-
}
30-
function flexClient() public view override returns (MockFlexVotingClient) {
31-
return MockFlexVotingClient(client);
27+
function _deployFlexClient(address _governor) internal override {
28+
flexClient = new MockFlexVotingClient(_governor);
3229
}
3330
}
3431

3532
contract BlockNumberClock_Constructor is Constructor {
3633
function _timestampClock() internal pure override returns (bool) {
3734
return false;
3835
}
39-
function _deployClient(address _governor) internal override {
40-
client = address(new MockFlexVotingClient(_governor));
41-
}
42-
function flexClient() public view override returns (MockFlexVotingClient) {
43-
return MockFlexVotingClient(client);
36+
function _deployFlexClient(address _governor) internal override {
37+
flexClient = new MockFlexVotingClient(_governor);
4438
}
4539
}
4640

4741
contract BlockNumberClock__RawBalanceOf is _RawBalanceOf {
4842
function _timestampClock() internal pure override returns (bool) {
4943
return false;
5044
}
51-
function _deployClient(address _governor) internal override {
52-
client = address(new MockFlexVotingClient(_governor));
53-
}
54-
function flexClient() public view override returns (MockFlexVotingClient) {
55-
return MockFlexVotingClient(client);
45+
function _deployFlexClient(address _governor) internal override {
46+
flexClient = new MockFlexVotingClient(_governor);
5647
}
5748
}
5849

5950
contract BlockNumberClock__CastVoteReasonString is _CastVoteReasonString {
6051
function _timestampClock() internal pure override returns (bool) {
6152
return false;
6253
}
63-
function _deployClient(address _governor) internal override {
64-
client = address(new MockFlexVotingClient(_governor));
65-
}
66-
function flexClient() public view override returns (MockFlexVotingClient) {
67-
return MockFlexVotingClient(client);
54+
function _deployFlexClient(address _governor) internal override {
55+
flexClient = new MockFlexVotingClient(_governor);
6856
}
6957
}
7058

7159
contract BlockNumberClock__SelfDelegate is _SelfDelegate {
7260
function _timestampClock() internal pure override returns (bool) {
7361
return false;
7462
}
75-
function _deployClient(address _governor) internal override {
76-
client = address(new MockFlexVotingClient(_governor));
77-
}
78-
function flexClient() public view override returns (MockFlexVotingClient) {
79-
return MockFlexVotingClient(client);
63+
function _deployFlexClient(address _governor) internal override {
64+
flexClient = new MockFlexVotingClient(_governor);
8065
}
8166
}
8267

8368
contract BlockNumberClock__CheckpointRawBalanceOf is _CheckpointRawBalanceOf {
8469
function _timestampClock() internal pure override returns (bool) {
8570
return false;
8671
}
87-
function _deployClient(address _governor) internal override {
88-
client = address(new MockFlexVotingClient(_governor));
89-
}
90-
function flexClient() public view override returns (MockFlexVotingClient) {
91-
return MockFlexVotingClient(client);
72+
function _deployFlexClient(address _governor) internal override {
73+
flexClient = new MockFlexVotingClient(_governor);
9274
}
9375
}
9476

9577
contract BlockNumberClock_GetPastRawBalance is GetPastRawBalance {
9678
function _timestampClock() internal pure override returns (bool) {
9779
return false;
9880
}
99-
function _deployClient(address _governor) internal override {
100-
client = address(new MockFlexVotingClient(_governor));
101-
}
102-
function flexClient() public view override returns (MockFlexVotingClient) {
103-
return MockFlexVotingClient(client);
81+
function _deployFlexClient(address _governor) internal override {
82+
flexClient = new MockFlexVotingClient(_governor);
10483
}
10584
}
10685

10786
contract BlockNumber__CheckpointTotalBalance is _CheckpointTotalBalance {
10887
function _timestampClock() internal pure override returns (bool) {
10988
return false;
11089
}
111-
function _deployClient(address _governor) internal override {
112-
client = address(new MockFlexVotingClient(_governor));
113-
}
114-
function flexClient() public view override returns (MockFlexVotingClient) {
115-
return MockFlexVotingClient(client);
90+
function _deployFlexClient(address _governor) internal override {
91+
flexClient = new MockFlexVotingClient(_governor);
11692
}
11793
}
11894

11995
contract BlockNumberClock_GetPastTotalBalance is GetPastTotalBalance {
12096
function _timestampClock() internal pure override returns (bool) {
12197
return false;
12298
}
123-
function _deployClient(address _governor) internal override {
124-
client = address(new MockFlexVotingClient(_governor));
125-
}
126-
function flexClient() public view override returns (MockFlexVotingClient) {
127-
return MockFlexVotingClient(client);
99+
function _deployFlexClient(address _governor) internal override {
100+
flexClient = new MockFlexVotingClient(_governor);
128101
}
129102
}
130103

131104
contract BlockNumberClock_Withdraw is Withdraw {
132105
function _timestampClock() internal pure override returns (bool) {
133106
return false;
134107
}
135-
function _deployClient(address _governor) internal override {
136-
client = address(new MockFlexVotingClient(_governor));
137-
}
138-
function flexClient() public view override returns (MockFlexVotingClient) {
139-
return MockFlexVotingClient(client);
108+
function _deployFlexClient(address _governor) internal override {
109+
flexClient = new MockFlexVotingClient(_governor);
140110
}
141111
}
142112

143113
contract BlockNumberClock_Deposit is Deposit {
144114
function _timestampClock() internal pure override returns (bool) {
145115
return false;
146116
}
147-
function _deployClient(address _governor) internal override {
148-
client = address(new MockFlexVotingClient(_governor));
149-
}
150-
function flexClient() public view override returns (MockFlexVotingClient) {
151-
return MockFlexVotingClient(client);
117+
function _deployFlexClient(address _governor) internal override {
118+
flexClient = new MockFlexVotingClient(_governor);
152119
}
153120
}
154121

155122
contract BlockNumberClock_ExpressVote is ExpressVote {
156123
function _timestampClock() internal pure override returns (bool) {
157124
return false;
158125
}
159-
function _deployClient(address _governor) internal override {
160-
client = address(new MockFlexVotingClient(_governor));
161-
}
162-
function flexClient() public view override returns (MockFlexVotingClient) {
163-
return MockFlexVotingClient(client);
126+
function _deployFlexClient(address _governor) internal override {
127+
flexClient = new MockFlexVotingClient(_governor);
164128
}
165129
}
166130

167131
contract BlockNumberClock_CastVote is CastVote {
168132
function _timestampClock() internal pure override returns (bool) {
169133
return false;
170134
}
171-
function _deployClient(address _governor) internal override {
172-
client = address(new MockFlexVotingClient(_governor));
173-
}
174-
function flexClient() public view override returns (MockFlexVotingClient) {
175-
return MockFlexVotingClient(client);
135+
function _deployFlexClient(address _governor) internal override {
136+
flexClient = new MockFlexVotingClient(_governor);
176137
}
177138
}
178139

179140
contract BlockNumberClock_Borrow is Borrow {
180141
function _timestampClock() internal pure override returns (bool) {
181142
return false;
182143
}
183-
function _deployClient(address _governor) internal override {
184-
client = address(new MockFlexVotingClient(_governor));
185-
}
186-
function flexClient() public view override returns (MockFlexVotingClient) {
187-
return MockFlexVotingClient(client);
144+
function _deployFlexClient(address _governor) internal override {
145+
flexClient = new MockFlexVotingClient(_governor);
188146
}
189147
}
190148

@@ -193,166 +151,124 @@ contract TimestampClock_Deployment is Deployment {
193151
function _timestampClock() internal pure override returns (bool) {
194152
return true;
195153
}
196-
function _deployClient(address _governor) internal override {
197-
client = address(new MockFlexVotingClient(_governor));
198-
}
199-
function flexClient() public view override returns (MockFlexVotingClient) {
200-
return MockFlexVotingClient(client);
154+
function _deployFlexClient(address _governor) internal override {
155+
flexClient = new MockFlexVotingClient(_governor);
201156
}
202157
}
203158

204159
contract TimestampClock_Constructor is Constructor {
205160
function _timestampClock() internal pure override returns (bool) {
206161
return true;
207162
}
208-
function _deployClient(address _governor) internal override {
209-
client = address(new MockFlexVotingClient(_governor));
210-
}
211-
function flexClient() public view override returns (MockFlexVotingClient) {
212-
return MockFlexVotingClient(client);
163+
function _deployFlexClient(address _governor) internal override {
164+
flexClient = new MockFlexVotingClient(_governor);
213165
}
214166
}
215167

216168
contract TimestampClock__RawBalanceOf is _RawBalanceOf {
217169
function _timestampClock() internal pure override returns (bool) {
218170
return true;
219171
}
220-
function _deployClient(address _governor) internal override {
221-
client = address(new MockFlexVotingClient(_governor));
222-
}
223-
function flexClient() public view override returns (MockFlexVotingClient) {
224-
return MockFlexVotingClient(client);
172+
function _deployFlexClient(address _governor) internal override {
173+
flexClient = new MockFlexVotingClient(_governor);
225174
}
226175
}
227176

228177
contract TimestampClock__CastVoteReasonString is _CastVoteReasonString {
229178
function _timestampClock() internal pure override returns (bool) {
230179
return true;
231180
}
232-
function _deployClient(address _governor) internal override {
233-
client = address(new MockFlexVotingClient(_governor));
234-
}
235-
function flexClient() public view override returns (MockFlexVotingClient) {
236-
return MockFlexVotingClient(client);
181+
function _deployFlexClient(address _governor) internal override {
182+
flexClient = new MockFlexVotingClient(_governor);
237183
}
238184
}
239185

240186
contract TimestampClock__SelfDelegate is _SelfDelegate {
241187
function _timestampClock() internal pure override returns (bool) {
242188
return true;
243189
}
244-
function _deployClient(address _governor) internal override {
245-
client = address(new MockFlexVotingClient(_governor));
246-
}
247-
function flexClient() public view override returns (MockFlexVotingClient) {
248-
return MockFlexVotingClient(client);
190+
function _deployFlexClient(address _governor) internal override {
191+
flexClient = new MockFlexVotingClient(_governor);
249192
}
250193
}
251194

252195
contract TimestampClock__CheckpointRawBalanceOf is _CheckpointRawBalanceOf {
253196
function _timestampClock() internal pure override returns (bool) {
254197
return true;
255198
}
256-
function _deployClient(address _governor) internal override {
257-
client = address(new MockFlexVotingClient(_governor));
258-
}
259-
function flexClient() public view override returns (MockFlexVotingClient) {
260-
return MockFlexVotingClient(client);
199+
function _deployFlexClient(address _governor) internal override {
200+
flexClient = new MockFlexVotingClient(_governor);
261201
}
262202
}
263203

264204
contract TimestampClock_GetPastRawBalance is GetPastRawBalance {
265205
function _timestampClock() internal pure override returns (bool) {
266206
return true;
267207
}
268-
function _deployClient(address _governor) internal override {
269-
client = address(new MockFlexVotingClient(_governor));
270-
}
271-
function flexClient() public view override returns (MockFlexVotingClient) {
272-
return MockFlexVotingClient(client);
208+
function _deployFlexClient(address _governor) internal override {
209+
flexClient = new MockFlexVotingClient(_governor);
273210
}
274211
}
275212

276213
contract TimestampClock__CheckpointTotalBalance is _CheckpointTotalBalance {
277214
function _timestampClock() internal pure override returns (bool) {
278215
return true;
279216
}
280-
function _deployClient(address _governor) internal override {
281-
client = address(new MockFlexVotingClient(_governor));
282-
}
283-
function flexClient() public view override returns (MockFlexVotingClient) {
284-
return MockFlexVotingClient(client);
217+
function _deployFlexClient(address _governor) internal override {
218+
flexClient = new MockFlexVotingClient(_governor);
285219
}
286220
}
287221

288222
contract TimestampClock_GetPastTotalBalance is GetPastTotalBalance {
289223
function _timestampClock() internal pure override returns (bool) {
290224
return true;
291225
}
292-
function _deployClient(address _governor) internal override {
293-
client = address(new MockFlexVotingClient(_governor));
294-
}
295-
function flexClient() public view override returns (MockFlexVotingClient) {
296-
return MockFlexVotingClient(client);
226+
function _deployFlexClient(address _governor) internal override {
227+
flexClient = new MockFlexVotingClient(_governor);
297228
}
298229
}
299230

300231
contract TimestampClock_Withdraw is Withdraw {
301232
function _timestampClock() internal pure override returns (bool) {
302233
return true;
303234
}
304-
function _deployClient(address _governor) internal override {
305-
client = address(new MockFlexVotingClient(_governor));
306-
}
307-
function flexClient() public view override returns (MockFlexVotingClient) {
308-
return MockFlexVotingClient(client);
235+
function _deployFlexClient(address _governor) internal override {
236+
flexClient = new MockFlexVotingClient(_governor);
309237
}
310238
}
311239

312240
contract TimestampClock_Deposit is Deposit {
313241
function _timestampClock() internal pure override returns (bool) {
314242
return true;
315243
}
316-
function _deployClient(address _governor) internal override {
317-
client = address(new MockFlexVotingClient(_governor));
318-
}
319-
function flexClient() public view override returns (MockFlexVotingClient) {
320-
return MockFlexVotingClient(client);
244+
function _deployFlexClient(address _governor) internal override {
245+
flexClient = new MockFlexVotingClient(_governor);
321246
}
322247
}
323248

324249
contract TimestampClock_ExpressVote is ExpressVote {
325250
function _timestampClock() internal pure override returns (bool) {
326251
return true;
327252
}
328-
function _deployClient(address _governor) internal override {
329-
client = address(new MockFlexVotingClient(_governor));
330-
}
331-
function flexClient() public view override returns (MockFlexVotingClient) {
332-
return MockFlexVotingClient(client);
253+
function _deployFlexClient(address _governor) internal override {
254+
flexClient = new MockFlexVotingClient(_governor);
333255
}
334256
}
335257

336258
contract TimestampClock_CastVote is CastVote {
337259
function _timestampClock() internal pure override returns (bool) {
338260
return true;
339261
}
340-
function _deployClient(address _governor) internal override {
341-
client = address(new MockFlexVotingClient(_governor));
342-
}
343-
function flexClient() public view override returns (MockFlexVotingClient) {
344-
return MockFlexVotingClient(client);
262+
function _deployFlexClient(address _governor) internal override {
263+
flexClient = new MockFlexVotingClient(_governor);
345264
}
346265
}
347266

348267
contract TimestampClock_Borrow is Borrow {
349268
function _timestampClock() internal pure override returns (bool) {
350269
return true;
351270
}
352-
function _deployClient(address _governor) internal override {
353-
client = address(new MockFlexVotingClient(_governor));
354-
}
355-
function flexClient() public view override returns (MockFlexVotingClient) {
356-
return MockFlexVotingClient(client);
271+
function _deployFlexClient(address _governor) internal override {
272+
flexClient = new MockFlexVotingClient(_governor);
357273
}
358274
}

0 commit comments

Comments
 (0)