Merge pull request #29 from osmosis-labs/asset-removal

Corrupted Assets
osmosis-labs · Mar 28, 2024 · fcf45f1 · fcf45f1
2 parents 47bbb02 + c720190
commit fcf45f1
Show file tree

Hide file tree

Showing 15 changed files with 1,295 additions and 25 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/contracts/transmuter/Cargo.toml b/contracts/transmuter/Cargo.toml
@@ -55,5 +55,5 @@ thiserror = {version = "1.0.44"}
 
 [dev-dependencies]
 itertools = "0.12.0"
-osmosis-test-tube = "22.0.0"
+osmosis-test-tube = "22.1.0"
 rstest = "0.18.2"
diff --git a/contracts/transmuter/src/asset.rs b/contracts/transmuter/src/asset.rs
@@ -43,6 +43,7 @@ impl AssetConfig {
             amount: Uint128::zero(),
             denom: self.denom,
             normalization_factor: self.normalization_factor,
+            is_corrupted: false,
         })
     }
 }
@@ -52,6 +53,7 @@ pub struct Asset {
     amount: Uint128,
     denom: String,
     normalization_factor: Uint128,
+    is_corrupted: bool,
 }
 
 impl Asset {
@@ -64,6 +66,7 @@ impl Asset {
             amount: amount.into(),
             denom: denom.to_string(),
             normalization_factor: normalization_factor.into(),
+            is_corrupted: false,
         }
     }
 
@@ -112,6 +115,16 @@ impl Asset {
         Ok(self)
     }
 
+    pub fn mark_as_corrupted(&'_ mut self) -> &'_ Self {
+        self.is_corrupted = true;
+        self
+    }
+
+    pub fn unmark_as_corrupted(&'_ mut self) -> &'_ Self {
+        self.is_corrupted = false;
+        self
+    }
+
     pub fn denom(&self) -> &str {
         &self.denom
     }
@@ -124,6 +137,10 @@ impl Asset {
         self.normalization_factor
     }
 
+    pub fn is_corrupted(&self) -> bool {
+        self.is_corrupted
+    }
+
     pub fn config(&self) -> AssetConfig {
         AssetConfig {
             denom: self.denom.clone(),
@@ -144,6 +161,7 @@ impl Asset {
             amount,
             denom: denom.to_string(),
             normalization_factor,
+            is_corrupted: false,
         }
     }
 
@@ -350,6 +368,7 @@ mod tests {
                 amount: Uint128::zero(),
                 denom: "denom1".to_string(),
                 normalization_factor: Uint128::one(),
+                is_corrupted: false,
             }
         );
 
@@ -364,6 +383,7 @@ mod tests {
                 amount: Uint128::zero(),
                 denom: "denom2".to_string(),
                 normalization_factor: Uint128::from(1000000u128),
+                is_corrupted: false,
             }
         );
 
@@ -383,6 +403,7 @@ mod tests {
             amount: Uint128::zero(),
             denom: "denom1".to_string(),
             normalization_factor: Uint128::one(),
+            is_corrupted: false,
         };
 
         assert_eq!(
@@ -398,4 +419,19 @@ mod tests {
             ContractError::NormalizationFactorMustBePositive {}
         );
     }
+
+    #[test]
+    fn test_mark_as_corrupted() {
+        let mut asset = Asset {
+            amount: Uint128::zero(),
+            denom: "denom1".to_string(),
+            normalization_factor: Uint128::one(),
+            is_corrupted: false,
+        };
+
+        assert_eq!(asset.is_corrupted(), false);
+
+        assert_eq!(asset.mark_as_corrupted().is_corrupted(), true);
+        assert_eq!(asset.is_corrupted(), true);
+    }
 }