fixed transaction module

Author: nagarajm22

lib/ain-ocean/src/data_acces/test/mod.rs

@@ -1 +1,2 @@
 mod oracle_test;
+mod transaction_test;

lib/ain-ocean/src/data_acces/test/transaction_test.rs

@@ -0,0 +1,99 @@
+#[cfg(test)]
+mod tests {
+    use crate::data_acces::transaction::TransactionVinDb;
+    use crate::data_acces::transaction_vout::TransactionVoutDb;
+    use crate::database::db_manger::{ColumnFamilyOperations, RocksDB};
+    use crate::model::transaction::{Transaction, TransactionBlock};
+    use crate::model::transaction_vout::{TransactionVout, TransactionVoutScript};
+    use tempfile::tempdir;
+    use tokio::task;
+
+    fn setup_test_db() -> TransactionVinDb {
+        let temp_dir = tempdir().unwrap();
+        let db = RocksDB::new(temp_dir.path().to_str().unwrap()).unwrap();
+        TransactionVinDb { db }
+    }
+
+    // Sample transaction for testing
+    fn sample_transaction(id: &str) -> Transaction {
+        Transaction {
+            id: id.to_string(),
+            order: 1,
+            block: TransactionBlock {
+                hash: "sample_hash".to_string(),
+                height: 123,
+                time: 1000,
+                median_time: 1000,
+            },
+            txid: id.to_string(),
+            hash: "sample_transaction_hash".to_string(),
+            version: 1,
+            size: 200,
+            v_size: 200,
+            weight: 300,
+            total_vout_value: "1000".to_string(),
+            lock_time: 0,
+            vin_count: 2,
+            vout_count: 2,
+        }
+    }
+
+    fn sample_transaction_Vout() {
+        // Instantiate a sample TransactionVoutScript
+        let sample_script = TransactionVoutScript {
+            hex: "76a91488ac".to_string(),    // Sample hex string
+            r#type: "pubkeyhash".to_string(), // Sample type
+        };
+
+        // Instantiate a sample TransactionVout
+        let sample_vout = TransactionVout {
+            id: "vout1".to_string(),     // Sample ID
+            txid: "tx12345".to_string(), // Sample transaction ID
+            n: 0,                        // Sample index
+            value: "0.0001".to_string(), // Sample value in BTC or similar currency
+            token_id: 123,               // Sample token ID
+            script: sample_script,       // Use the sample script created above
+        };
+    }
+
+    #[tokio::test]
+    async fn test_store_transaction() {
+        let txn_vin_db = setup_test_db();
+
+        let test_transaction = sample_transaction("tx1");
+
+        let result = txn_vin_db.store(test_transaction).await;
+        assert!(result.is_ok());
+    }
+
+    #[tokio::test]
+    async fn test_store_get_transaction() {
+        let txn_vin_db = setup_test_db();
+
+        let test_transaction = sample_transaction("tx1");
+        let trx_id = test_transaction.id.clone();
+
+        let result = txn_vin_db.store(test_transaction.clone()).await;
+        assert!(result.is_ok());
+
+        let result = txn_vin_db.get(trx_id).await.unwrap().unwrap();
+        assert_eq!(test_transaction, result);
+    }
+
+    #[tokio::test]
+    async fn test_store_get_delete_transaction() {
+        let txn_vin_db = setup_test_db();
+
+        let test_transaction = sample_transaction("tx1");
+        let trx_id = test_transaction.id.clone();
+
+        let result = txn_vin_db.store(test_transaction.clone()).await;
+        assert!(result.is_ok());
+
+        let result = txn_vin_db.get(trx_id.clone()).await.unwrap().unwrap();
+        assert_eq!(test_transaction, result);
+
+        let result = txn_vin_db.delete(trx_id).await;
+        assert!(result.is_ok());
+    }
+}

lib/ain-ocean/src/data_acces/transaction_vout.rs

@@ -5,11 +5,11 @@ use anyhow::{anyhow, Result};
 use serde::{Deserialize, Serialize};
 
 #[derive(Debug)]
-pub struct TransactionVinDb {
+pub struct TransactionVoutDb {
     pub db: RocksDB,
 }
 
-impl TransactionVinDb {
+impl TransactionVoutDb {
     pub async fn get(&self, txid: String, n: i64) -> Result<Option<TransactionVout>> {
         match self.db.get("transaction_vout", txid.as_bytes()) {
             Ok(Some(value)) => {

lib/ain-ocean/src/model/transaction.rs

@@ -1,6 +1,6 @@
 use serde::{Deserialize, Serialize};
 
-#[derive(Serialize, Deserialize, Debug, Default)]
+#[derive(Serialize, Deserialize, Debug, Default, PartialEq, Clone)]
 #[serde(rename_all = "camelCase")]
 pub struct Transaction {
     pub id: String,
@@ -18,7 +18,7 @@ pub struct Transaction {
     pub vout_count: i32,
 }
 
-#[derive(Serialize, Deserialize, Debug, Default)]
+#[derive(Serialize, Deserialize, Debug, Default, PartialEq, Clone)]
 #[serde(rename_all = "camelCase")]
 pub struct TransactionBlock {
     pub hash: String,