transpile: Add enums.c snapshot test

Author: Rua

c2rust-transpile/tests/snapshots.rs

@@ -334,6 +334,11 @@ fn test_empty_init() {
     transpile("empty_init.c").run();
 }
 
+#[test]
+fn test_enums() {
+    transpile("enums.c").run();
+}
+
 #[test]
 fn test_exprs() {
     transpile("exprs.c").run();

c2rust-transpile/tests/snapshots/enums.c

@@ -0,0 +1,35 @@
+typedef enum { Foo0, Foo1 } Foo;
+enum Bar { Bar1 = -1, Bar0 };
+
+void test_enums(void) {
+    Foo foo = Foo0;
+    enum Bar bar = Bar0;
+    foo = Foo1;
+    bar = Bar1;
+
+    // Assign enum constant of wrong type
+    foo = Bar0;
+    bar = Foo0;
+
+    // Assign integer that matches a constant
+    foo = 1;
+    bar = 1;
+
+    // Assign integer that doesn't match any constant
+    foo = 3;
+    bar = 3;
+
+    // Arithmetic
+    foo -= 2;
+    bar -= 2;
+
+    Foo e = Foo1;
+
+    // Compare to same type
+    int enum_enum = e == foo;
+    int enum_constant = e == Foo0;
+
+    // Compare to wrong type
+    int wrong_enum_enum = e == bar;
+    int wrong_enum_constant = e == Bar0;
+}

c2rust-transpile/tests/snapshots/snapshots__transpile@enums.c.2021.snap

@@ -0,0 +1,44 @@
+---
+source: c2rust-transpile/tests/snapshots.rs
+expression: cat tests/snapshots/enums.2021.rs
+---
+#![allow(
+    dead_code,
+    non_camel_case_types,
+    non_snake_case,
+    non_upper_case_globals,
+    unused_assignments,
+    unused_mut
+)]
+pub type Foo = ::core::ffi::c_uint;
+pub const Foo1: Foo = 1;
+pub const Foo0: Foo = 0;
+pub type Bar = ::core::ffi::c_int;
+pub const Bar0: Bar = 0;
+pub const Bar1: Bar = -1;
+#[no_mangle]
+pub unsafe extern "C" fn test_enums() {
+    let mut foo: Foo = Foo0;
+    let mut bar: Bar = Bar0;
+    foo = Foo1;
+    bar = Bar1;
+    foo = Bar0 as ::core::ffi::c_int as Foo;
+    bar = Foo0 as ::core::ffi::c_int as Bar;
+    foo = Foo1;
+    bar = 1 as Bar;
+    foo = 3 as Foo;
+    bar = 3 as Bar;
+    foo = (foo as ::core::ffi::c_uint).wrapping_sub(2 as ::core::ffi::c_uint) as Foo;
+    bar = (bar as ::core::ffi::c_int - 2 as ::core::ffi::c_int) as Bar;
+    let mut e: Foo = Foo1;
+    let mut enum_enum: ::core::ffi::c_int =
+        (e as ::core::ffi::c_uint == foo as ::core::ffi::c_uint) as ::core::ffi::c_int;
+    let mut enum_constant: ::core::ffi::c_int = (e as ::core::ffi::c_uint
+        == Foo0 as ::core::ffi::c_int as ::core::ffi::c_uint)
+        as ::core::ffi::c_int;
+    let mut wrong_enum_enum: ::core::ffi::c_int =
+        (e as ::core::ffi::c_uint == bar as ::core::ffi::c_uint) as ::core::ffi::c_int;
+    let mut wrong_enum_constant: ::core::ffi::c_int = (e as ::core::ffi::c_uint
+        == Bar0 as ::core::ffi::c_int as ::core::ffi::c_uint)
+        as ::core::ffi::c_int;
+}

c2rust-transpile/tests/snapshots/snapshots__transpile@enums.c.2024.snap

@@ -0,0 +1,45 @@
+---
+source: c2rust-transpile/tests/snapshots.rs
+expression: cat tests/snapshots/enums.2024.rs
+---
+#![allow(
+    dead_code,
+    non_camel_case_types,
+    non_snake_case,
+    non_upper_case_globals,
+    unsafe_op_in_unsafe_fn,
+    unused_assignments,
+    unused_mut
+)]
+pub type Foo = ::core::ffi::c_uint;
+pub const Foo1: Foo = 1;
+pub const Foo0: Foo = 0;
+pub type Bar = ::core::ffi::c_int;
+pub const Bar0: Bar = 0;
+pub const Bar1: Bar = -1;
+#[unsafe(no_mangle)]
+pub unsafe extern "C" fn test_enums() {
+    let mut foo: Foo = Foo0;
+    let mut bar: Bar = Bar0;
+    foo = Foo1;
+    bar = Bar1;
+    foo = Bar0 as ::core::ffi::c_int as Foo;
+    bar = Foo0 as ::core::ffi::c_int as Bar;
+    foo = Foo1;
+    bar = 1 as Bar;
+    foo = 3 as Foo;
+    bar = 3 as Bar;
+    foo = (foo as ::core::ffi::c_uint).wrapping_sub(2 as ::core::ffi::c_uint) as Foo;
+    bar = (bar as ::core::ffi::c_int - 2 as ::core::ffi::c_int) as Bar;
+    let mut e: Foo = Foo1;
+    let mut enum_enum: ::core::ffi::c_int =
+        (e as ::core::ffi::c_uint == foo as ::core::ffi::c_uint) as ::core::ffi::c_int;
+    let mut enum_constant: ::core::ffi::c_int = (e as ::core::ffi::c_uint
+        == Foo0 as ::core::ffi::c_int as ::core::ffi::c_uint)
+        as ::core::ffi::c_int;
+    let mut wrong_enum_enum: ::core::ffi::c_int =
+        (e as ::core::ffi::c_uint == bar as ::core::ffi::c_uint) as ::core::ffi::c_int;
+    let mut wrong_enum_constant: ::core::ffi::c_int = (e as ::core::ffi::c_uint
+        == Bar0 as ::core::ffi::c_int as ::core::ffi::c_uint)
+        as ::core::ffi::c_int;
+}