add unit tests
git-svn-id: https://llvm.org/svn/llvm-project/compiler-rt/trunk@110278 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/BlocksRuntime/tests/byrefcopycopy.c b/BlocksRuntime/tests/byrefcopycopy.c
new file mode 100644
index 0000000..d6fafc1
--- /dev/null
+++ b/BlocksRuntime/tests/byrefcopycopy.c
@@ -0,0 +1,46 @@
+//
+// The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+
+// CONFIG rdar://6255170
+
+#include <stdio.h>
+#include <stdbool.h>
+#include <stdlib.h>
+#include <Block.h>
+#include <Block_private.h>
+#include <assert.h>
+
+
+int
+main(int argc, char *argv[])
+{
+ __block int var = 0;
+ int shouldbe = 0;
+ void (^b)(void) = ^{ var++; /*printf("var is at %p with value %d\n", &var, var);*/ };
+ __typeof(b) _b;
+ //printf("before copy...\n");
+ b(); ++shouldbe;
+ size_t i;
+
+ for (i = 0; i < 10; i++) {
+ _b = Block_copy(b); // make a new copy each time
+ assert(_b);
+ ++shouldbe;
+ _b(); // should still update the stack
+ Block_release(_b);
+ }
+
+ //printf("after...\n");
+ b(); ++shouldbe;
+
+ if (var != shouldbe) {
+ printf("Hmm, var is %d but should be %d\n", var, shouldbe);
+ return 1;
+ }
+ printf("%s: Success!!\n", argv[0]);
+
+ return 0;
+}