test(instructions): add arithmetic operations test

Author: avrabe

wrt-host/Cargo.toml

@@ -23,6 +23,7 @@ parking_lot = { version = "0.12", optional = true }
 default = ["std"]
 std = ["log", "parking_lot", "wrt-error/std", "wrt-types/std", "wrt-intercept/std"]
 no_std = ["wrt-error/no_std", "wrt-types/no_std", "wrt-intercept/no_std"]
+alloc = ["wrt-error/alloc", "wrt-types/alloc", "wrt-intercept/alloc"]
 kani = ["wrt-intercept/kani"]
 
 [lints.rust]

wrt-instructions/Cargo.toml

@@ -17,6 +17,7 @@ no_std = [
     "wrt-error/no_std", 
     "wrt-types/no_std",
     "wrt-runtime/no_std",
+    "alloc",
 ]
 optimize = []  # Optional feature to bypass some safety checks in performance-critical paths
 safety = []    # Additional safety features for ASIL-B compliance

wrt-instructions/src/arithmetic_test.rs

@@ -0,0 +1,39 @@
+use crate::{
+    arithmetic_ops::{ArithmeticContext, ArithmeticOp},
+    instruction_traits::PureInstruction,
+    Error, Value,
+};
+
+struct SimpleContext {
+    stack: Vec<Value>,
+}
+
+impl SimpleContext {
+    fn new() -> Self {
+        Self { stack: Vec::new() }
+    }
+}
+
+impl ArithmeticContext for SimpleContext {
+    fn push_arithmetic_value(&mut self, value: Value) -> crate::Result<()> {
+        self.stack.push(value);
+        Ok(())
+    }
+
+    fn pop_arithmetic_value(&mut self) -> crate::Result<Value> {
+        self.stack
+            .pop()
+            .ok_or_else(|| Error::new(wrt_error::kinds::StackUnderflow))
+    }
+}
+
+#[test]
+fn test_i32_add() {
+    let mut context = SimpleContext::new();
+
+    // Test i32.add
+    context.push_arithmetic_value(Value::I32(2)).unwrap();
+    context.push_arithmetic_value(Value::I32(3)).unwrap();
+    ArithmeticOp::I32Add.execute(&mut context).unwrap();
+    assert_eq!(context.pop_arithmetic_value().unwrap(), Value::I32(5));
+}