libpthread-454.40.3.tar.gz

[apple/libpthread.git] / tests / tsd.c

diff --git a/tests/tsd.c b/tests/tsd.c
index 998d75d63c99b840423f67cf619cfff4013332b6..d20dcf156db087695ffd019c9d157897d52732dc 100644 (file)
--- a/tests/tsd.c
+++ b/tests/tsd.c
@@ -1,55 +1,69 @@
-#include <assert.h>
 #include <pthread.h>
 #include <stdio.h>
+#include <sys/sysctl.h>
 
-void *ptr = NULL;
+#include "darwintest_defaults.h"
 
-void destructor(void *value)
+static void *ptr = NULL;
+
+static void destructor(void *value)
 {
        ptr = value;
 }
 
-void *thread(void *param)
+static void *thread(void *param)
 {
-       int res;
-
        pthread_key_t key = *(pthread_key_t *)param;
-       res = pthread_setspecific(key, (void *)0x12345678);
-       assert(res == 0);
+       T_ASSERT_POSIX_ZERO(pthread_setspecific(key, (void *)0x12345678), NULL);
        void *value = pthread_getspecific(key);
 
-       pthread_key_t key2;
-       res = pthread_key_create(&key, NULL);
-       assert(res == 0);
-       res = pthread_setspecific(key, (void *)0x55555555);
-       assert(res == 0);
+       T_ASSERT_POSIX_ZERO(pthread_key_create(&key, NULL), NULL);
+       T_ASSERT_POSIX_ZERO(pthread_setspecific(key, (void *)0x55555555), NULL);
 
        return value;
 }
 
-int main(int argc, char *argv[])
+T_DECL(tsd, "tsd",
+       T_META_ALL_VALID_ARCHS(YES))
 {
-       int res;
        pthread_key_t key;
 
-       res = pthread_key_create(&key, destructor);
-       assert(res == 0);
-       printf("key = %ld\n", key);
+       T_ASSERT_POSIX_ZERO(pthread_key_create(&key, destructor), NULL);
+       T_LOG("key = %ld", key);
 
        pthread_t p = NULL;
-       res = pthread_create(&p, NULL, thread, &key);
-       assert(res == 0);
+       T_ASSERT_POSIX_ZERO(pthread_create(&p, NULL, thread, &key), NULL);
 
        void *value = NULL;
-       res = pthread_join(p, &value);
-       printf("value = %p\n", value);
-       printf("ptr = %p\n", ptr);
+       T_ASSERT_POSIX_ZERO(pthread_join(p, &value), NULL);
+       T_LOG("value = %p; ptr = %p\n", value, ptr);
 
-       assert(ptr == value);
+       T_EXPECT_EQ(ptr, value, NULL);
 
-       res = pthread_key_delete(key);
-       assert(res == 0);
+       T_ASSERT_POSIX_ZERO(pthread_key_delete(key), NULL);
+}
 
-       return 0;
+static uint32_t
+get_ncpu(void)
+{
+       static uint32_t activecpu;
+       if (!activecpu) {
+               uint32_t n;
+               size_t s = sizeof(activecpu);
+               sysctlbyname("hw.activecpu", &n, &s, NULL, 0);
+               activecpu = n;
+       }
+       return activecpu;
 }
 
+T_DECL(cpuid, "cpu id", T_META_ALL_VALID_ARCHS(YES))
+{
+       pthread_t child;
+
+       size_t cpu_id;
+       if (pthread_cpu_number_np(&cpu_id)) {
+               T_FAIL("Should not fail to get CPU id");
+       }
+
+       T_ASSERT_LE(cpu_id, get_ncpu(), "Got a valid CPU id");
+}