replicating with receiver wins point

resgroup · Sep 29, 2020 · ba1deb6 · ba1deb6
1 parent 49eee7e
commit ba1deb6
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/src/kata.py b/src/kata.py
@@ -37,11 +37,37 @@ def server_wins_point(self):
             return GameScore(
                 server_score=PlayerScore.Wins,
                 receiver_score=self.receiver_score)
-        elif self.server_score == PlayerScore.Fourty and self.server_score == PlayerScore.Fourty:
+        elif self.server_score == PlayerScore.Fourty and self.receiver_score == PlayerScore.Fourty:
             return GameScore(
                 server_score=PlayerScore.Advantage,
-                receiver_score=PlayerScore.Fourty)
+                receiver_score=self.receiver_score)
         elif self.server_score == PlayerScore.Advantage:
             return GameScore(
                 server_score=PlayerScore.Wins,
                 receiver_score=self.receiver_score)
+
+    def receiver_wins_point(self):
+        if self.receiver_score == PlayerScore.Zero:
+            return GameScore(
+                server_score=self.server_score,
+                receiver_score=PlayerScore.Fifteen)
+        elif self.receiver_score == PlayerScore.Fifteen:
+            return GameScore(
+                server_score=self.server_score,
+                receiver_score=PlayerScore.Thirty)
+        elif self.receiver_score == PlayerScore.Thirty:
+            return GameScore(
+                server_score=self.server_score,
+                receiver_score=PlayerScore.Fourty)
+        elif self.receiver_score == PlayerScore.Fourty and self.server_score != PlayerScore.Fourty:
+            return GameScore(
+                server_score=self.server_score,
+                receiver_score=PlayerScore.Wins)
+        elif self.receiver_score == PlayerScore.Fourty and self.server_score == PlayerScore.Fourty:
+            return GameScore(
+                server_score=self.server_score,
+                receiver_score=PlayerScore.Advantage)
+        elif self.receiver_score == PlayerScore.Advantage:
+            return GameScore(
+                server_score=self.server_score,
+                receiver_score=PlayerScore.Wins)
diff --git a/tests/test_kata.py b/tests/test_kata.py
@@ -113,7 +113,7 @@ def test_server_wins(current_score, expected_score):
 ]
 
 
-@pytest.mark.parametrize("current_score,expected_score", server_wins)
+@pytest.mark.parametrize("current_score, expected_score", receiver_wins)
 def test_receiver_wins(current_score, expected_score):
     new_score = current_score.receiver_wins_point()