enums support member functions

2 files changed, 31 insertions, 0 deletions

diff --git a/test/cases/enum_with_members.zig b/test/cases/enum_with_members.zig
new file mode 100644
index 0000000000..cff5883b19
--- /dev/null
+++ b/test/cases/enum_with_members.zig
@@ -0,0 +1,30 @@
+const std = @import("std");
+const assert = std.debug.assert;
+const io = std.io;
+const str = std.str;
+
+enum ET {
+    SINT: i32,
+    UINT: u32,
+
+    pub fn print(a: &ET, buf: []u8) -> %usize {
+        return switch (*a) {
+            SINT => |x| { io.bufPrintInt(i32, buf, x) },
+            UINT => |x| { io.bufPrintInt(u32, buf, x) },
+        }
+    }
+}
+
+#attribute("test")
+fn enumWithMembers() {
+    const a = ET.SINT { -42 };
+    const b = ET.UINT { 42 };
+    var buf: [20]u8 = undefined;
+
+    assert(%%a.print(buf) == 3);
+    assert(str.eql(buf[0...3], "-42"));
+
+    assert(%%b.print(buf) == 2);
+    assert(str.eql(buf[0...2], "42"));
+}
+
diff --git a/test/self_hosted.zig b/test/self_hosted.zig
index 7b58684b05..80920758a4 100644
--- a/test/self_hosted.zig
+++ b/test/self_hosted.zig
@@ -13,6 +13,7 @@ const test_var_params = @import("cases/var_params.zig");
 const test_const_slice_child = @import("cases/const_slice_child.zig");
 const test_switch_prong_implicit_cast = @import("cases/switch_prong_implicit_cast.zig");
 const test_switch_prong_err_enum = @import("cases/switch_prong_err_enum.zig");
+const test_enum_with_members = @import("cases/enum_with_members.zig");
 
 // normal comment
 /// this is a documentation comment