1 files changed, 212 insertions, 0 deletions
diff --git a/tests/test_switch.sh b/tests/test_switch.sh
new file mode 100644
index 0000000..d5d3dae
--- /dev/null
+++ b/tests/test_switch.sh
@@ -0,0 +1,212 @@
+#!/bin/bash
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 2;
+    int result = 0;
+
+    switch (x) {
+    case 1:
+        result = 10;
+        break;
+    case 2:
+        result = 20;
+        break;
+    case 3:
+        result = 30;
+        break;
+    }
+
+    ASSERT_EQ(20, result);
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 5;
+    int result = 0;
+
+    switch (x) {
+    case 1:
+        result = 10;
+        break;
+    case 2:
+        result = 20;
+        break;
+    default:
+        result = 99;
+        break;
+    }
+
+    ASSERT_EQ(99, result);
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 2;
+    int result = 0;
+
+    switch (x) {
+    case 1:
+        result = result + 10;
+    case 2:
+        result = result + 20;
+    case 3:
+        result = result + 30;
+        break;
+    }
+
+    ASSERT_EQ(50, result);  // 20 + 30 due to fall-through
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 1;
+    int y = 2;
+    int result = 0;
+
+    switch (x) {
+    case 1:
+        switch (y) {
+        case 1:
+            result = 11;
+            break;
+        case 2:
+            result = 12;
+            break;
+        }
+        break;
+    case 2:
+        result = 20;
+        break;
+    }
+
+    ASSERT_EQ(12, result);
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int a = 3;
+    int b = 2;
+    int result = 0;
+
+    switch (a + b) {
+    case 4:
+        result = 40;
+        break;
+    case 5:
+        result = 50;
+        break;
+    case 6:
+        result = 60;
+        break;
+    }
+
+    ASSERT_EQ(50, result);
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 2;
+    int result = 0;
+    int temp = 0;
+
+    switch (x) {
+    case 1:
+        temp = 5;
+        result = temp * 2;
+        break;
+    case 2:
+        temp = 10;
+        result = temp * 2;
+        break;
+    case 3:
+        temp = 15;
+        result = temp * 2;
+        break;
+    }
+
+    ASSERT_EQ(20, result);
+    ASSERT_EQ(10, temp);
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 1;
+    int result = 0;
+
+    switch (x) {
+    case 1: {
+        int local = 100;
+        result = local;
+        break;
+    }
+    case 2: {
+        int local = 200;
+        result = local;
+        break;
+    }
+    }
+
+    ASSERT_EQ(100, result);
+}
+EOF
+
+test_exit_code 0 <<'EOF'
+#include "../../helpers.h"
+
+int main() {
+    int x = 10;
+    int result = 42;
+
+    switch (x) {
+    case 1:
+        result = 10;
+        break;
+    case 2:
+        result = 20;
+        break;
+    }
+
+    ASSERT_EQ(42, result);
+}
+EOF
+
+cat <<'EOF' > expected
+main.c:2: 'case' label not within a switch statement
+EOF
+test_compile_error <<'EOF'
+int main() {
+    case 1:
+        return 0;
+}
+EOF
+
+cat <<'EOF' > expected
+main.c:2: 'default' label not within a switch statement
+EOF
+test_compile_error <<'EOF'
+int main() {
+    default:
+        return 0;
+}
+EOF