update to account for proto treating sp override as a boolean

helium · Feb 12, 2025 · 57df94e · 57df94e
1 parent e758b44
commit 57df94e
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/hex_assignments/src/assignment.rs b/hex_assignments/src/assignment.rs
@@ -32,6 +32,16 @@ impl From<Assignment> for ProtoAssignment {
     }
 }
 
+impl Into<bool> for Assignment {
+    fn into(self) -> bool {
+        match self {
+            Assignment::A => true,
+            Assignment::B => true,
+            Assignment::C => false,
+        }
+    }
+}
+
 impl From<Assignment> for i32 {
     fn from(assignment: Assignment) -> i32 {
         ProtoAssignment::from(assignment) as i32

diff --git a/mobile_verifier/tests/integrations/boosting_oracles.rs b/mobile_verifier/tests/integrations/boosting_oracles.rs
@@ -194,7 +194,7 @@ async fn test_footfall_and_urbanization_report(pool: PgPool) -> anyhow::Result<(
         urbanized.insert(hex_cell(&hex.location), hex.urbanized().into());
         footfall.insert(hex_cell(&hex.location), hex.footfall().into());
         landtype.insert(hex_cell(&hex.location), hex.landtype().into());
-        if hex.service_provider_override == Assignment::A as i32 {
+        if hex.service_provider_override {
             service_provider_override.insert(hex_cell(&hex.location));
         }
     }