Bladeren bron

Add perk upgrade function to perks module

JoostSijm 4 jaren geleden
bovenliggende
commit
5696c2e766
2 gewijzigde bestanden met toevoegingen van 29 en 0 verwijderingen
  1. 22 0
      rival_regions_wrapper/api_wrapper/perks.py
  2. 7 0
      tests/test_rival_regions_wrapper.py

+ 22 - 0
rival_regions_wrapper/api_wrapper/perks.py

@@ -42,3 +42,25 @@ class Perks(object):
             'upgrade_perk': upgrade_perk
         }
         return perks
+
+    @staticmethod
+    def upgrade(perk, upgrade_type):
+        """Craft item"""
+        perk_keys = {
+            'strength': 1,
+            'education': 2,
+            'endurance': 3,
+        }
+        if isinstance(perk, str) and perk in perk_keys:
+            perk_keys = perk_keys[perk]
+
+        upgrade_type_keys = {
+            'money': 1,
+            'gold': 2,
+        }
+        if isinstance(upgrade_type, str) and upgrade_type in upgrade_type_keys:
+            upgrade_type = upgrade_type_keys[upgrade_type]
+
+        MIDDLEWARE.post('perks/up/{}/{}'.format(perk, upgrade_type))
+        return True
+        

+ 7 - 0
tests/test_rival_regions_wrapper.py

@@ -112,6 +112,13 @@ def test_perks_info(perks_keys):
     assert isinstance(response['upgrade_date'], datetime), "upgrade_date should be a date"
     assert isinstance(response['upgrade_perk'], int), "upgrade_perk should be an int"
 
+@pytest.mark.skip(reason="Update request")
+def test_perks_upgrade():
+    """Test an API call to upgrade perk"""
+    perk = 'endurance'
+    upgrade_type = 'money'
+    Perks.upgrade(perk, upgrade_type )
+
 @pytest.fixture
 def craft_keys():
     """Standard keys for craft"""