Refactor; defer with fblocks

1 files changed, 37 insertions, 0 deletions

diff --git a/source/tests/defer.test.c b/source/tests/defer.test.c
new file mode 100644
index 0000000..0eb8557
--- /dev/null
+++ b/source/tests/defer.test.c
@@ -0,0 +1,37 @@
+#if defined(__clang__) || defined(__APPLE__)
+#  include "../kit/defer.h"
+
+#  define KIT_TEST_FILE defer_block
+#  include "../kit/test.h"
+
+TEST("defer") {
+  int defer_ref x = 1;
+
+  {
+    defer {
+      x = 2;
+    };
+
+    x = 3;
+  }
+
+  REQUIRE_EQ(x, 2);
+}
+
+TEST("defer capture") {
+  int defer_ref x = 1;
+
+  {
+    defer {
+      if (x == 3)
+        x = 2;
+    };
+
+    x = 3;
+  }
+
+  REQUIRE_EQ(x, 2);
+}
+
+#  undef KIT_TEST_FILE
+#endif