Only invalidate unasserted gateways if they are from WiFi radios

helium · Nov 14, 2023 · a5c6774 · a5c6774
1 parent f4ebc65
commit a5c6774
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/mobile_verifier/src/heartbeats/mod.rs b/mobile_verifier/src/heartbeats/mod.rs
@@ -496,21 +496,18 @@ pub async fn validate_heartbeat(
                 proto::HeartbeatValidity::GatewayNotFound,
             ))
         }
-        GatewayResolution::GatewayNotAsserted => {
+        GatewayResolution::GatewayNotAsserted if heartbeat.hb_type == HbType::Wifi => {
             return Ok((
                 cell_type,
                 None,
                 None,
                 proto::HeartbeatValidity::GatewayNotAsserted,
             ))
         }
-        GatewayResolution::AssertedLocation(location) => {
-            if heartbeat.hb_type == HbType::Wifi {
-                Some(heartbeat.asserted_distance(location)?)
-            } else {
-                None
-            }
+        GatewayResolution::AssertedLocation(location) if heartbeat.hb_type == HbType::Wifi => {
+            Some(heartbeat.asserted_distance(location)?)
         }
+        _ => None,
     };
 
     /*