wip

Signed-off-by: chrismaree <[email protected]>

src/HoneyPot.sol

@@ -30,6 +30,7 @@ contract HoneyPot is Ownable {
 
  function createHoneyPot() external payable {
  require(honeyPots[msg.sender].liquidationPrice == 0, "Liquidation price already set for this user");
+ require(msg.value > 0, "No value sent");
 
  (, int256 currentPrice,,,) = oracle.latestRoundData();
 
@@ -55,6 +56,7 @@ contract HoneyPot is Ownable {
  HoneyPotDetails storage userPot = honeyPots[honeyPotCreator];
 
  require(currentPrice != userPot.liquidationPrice, "Liquidation price reached for this user");
+ require(userPot.balance > 0, "No balance to withdraw");
 
  _emptyPotForUser(honeyPotCreator, msg.sender);
  emit HoneyPotEmptied(honeyPotCreator, msg.sender, userPot.balance);

test/HoneyPot.sol

@@ -171,4 +171,19 @@ contract HoneyPotTest is CommonTest {
 
  assertTrue(liquidatorBalanceAfter == liquidatorBalanceBefore + honeyPotBalance);
  }
+
+ function testCreateHoneyPotWithNoValue() public {
+ vm.expectRevert("No value sent");
+ honeyPot.createHoneyPot{value: 0}();
+ }
+
+ function testEmptyHoneyPotWithZeroBalance() public {
+ // Assuming honeyPot has been created before
+ // Reset HoneyPot for the caller to ensure balance is 0
+ honeyPot.resetPot();
+
+ vm.prank(liquidator);
+ vm.expectRevert("No balance to withdraw");
+ honeyPot.emptyHoneyPot(address(this));
+ }
 }