Browse Source

Working on test for storage

JoostSijm 4 years ago
parent
commit
522ad6edc4
1 changed files with 9 additions and 0 deletions
  1. 9 0
      tests/test_rival_regions_wrapper.py

+ 9 - 0
tests/test_rival_regions_wrapper.py

@@ -27,3 +27,12 @@ def test_profile_info(profile_keys):
     assert isinstance(response['strenght'], int), "strenght should be a int"
     assert isinstance(response['education'], int), "education should be a int"
     assert isinstance(response['endurance'], int), "endurance should be a int"
+
+@pytest.fixture
+def storage_keys():
+    """Standard keys for storage"""
+    return [
+        'profile_id', 'oil', 'ore', 'uranium', 'diamonds', 'liquid_oxygen', 'rivalium',
+        'antirad', 'energy_drink', 'spacerockets', 'lss', 'tanks', 'aircrafts',
+        'missiles', 'bombers', 'battleships', 'laser_drones', 'moon_tanks', 'space_stations'
+    ]