Add codegen tests

Signed-off-by: Alex Saveau <[email protected]>

Author: SUPERCILEX

tests/codegen/string_reserve_push.rs

@@ -0,0 +1,61 @@
+// compile-flags: -O
+
+#![crate_type = "lib"]
+
+use std::arch::asm;
+
+#[no_mangle]
+// CHECK-LABEL: @reserve_push_str(
+pub fn reserve_push_str(v: &mut String, s: &str) {
+    // CHECK: do_reserve_and_handle
+    // CHECK: "nop"
+    // CHECK-NOT: do_reserve_and_handle
+    // CHECK: ret
+    v.reserve(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.push_str(s);
+}
+
+#[no_mangle]
+// CHECK-LABEL: @reserve_exact_push_str(
+pub fn reserve_exact_push_str(v: &mut String, s: &str) {
+    // CHECK: finish_grow
+    // CHECK: "nop"
+    // CHECK-NOT: finish_grow
+    // CHECK: ret
+    v.reserve_exact(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.push_str(s);
+}
+
+#[no_mangle]
+// CHECK-LABEL: @try_reserve_push_str(
+pub fn try_reserve_push_str(v: &mut String, s: &str) {
+    // CHECK: finish_grow
+    // CHECK: "nop"
+    // CHECK-NOT: finish_grow
+    // CHECK: ret
+    v.try_reserve(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.push_str(s);
+}
+
+#[no_mangle]
+// CHECK-LABEL: @try_reserve_exact_push_str(
+pub fn try_reserve_exact_push_str(v: &mut String, s: &str) {
+    // CHECK: finish_grow
+    // CHECK: "nop"
+    // CHECK-NOT: finish_grow
+    // CHECK: ret
+    v.try_reserve_exact(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.push_str(s);
+}

tests/codegen/vec_reserve_extend.rs

@@ -0,0 +1,61 @@
+// compile-flags: -O
+
+#![crate_type = "lib"]
+
+use std::arch::asm;
+
+#[no_mangle]
+// CHECK-LABEL: @reserve_extend(
+pub fn reserve_extend(v: &mut Vec<u8>, s: &[u8]) {
+    // CHECK: do_reserve_and_handle
+    // CHECK: "nop"
+    // CHECK-NOT: do_reserve_and_handle
+    // CHECK: ret
+    v.reserve(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.extend_from_slice(s);
+}
+
+#[no_mangle]
+// CHECK-LABEL: @reserve_exact_extend(
+pub fn reserve_exact_extend(v: &mut Vec<u8>, s: &[u8]) {
+    // CHECK: finish_grow
+    // CHECK: "nop"
+    // CHECK-NOT: finish_grow
+    // CHECK: ret
+    v.reserve_exact(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.extend_from_slice(s);
+}
+
+#[no_mangle]
+// CHECK-LABEL: @try_reserve_extend(
+pub fn try_reserve_extend(v: &mut Vec<u8>, s: &[u8]) {
+    // CHECK: finish_grow
+    // CHECK: "nop"
+    // CHECK-NOT: finish_grow
+    // CHECK: ret
+    v.try_reserve(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.extend_from_slice(s);
+}
+
+#[no_mangle]
+// CHECK-LABEL: @try_reserve_exact_extend(
+pub fn try_reserve_exact_extend(v: &mut Vec<u8>, s: &[u8]) {
+    // CHECK: finish_grow
+    // CHECK: "nop"
+    // CHECK-NOT: finish_grow
+    // CHECK: ret
+    v.try_reserve_exact(s.len());
+    unsafe {
+        asm!("nop"); // Used to make the check logic easier
+    }
+    v.extend_from_slice(s);
+}