Merge pull request #20 from jplacht/fix-planet-distances

fix #19 distance results on planet search

Author: jplacht

fio_wrapper/endpoints/endpoints_v1/planet.py

@@ -169,7 +169,7 @@ def search(
                 distance_checks=distance_checks
             ):
                 raise PlanetSearchDistanceChecksInvalid(
-                    "Invalid distance checks. Can check for up to 3 distances."
+                    "Invalid distance checks. Can check for up to 10 distances."
                 )
 
         (status, data) = self.adapter.post(

fio_wrapper/models/planet_models.py

@@ -106,6 +106,7 @@ class PlanetFull(Planet):
     PlanetTier: int
     UserNameSubmitted: str
     Timestamp: NaiveDatetime
+    DistanceResults: Optional[List[int]]
 
 
 class PlanetFullList(RootModel):

fio_wrapper/validators.py

@@ -98,7 +98,7 @@ def validate_planet_search_distance_checks(
     if distance_checks is None:
         return False
 
-    if len(distance_checks) > 3:
+    if len(distance_checks) > 10:
         return False
 
     # ensure all distances are str

tests/test_planet_v1.py

@@ -645,6 +645,7 @@ def planet_full_1() -> Dict:
         "PlanetTier": 0,
         "UserNameSubmitted": "STRAKE",
         "Timestamp": "2023-07-15T00:21:41.935321",
+        "DistanceResults": [],
     }
 
 

tests/test_validators.py

@@ -47,8 +47,8 @@ def test_validate_planet_search_materials(input, expected) -> None:
     "input, expected",
     [
         ([], True),
-        (["1", "2", "3"], True),
-        (["1", "2", "3", "4"], False),
+        (["1", "2", "3", "4", "5", "6", "7", "8", "9", "10"], True),
+        (["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11"], False),
         (None, False),
     ],
 )