src/util/test_mq.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2012 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 /**
  22  * @file set/test_mq.c
  23  * @brief simple tests for mq
  24  */
  25 #include "platform.h"
  26 #include "gnunet_util_lib.h"
  27 #include "gnunet_testing_lib.h"
  28
  29
  30 GNUNET_NETWORK_STRUCT_BEGIN
  31
  32 struct MyMessage
  33 {
  34   struct GNUNET_MessageHeader header;
  35   uint32_t x GNUNET_PACKED;
  36 };
  37
  38 GNUNET_NETWORK_STRUCT_END
  39
  40 void
  41 test1 (void)
  42 {
  43   struct GNUNET_MQ_Message *mqm;
  44   struct MyMessage *mm;
  45
  46   mm = NULL;
  47   mqm = NULL;
  48
  49   mqm = GNUNET_MQ_msg (mm, 42);
  50   GNUNET_assert (NULL != mqm);
  51   GNUNET_assert (NULL != mm);
  52   GNUNET_assert (42 == ntohs (mm->header.type));
  53   GNUNET_assert (sizeof (struct MyMessage) == ntohs (mm->header.size));
  54 }
  55
  56
  57 void
  58 test2 (void)
  59 {
  60   struct GNUNET_MQ_Message *mqm;
  61   struct MyMessage *mm;
  62   int res;
  63   char *s = "foo";
  64
  65   mqm = GNUNET_MQ_msg (mm, 42);
  66   res = GNUNET_MQ_nest (mqm, s, strlen(s));
  67   GNUNET_assert (GNUNET_OK == res);
  68   res = GNUNET_MQ_nest (mqm, s, strlen(s));
  69   GNUNET_assert (GNUNET_OK == res);
  70   res = GNUNET_MQ_nest (mqm, NULL, 0);
  71   GNUNET_assert (GNUNET_OK == res);
  72
  73   GNUNET_assert (strlen (s) * 2 + sizeof (struct MyMessage) == ntohs (mm->header.size));
  74
  75   res = GNUNET_MQ_nest_mh (mqm, &mm->header);
  76   GNUNET_assert (GNUNET_OK == res);
  77   GNUNET_assert (2 * (strlen (s) * 2 + sizeof (struct MyMessage)) == ntohs (mm->header.size));
  78
  79   res = GNUNET_MQ_nest (mqm, (void *) 0xF00BA, 0xFFF0);
  80   GNUNET_assert (GNUNET_OK != res);
  81
  82   GNUNET_MQ_discard (mqm);
  83 }
  84
  85
  86 void
  87 test3 (void)
  88 {
  89   struct GNUNET_MQ_Message *mqm;
  90   struct GNUNET_MessageHeader *mh;
  91
  92   mqm = GNUNET_MQ_msg_header (42);
  93   /* how could the above be checked? */
  94
  95   GNUNET_MQ_discard (mqm);
  96
  97   mqm = GNUNET_MQ_msg_header_extra (mh, 20, 42);
  98   GNUNET_assert (42 == ntohs (mh->type));
  99   GNUNET_assert (sizeof (struct GNUNET_MessageHeader) + 20 == ntohs (mh->size));
 100 }
 101
 102
 103 int
 104 main (int argc, char **argv)
 105 {
 106
 107   GNUNET_log_setup ("test-mq", "INFO", NULL);
 108   test1 ();
 109   test2 ();
 110   test3 ();
 111
 112   return 0;
 113 }
 114