fix tests

chain/src/repository/balance.rs

@@ -80,10 +80,13 @@ mod tests {
     use namada_sdk::token::Amount as NamadaAmount;
     use namada_sdk::uint::MAX_SIGNED_VALUE;
     use orm::balances::BalanceDb;
+    use orm::blocks::BlockInsertDb;
+    use orm::schema::blocks;
     use orm::views::balances;
     use shared::balance::{Amount, Balance};
     use shared::id::Id;
     use shared::token::IbcToken;
+    use std::collections::HashSet;
     use test_helpers::db::TestDb;
 
     use super::*;
@@ -130,6 +133,8 @@ mod tests {
 
             insert_tokens(conn, vec![token.clone()])?;
 
+            seed_blocks_from_balances(conn, &vec![balance.clone()])?;
+
             insert_balances(conn, vec![balance.clone()])?;
 
             let queried_balance = query_balance_by_address(conn, owner, token)?;
@@ -175,6 +180,7 @@ mod tests {
                 ..(balance.clone())
             };
 
+            seed_blocks_from_balances(conn, &vec![new_balance.clone()])?;
             insert_balances(conn, vec![new_balance])?;
 
             let queried_balance =
@@ -376,6 +382,8 @@ mod tests {
 
             seed_tokens_from_balance(conn, fake_balances.clone())?;
 
+            seed_blocks_from_balances(conn, &fake_balances)?;
+
             insert_balances(conn, fake_balances.clone())?;
 
             assert_eq!(query_all_balances(conn)?.len(), fake_balances.len());
@@ -410,6 +418,7 @@ mod tests {
 
             insert_tokens(conn, vec![token.clone()])?;
 
+            seed_blocks_from_balances(conn, &vec![balance.clone()])?;
             insert_balances(conn, vec![balance.clone()])?;
 
             let queried_balance = query_balance_by_address(conn, owner, token)?;
@@ -442,6 +451,8 @@ mod tests {
 
             insert_tokens(conn, vec![token])?;
 
+            seed_blocks_from_balances(conn, &balances)?;
+
             let res = insert_balances(conn, balances);
 
             assert!(res.is_ok());
@@ -475,6 +486,8 @@ mod tests {
 
             seed_tokens_from_balance(conn, balances.clone())?;
 
+            seed_blocks_from_balances(conn, &balances)?;
+
             let res = insert_balances(conn, balances);
 
             assert!(res.is_ok());
@@ -500,12 +513,33 @@ mod tests {
         anyhow::Ok(())
     }
 
+    fn seed_blocks_from_balances(
+        conn: &mut PgConnection,
+        balances: &Vec<Balance>,
+    ) -> anyhow::Result<()> {
+        for height in balances
+            .into_iter()
+            .map(|balance| balance.height as i32)
+            .collect::<HashSet<_>>()
+        {
+            diesel::insert_into(blocks::table)
+                .values::<&BlockInsertDb>(&BlockInsertDb::fake(height))
+                .on_conflict_do_nothing()
+                .execute(conn)
+                .context("Failed to insert block in db")?;
+        }
+
+        anyhow::Ok(())
+    }
+
     fn seed_balance(
         conn: &mut PgConnection,
         balances: Vec<Balance>,
     ) -> anyhow::Result<()> {
         seed_tokens_from_balance(conn, balances.clone())?;
 
+        seed_blocks_from_balances(conn, &balances)?;
+
         diesel::insert_into(balance_changes::table)
             .values::<&Vec<BalanceChangesInsertDb>>(
                 &balances

orm/src/blocks.rs

@@ -39,3 +39,18 @@ impl From<(Block, TendermintBlockResponse)> for BlockInsertDb {
         }
     }
 }
+
+impl BlockInsertDb {
+    pub fn fake(height: i32) -> Self {
+        Self {
+            height,
+            hash: height.to_string(), // fake hash but ensures uniqueness with height
+            app_hash: "fake_app_hash".to_string(), // doesn't require uniqueness
+            timestamp: chrono::DateTime::from_timestamp(0, 0)
+                .unwrap()
+                .naive_utc(),
+            proposer: "fake_proposer".to_string(),
+            epoch: 0,
+        }
+    }
+}

shared/src/balance.rs

@@ -110,7 +110,7 @@ impl Balance {
             owner: Id::Account(address.to_string()),
             token,
             amount: Amount::fake(),
-            height: (0..10000).fake::<u32>(),
+            height: 0,
         }
     }
 }