cli support

Author: rahxephon89

aptos-move/move-examples/cli-e2e-tests/common/sources/cli_e2e_tests.move

@@ -81,6 +81,21 @@ module addr::cli_e2e_tests {
         name: String,
     }
 
+    /// A public enum for testing enum arguments - simple variants
+    public enum Color has copy, drop {
+        Red,
+        Green,
+        Blue,
+        RGB { r: u8, g: u8, b: u8 },
+    }
+
+    /// A public enum with struct-containing variants
+    public enum Shape has copy, drop {
+        Circle { center: Point, radius: u64 },
+        Rectangle { rect: Rectangle },
+        Point { point: Point },
+    }
+
     fun init_module(account: &signer) {
         let collection = string::utf8(b"Hero Quest");
         collection::create_unlimited_collection(
@@ -389,7 +404,7 @@ module addr::cli_e2e_tests {
     /// Test entry function that takes a Point argument
     public entry fun test_struct_point(_account: &signer, p: Point) {
         // Verify the point values are valid
-        assert!(p.x >= 0 && p.y >= 0, 100);
+        assert!(p.x > 0 && p.y > 0, 100);
     }
 
     /// Test entry function that takes a Rectangle argument
@@ -451,4 +466,68 @@ module addr::cli_e2e_tests {
         // Verify name
         assert!(string::bytes(&data.name) == &b"test_data", 116);
     }
+
+    // Test entry functions for enum transaction arguments
+
+    /// Test entry function that takes a simple enum variant (no fields)
+    public entry fun test_enum_color_simple(_account: &signer, color: Color) {
+        // Verify it's the Red variant
+        assert!(color is Color::Red, 120);
+    }
+
+    /// Test entry function that takes an enum variant with fields
+    public entry fun test_enum_color_rgb(_account: &signer, color: Color) {
+        // Verify it's the RGB variant with expected values
+        assert!(color is Color::RGB, 121);
+        match (color) {
+            Color::RGB { r, g, b } => {
+                assert!(r == 255, 122);
+                assert!(g == 128, 123);
+                assert!(b == 0, 124);
+            },
+            _ => abort 125,
+        }
+    }
+
+    /// Test entry function that takes a Shape enum with Point variant
+    public entry fun test_enum_shape_point(_account: &signer, shape: Shape) {
+        // Verify it's the Point variant
+        assert!(shape is Shape::Point, 126);
+        match (shape) {
+            Shape::Point { point } => {
+                assert!(point.x == 100, 127);
+                assert!(point.y == 200, 128);
+            },
+            _ => abort 129,
+        }
+    }
+
+    /// Test entry function that takes a Shape enum with Circle variant
+    public entry fun test_enum_shape_circle(_account: &signer, shape: Shape) {
+        // Verify it's the Circle variant
+        assert!(shape is Shape::Circle, 130);
+        match (shape) {
+            Shape::Circle { center, radius } => {
+                assert!(center.x == 50, 131);
+                assert!(center.y == 50, 132);
+                assert!(radius == 25, 133);
+            },
+            _ => abort 134,
+        }
+    }
+
+    /// Test entry function with mixed primitive and enum arguments
+    public entry fun test_mixed_with_enum(_account: &signer, num: u64, color: Color, p: Point) {
+        assert!(num == 999, 135);
+        assert!(color is Color::Green, 136);
+        assert!(p.x == 10 && p.y == 20, 137);
+    }
+
+    /// Test entry function with Option<Color> enum
+    public entry fun test_option_enum(_account: &signer, opt: Option<Color>) {
+        // Verify it's Some and contains Green variant
+        assert!(option::is_some(&opt), 138);
+        let color = option::destroy_some(opt);
+        assert!(color is Color::Green, 139);
+    }
 }