From 9941a636bbce87ca556149be75bcb9b81e86815e Mon Sep 17 00:00:00 2001
From: Macpie <pierre@nova-labs.com>
Date: Wed, 16 Oct 2024 12:52:33 -0700
Subject: [PATCH] Fix test

---
 mobile_verifier/tests/integrations/last_location.rs | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/mobile_verifier/tests/integrations/last_location.rs b/mobile_verifier/tests/integrations/last_location.rs
index c04d0678d..723a1bbef 100644
--- a/mobile_verifier/tests/integrations/last_location.rs
+++ b/mobile_verifier/tests/integrations/last_location.rs
@@ -8,7 +8,10 @@ use helium_proto::services::poc_mobile::{self as proto, LocationSource};
 use mobile_verifier::{
     coverage::{CoverageObject, CoverageObjectCache},
     geofence::GeofenceValidator,
-    heartbeats::{last_location::LocationCache, HbType, Heartbeat, ValidatedHeartbeat},
+    heartbeats::{
+        last_location::{Key, LocationCache},
+        HbType, Heartbeat, ValidatedHeartbeat,
+    },
 };
 use rust_decimal_macros::dec;
 use sqlx::{PgPool, Postgres, Transaction};
@@ -122,7 +125,9 @@ async fn heartbeat_will_use_last_good_location_from_db(pool: PgPool) -> anyhow::
         dec!(1.0)
     );
 
-    location_cache.delete_last_location(&hotspot).await;
+    location_cache
+        .delete_last_location(Key::WifiPubKey(hotspot.clone()))
+        .await;
     transaction = pool.begin().await?;
     validated_heartbeat_1.clone().save(&mut transaction).await?;
     transaction.commit().await?;