feat: using pytest

Author: heyloh

src/auction/domain.py

@@ -3,13 +3,27 @@
 
 class User:
 
-    def __init__(self, name):
+    def __init__(self, name, wallet=500.0):
         self.__name = name
+        self.__wallet = wallet
 
     @property
     def name(self):
         return self.__name
 
+    @property
+    def wallet(self):
+        return self.__wallet
+
+    def propose_bid(self, auction, value):
+        if value > self.__wallet:
+            raise ValueError(f'User tried to propose R${value} having only R${self.__wallet}.')
+
+        bid = Bid(self, value)
+        auction.propose(bid)
+
+        self.__wallet -= value
+
 
 class Bid:
 
@@ -28,7 +42,7 @@ def __init__(self, description):
 
     def propose(self, bid: Bid):
         if not self.__bids or self.__bids[-1].user.name != bid.user.name and bid.value > self.__bids[
-                -1].value:
+            -1].value:
             if bid.value > self.highest_bid:
                 self.highest_bid = bid.value
 

tests/__init__.py

@@ -0,0 +1,29 @@
+import pytest
+from pytest import raises
+
+from src.auction.domain import User, Auction
+
+
+@pytest.fixture
+def generic_user():
+    return User('Generic User', 100.0)
+
+
+@pytest.fixture
+def auction():
+    return Auction('Generic Auction')
+
+
+def test_should_subtract_value_from_user_wallet_when_proposing_a_bid(generic_user, auction):
+    generic_user.propose_bid(auction, 50.0)
+    assert generic_user.wallet == 50.0
+
+
+def test_should_allow_propose_bid_when_the_value_is_equal_to_the_value_on_wallet(generic_user, auction):
+    generic_user.propose_bid(auction, 100.0)
+    assert generic_user.wallet == 0.0
+
+
+def test_should_not_allow_propose_bid_when_the_value_is_higher_than_the_value_on_wallet(generic_user, auction):
+    with raises(ValueError):
+        generic_user.propose_bid(auction, 150.0)