test: brc20 augment

hirosystems · Sep 17, 2024 · 2701cea · 2701cea
1 parent 0a26bf6
commit 2701cea
Show file tree

Hide file tree

Showing 3 changed files with 104 additions and 3 deletions.
diff --git a/components/ordhook-core/src/core/meta_protocols/brc20/db.rs b/components/ordhook-core/src/core/meta_protocols/brc20/db.rs
@@ -616,3 +616,88 @@ pub fn write_augmented_block_to_brc20_db(
     insert_token_rows(&tokens, db_conn, ctx);
     insert_ledger_rows(&ledger_rows, db_conn, ctx);
 }
+
+#[cfg(test)]
+mod test {
+    use chainhook_sdk::{
+        types::{
+            Brc20Operation, Brc20TokenDeployData, OrdinalInscriptionNumber,
+            OrdinalInscriptionRevealData, OrdinalOperation,
+        },
+        utils::Context,
+    };
+
+    use crate::{
+        config::Config,
+        core::test_builders::{TestBlockBuilder, TestTransactionBuilder},
+        db::{drop_all_dbs, initialize_sqlite_dbs},
+    };
+
+    use super::{get_token, write_augmented_block_to_brc20_db};
+
+    #[test]
+    fn writes_augmented_block_to_db() {
+        let ctx = Context::empty();
+        let mut config = Config::test_default();
+        config.meta_protocols.brc20 = true;
+        drop_all_dbs(&config);
+        let sqlite_dbs = initialize_sqlite_dbs(&config, &ctx);
+
+        let block = TestBlockBuilder::new()
+            .hash("0xb61b0172d95e266c18aea0c624db987e971a5d6d4ebc2aaed85da4642d635735".to_string())
+            .height(850000)
+            .add_transaction(TestTransactionBuilder::new().build())
+            .add_transaction(
+                TestTransactionBuilder::new()
+                    .hash("0xc62d436323e14cdcb91dd21cb7814fd1ac5b9ecb6e3cc6953b54c02a343f7ec9".to_string())
+                    .add_ordinal_operation(OrdinalOperation::InscriptionRevealed(OrdinalInscriptionRevealData {
+                        content_bytes: "0x7b200a20202270223a20226272632d3230222c0a2020226f70223a20226465706c6f79222c0a2020227469636b223a20226f726469222c0a2020226d6178223a20223231303030303030222c0a2020226c696d223a202231303030220a7d".to_string(),
+                        content_type: "text/plain;charset=utf-8".to_string(),
+                        content_length: 94,
+                        inscription_number: OrdinalInscriptionNumber { classic: 0, jubilee: 0 },
+                        inscription_fee: 0,
+                        inscription_output_value: 0,
+                        inscription_id: "b61b0172d95e266c18aea0c624db987e971a5d6d4ebc2aaed85da4642d635735i0".to_string(),
+                        inscription_input_index: 0,
+                        inscription_pointer: None,
+                        inscriber_address: None,
+                        delegate: None,
+                        metaprotocol: None,
+                        metadata: None,
+                        parent: None,
+                        ordinal_number: 0,
+                        ordinal_block_height: 0,
+                        ordinal_offset: 0,
+                        tx_index: 0,
+                        transfers_pre_inscription: 0,
+                        satpoint_post_inscription: "".to_string(),
+                        curse_type: None,
+                    }))
+                    .brc20_operation(Some(Brc20Operation::Deploy(Brc20TokenDeployData {
+                        tick: "ordi".to_string(), 
+                        max: "21000".to_string(), 
+                        lim: "1000".to_string(), 
+                        dec: "0".to_string(), 
+                        address: "3K9KZZPB8NRwZVP5wNKX4VYhnswrJxpgZ4".to_string(), 
+                        inscription_id: "b61b0172d95e266c18aea0c624db987e971a5d6d4ebc2aaed85da4642d635735i0".to_string(), 
+                        self_mint: false
+                    })))
+                    .build()
+                )
+            .build();
+        write_augmented_block_to_brc20_db(&block, sqlite_dbs.brc20.as_ref().unwrap(), &ctx);
+
+        let deploy = get_token("ordi", sqlite_dbs.brc20.as_ref().unwrap(), &ctx).unwrap();
+        assert_eq!(
+            deploy.inscription_id,
+            "b61b0172d95e266c18aea0c624db987e971a5d6d4ebc2aaed85da4642d635735i0"
+        );
+        assert_eq!(deploy.tick, "ordi");
+        assert_eq!(deploy.display_tick, "ordi");
+        assert_eq!(deploy.max, 21000.0);
+        assert_eq!(deploy.lim, 1000.0);
+        assert_eq!(deploy.dec, 0);
+        assert_eq!(deploy.inscription_number, 0);
+        assert_eq!(deploy.address, "3K9KZZPB8NRwZVP5wNKX4VYhnswrJxpgZ4");
+    }
+}
diff --git a/components/ordhook-core/src/core/pipeline/processors/block_archiving.rs b/components/ordhook-core/src/core/pipeline/processors/block_archiving.rs
@@ -105,7 +105,11 @@ mod test {
             pipeline::PostProcessorCommand,
             test_builders::{TestBlockBuilder, TestTransactionBuilder},
         },
-        db::{blocks::{find_block_bytes_at_block_height, open_blocks_db_with_retry}, cursor::BlockBytesCursor, drop_all_dbs, initialize_sqlite_dbs},
+        db::{
+            blocks::{find_block_bytes_at_block_height, open_blocks_db_with_retry},
+            cursor::BlockBytesCursor,
+            drop_all_dbs, initialize_sqlite_dbs,
+        },
     };
 
     use super::start_block_archiving_processor;

diff --git a/components/ordhook-core/src/core/test_builders.rs b/components/ordhook-core/src/core/test_builders.rs
@@ -1,7 +1,7 @@
 use chainhook_sdk::types::{
     bitcoin::{OutPoint, TxIn, TxOut},
     BitcoinBlockData, BitcoinBlockMetadata, BitcoinNetwork, BitcoinTransactionData,
-    BitcoinTransactionMetadata, BlockIdentifier, OrdinalInscriptionNumber,
+    BitcoinTransactionMetadata, BlockIdentifier, Brc20Operation, OrdinalInscriptionNumber,
     OrdinalInscriptionRevealData, OrdinalOperation, TransactionIdentifier,
 };
 
@@ -65,6 +65,7 @@ pub struct TestTransactionBuilder {
     inputs: Vec<TxIn>,
     outputs: Vec<TxOut>,
     ordinal_operations: Vec<OrdinalOperation>,
+    brc20_operation: Option<Brc20Operation>,
 }
 
 impl TestTransactionBuilder {
@@ -74,6 +75,7 @@ impl TestTransactionBuilder {
             ordinal_operations: vec![],
             inputs: vec![],
             outputs: vec![],
+            brc20_operation: None,
         }
     }
 
@@ -137,6 +139,16 @@ impl TestTransactionBuilder {
         self
     }
 
+    pub fn add_ordinal_operation(mut self, ordinal_operation: OrdinalOperation) -> Self {
+        self.ordinal_operations.push(ordinal_operation);
+        self
+    }
+
+    pub fn brc20_operation(mut self, brc20_operation: Option<Brc20Operation>) -> Self {
+        self.brc20_operation = brc20_operation;
+        self
+    }
+
     pub fn build(self) -> BitcoinTransactionData {
         BitcoinTransactionData {
             transaction_identifier: TransactionIdentifier { hash: self.hash },
@@ -146,7 +158,7 @@ impl TestTransactionBuilder {
                 outputs: self.outputs,
                 ordinal_operations: self.ordinal_operations,
                 stacks_operations: vec![],
-                brc20_operation: None,
+                brc20_operation: self.brc20_operation,
                 proof: None,
                 fee: 0,
                 index: 0,