Undo change to vyper test

crytic · Oct 12, 2023 · ef42535 · ef42535
1 parent a8b933c
commit ef42535
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/vyper/auction.vy b/tests/vyper/auction.vy
@@ -37,6 +37,11 @@ pendingReturns: HashMap[address, uint256]
 # Create a blinded auction with `_biddingTime` seconds bidding time and
 # `_revealTime` seconds reveal time on behalf of the beneficiary address
 # `_beneficiary`.
+@external
+def __init__(_beneficiary: address, _biddingTime: uint256, _revealTime: uint256):
+    self.beneficiary = _beneficiary
+    self.biddingEnd = block.timestamp + _biddingTime
+    self.revealEnd = self.biddingEnd + _revealTime
 
 
 # Place a blinded bid with:
@@ -170,4 +175,4 @@ def auctionEnd():
     self.ended = True
 
     # Transfer funds to beneficiary
-    send(self.beneficiary, self.highestBid)
+    send(self.beneficiary, self.highestBid)