libdispatch update

[platform/upstream/gcd.git] / dispatch-1.0 / testing / dispatch_read.c

diff --git a/dispatch-1.0/testing/dispatch_read.c b/dispatch-1.0/testing/dispatch_read.c
index 7d1a8e6..42c2a50 100644 (file)
--- a/dispatch-1.0/testing/dispatch_read.c
+++ b/dispatch-1.0/testing/dispatch_read.c
@@ -1,26 +1,25 @@
 /*
- * Copyright (c) 2008-2009 Apple Inc. All rights reserved.
+ * Copyright (c) 2008-2011 Apple Inc. All rights reserved.
  *
  * @APPLE_APACHE_LICENSE_HEADER_START@
- * 
+ *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *     http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
- * 
+ *
  * @APPLE_APACHE_LICENSE_HEADER_END@
  */
 
-#include "config/config.h"
-
 #include <sys/stat.h>
+#include <sys/types.h>
 #include <assert.h>
 #include <fcntl.h>
 #include <stdio.h>
@@ -30,18 +29,27 @@
 
 #include <dispatch/dispatch.h>
 
+#include <bsdtests.h>
 #include "dispatch_test.h"
 
 static size_t bytes_total;
 static size_t bytes_read;
 
-int main(void)
+void
+test_fin(void *cxt)
+{
+       test_ptr("test_fin run", cxt, cxt);
+       test_stop();
+}
+
+int
+main(void)
 {
        const char *path = "/usr/share/dict/words";
        struct stat sb;
 
-       test_start("Dispatch Source Read");
-       
+       dispatch_test_start("Dispatch Source Read");
+
        int infd = open(path, O_RDONLY);
        if (infd == -1) {
                perror(path);
@@ -58,39 +66,46 @@ int main(void)
                exit(EXIT_FAILURE);
        }
 
+       if (!dispatch_test_check_evfilt_read_for_fd(infd)) {
+               test_skip("EVFILT_READ kevent not firing for test file");
+               test_fin(NULL);
+       }
+
        dispatch_queue_t main_q = dispatch_get_main_queue();
        test_ptr_notnull("dispatch_get_main_queue", main_q);
 
-       dispatch_source_t reader;
-       
-       reader = dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, infd, 0, main_q);
-       test_ptr_notnull("DISPATCH_SOURCE_TYPE_READ", reader);
-       
+       dispatch_source_t reader = dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, infd, 0, main_q);
+       test_ptr_notnull("dispatch_source_create", reader);
+       assert(reader);
+
        dispatch_source_set_event_handler(reader, ^{
-                       size_t estimated = dispatch_source_get_data(reader);
-                       printf("bytes available: %zu\n", estimated);
-                       const ssize_t bufsiz = 1024*500; // 500 KB buffer
-                       static char buffer[1024*500];   // 500 KB buffer
-                       ssize_t actual = read(infd, buffer, sizeof(buffer));
+               size_t estimated = dispatch_source_get_data(reader);
+               fprintf(stderr, "bytes available: %zu\n", estimated);
+               test_double_less_than_or_equal("estimated", estimated, bytes_total - bytes_read);
+               const ssize_t bufsiz = 1024*500; // 500 KB buffer
+               static char buffer[1024*500];   // 500 KB buffer
+               ssize_t actual = read(infd, buffer, sizeof(buffer));
+               bytes_read += actual;
+               printf("bytes read: %zd\n", actual);
+               if (actual < bufsiz) {
+                       actual = read(infd, buffer, sizeof(buffer));
                        bytes_read += actual;
-                       printf("bytes read: %zd\n", actual);
-                       if (actual < bufsiz) {
-                               actual = read(infd, buffer, sizeof(buffer));
-                               bytes_read += actual;
-                               // confirm EOF condition
-                               test_long("EOF", actual, 0);
-                               dispatch_source_cancel(reader);
-                               dispatch_release(reader);
-                       }
+                       // confirm EOF condition
+                       test_long("EOF", actual, 0);
+                       dispatch_source_cancel(reader);
+               }
        });
-       
+
        dispatch_source_set_cancel_handler(reader, ^{
                test_long("Bytes read", bytes_read, bytes_total);
                int res = close(infd);
                test_errno("close", res == -1 ? errno : 0, 0);
-               test_stop();
+               dispatch_release(reader);
        });
 
+       dispatch_set_context(reader, reader);
+       dispatch_set_finalizer_f(reader, test_fin);
+
        dispatch_resume(reader);
 
        dispatch_main();