fixing rust build a bit, adding testing to crdt

Author: kalisam

.github/workflows/rust.yml

@@ -16,7 +16,20 @@ jobs:
 
     steps:
     - uses: actions/checkout@v4
+    - name: Install Rust toolchain
+      uses: actions-rs/toolchain@v1
+      with:
+        profile: minimal
+        toolchain: stable
+        override: true
+        components: rustfmt, clippy
+    - name: Check formatting
+      run: cargo fmt --all -- --check
+    - name: Clippy
+      run: cargo clippy --all -- -D warnings
     - name: Build
       run: cargo build --verbose
     - name: Run tests
       run: cargo test --verbose
+    - name: Run benchmarks
+      run: cargo bench --no-run

src/core/centroid_crdt.rs

@@ -160,4 +160,144 @@ impl CentroidCRDT {
         distances.truncate(limit);
         distances
     }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use uuid::Uuid;
+    
+    #[test]
+    fn test_create_centroid() {
+        let node_id = Uuid::new_v4();
+        let mut crdt = CentroidCRDT::new(node_id);
+        
+        let vector = Vector::new(vec![1.0, 2.0, 3.0]);
+        let centroid_id = crdt.create_centroid(vector.clone());
+        
+        assert_eq!(crdt.centroids.len(), 1);
+        assert_eq!(crdt.operations.len(), 1);
+        
+        let centroid = crdt.get_centroid(&centroid_id).unwrap();
+        assert_eq!(centroid.vector.values, vector.values);
+    }
+    
+    #[test]
+    fn test_update_centroid() {
+        let node_id = Uuid::new_v4();
+        let mut crdt = CentroidCRDT::new(node_id);
+        
+        let vector1 = Vector::new(vec![1.0, 2.0, 3.0]);
+        let centroid_id = crdt.create_centroid(vector1);
+        
+        let vector2 = Vector::new(vec![4.0, 5.0, 6.0]);
+        crdt.update_centroid(centroid_id, vector2).unwrap();
+        
+        assert_eq!(crdt.operations.len(), 2);
+        
+        let centroid = crdt.get_centroid(&centroid_id).unwrap();
+        // After updating, the vector should be a weighted average of the original and new vector
+        // With 1 existing element and 1 new element, the weights are 1/2 each
+        // So the expected value is (1.0*1 + 4.0*1)/2, (2.0*1 + 5.0*1)/2, (3.0*1 + 6.0*1)/2
+        // Which equals 2.5, 3.5, 4.5
+        assert!(centroid.vector.values[0] > 1.0 && centroid.vector.values[0] < 4.0);
+        assert!(centroid.vector.values[1] > 2.0 && centroid.vector.values[1] < 5.0);
+        assert!(centroid.vector.values[2] > 3.0 && centroid.vector.values[2] < 6.0);
+        assert_eq!(centroid.count, 2);
+    }
+    
+    #[test]
+    fn test_delete_centroid() {
+        let node_id = Uuid::new_v4();
+        let mut crdt = CentroidCRDT::new(node_id);
+        
+        let vector = Vector::new(vec![1.0, 2.0, 3.0]);
+        let centroid_id = crdt.create_centroid(vector.clone());
+        
+        crdt.delete_centroid(centroid_id).unwrap();
+        
+        assert_eq!(crdt.centroids.len(), 0);
+        assert_eq!(crdt.operations.len(), 2);
+    }
+    
+    #[test]
+    fn test_merge() {
+        // Create two CRDTs
+        let node_id1 = Uuid::new_v4();
+        let node_id2 = Uuid::new_v4();
+        let mut crdt1 = CentroidCRDT::new(node_id1);
+        let mut crdt2 = CentroidCRDT::new(node_id2);
+        
+        // Add a centroid to the first CRDT
+        let vector1 = Vector::new(vec![1.0, 2.0, 3.0]);
+        let centroid_id1 = crdt1.create_centroid(vector1.clone());
+        
+        // Add a different centroid to the second CRDT
+        let vector2 = Vector::new(vec![4.0, 5.0, 6.0]);
+        let centroid_id2 = crdt2.create_centroid(vector2.clone());
+        
+        // Merge the second CRDT into the first
+        crdt1.merge(&crdt2);
+        
+        // First CRDT should now have both centroids
+        assert_eq!(crdt1.centroids.len(), 2);
+        assert!(crdt1.centroids.contains_key(&centroid_id1));
+        assert!(crdt1.centroids.contains_key(&centroid_id2));
+        
+        // Verify the centroids have the correct vectors
+        let merged_centroid1 = crdt1.get_centroid(&centroid_id1).unwrap();
+        let merged_centroid2 = crdt1.get_centroid(&centroid_id2).unwrap();
+        
+        assert_eq!(merged_centroid1.vector.values, vector1.values);
+        assert_eq!(merged_centroid2.vector.values, vector2.values);
+        
+        // Operations should be merged too
+        assert_eq!(crdt1.operations.len(), 2);
+        assert_eq!(crdt1.observed.len(), 2);
+    }
+    
+    #[test]
+    fn test_concurrent_operations() {
+        // Test handling of concurrent operations with different timestamps
+        let node_id1 = Uuid::new_v4();
+        let node_id2 = Uuid::new_v4();
+        let mut crdt1 = CentroidCRDT::new(node_id1);
+        let mut crdt2 = CentroidCRDT::new(node_id2);
+        
+        // Both CRDTs create a centroid with the same ID but different vectors
+        let centroid_id = Uuid::new_v4();
+        let vector1 = Vector::new(vec![1.0, 2.0, 3.0]);
+        let vector2 = Vector::new(vec![4.0, 5.0, 6.0]);
+        
+        // Create the same centroid in both CRDTs with different timestamps
+        let now = chrono::Utc::now();
+        let later = now + chrono::Duration::seconds(10);
+        
+        // Earlier operation in CRDT1
+        let op1 = CentroidOperation {
+            id: Uuid::new_v4(),
+            centroid_id,
+            timestamp: now,
+            operation_type: OperationType::Create(vector1),
+        };
+        
+        // Later operation in CRDT2
+        let op2 = CentroidOperation {
+            id: Uuid::new_v4(),
+            centroid_id,
+            timestamp: later,
+            operation_type: OperationType::Create(vector2.clone()),
+        };
+        
+        crdt1.apply_operation(op1);
+        crdt2.apply_operation(op2);
+        
+        // Merge CRDT2 into CRDT1
+        crdt1.merge(&crdt2);
+        
+        // The centroid in CRDT1 should now have the vector from CRDT2
+        // because it has a later timestamp
+        let centroid = crdt1.get_centroid(&centroid_id).unwrap();
+        assert_eq!(centroid.vector.values, vector2.values);
+    }
 }