ccan/oserver/test/run.c

   1 #include <ccan/failtest/failtest_override.h>
   2 #include <ccan/oserver/oserver.c>
   3 #include <ccan/oserver/oserver.h>
   4 #include <ccan/str/str.h>
   5 #include <ccan/foreach/foreach.h>
   6 #include <ccan/tap/tap.h>
   7 #include <sys/types.h>
   8 #include <sys/stat.h>
   9 #include <ccan/failtest/failtest.h>
  10 #include <fcntl.h>
  11 #include <string.h>
  12 #include <ccan/failtest/failtest_undo.h>
  13
  14 static void exit_test(void)
  15 {
  16         failtest_exit(exit_status());
  17 }
  18
  19 int main(int argc, char *argv[])
  20 {
  21         int fd;
  22         char buf[200];
  23         const char *input;
  24
  25         /* This is how many tests you plan to run */
  26         plan_tests(3 * 6);
  27         failtest_init(argc, argv);
  28         tap_fail_callback = exit_test;
  29
  30         foreach_ptr(input,
  31                     "This is a test\n",
  32                     "This is a test\r",
  33                     "This is a test\r\n",
  34                     "This is a test\nWith extra",
  35                     "This is a test\rWith extra",
  36                     "This is a test\r\nWith extra") {
  37                 fd = open("run-fd", O_RDWR|O_CREAT|O_TRUNC, 0600);
  38
  39                 write(fd, input, strlen(input));
  40                 lseek(fd, 0, SEEK_SET);
  41
  42                 ok1(oserver_serve(fd));
  43
  44                 lseek(fd, 0, SEEK_SET);
  45                 buf[read(fd, buf, sizeof(buf)-1)] = '\0';
  46
  47                 ok1(strncmp(buf, input, strlen("This is a test")) == 0);
  48                 ok1(streq(buf + strlen("This is a test") + 1,
  49                           "Louder, like this: 'THIS IS A TEST'\r\n"));
  50         }
  51
  52         /* This exits depending on whether all tests passed */
  53         failtest_exit(exit_status());
  54 }