src/set/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 #include "mq.h"
  29
  30
  31 GNUNET_NETWORK_STRUCT_BEGIN
  32
  33 struct MyMessage
  34 {
  35   struct GNUNET_MessageHeader header;
  36   uint32_t x GNUNET_PACKED;
  37 };
  38
  39 GNUNET_NETWORK_STRUCT_END
  40
  41 void
  42 test1 (void)
  43 {
  44   struct GNUNET_MQ_Message *mqm;
  45   struct MyMessage *mm;
  46
  47   mm = NULL;
  48   mqm = NULL;
  49
  50   mqm = GNUNET_MQ_msg (mm, 42);
  51   GNUNET_assert (NULL != mqm);
  52   GNUNET_assert (NULL != mm);
  53   GNUNET_assert (42 == ntohs (mm->header.type));
  54   GNUNET_assert (sizeof (struct MyMessage) == ntohs (mm->header.size));
  55 }
  56
  57
  58 void
  59 test2 (void)
  60 {
  61   struct GNUNET_MQ_Message *mqm;
  62   struct MyMessage *mm;
  63   int res;
  64   char *s = "foo";
  65
  66   mqm = GNUNET_MQ_msg (mm, 42);
  67   res = GNUNET_MQ_nest (mqm, s, strlen(s));
  68   GNUNET_assert (GNUNET_OK == res);
  69   res = GNUNET_MQ_nest (mqm, s, strlen(s));
  70   GNUNET_assert (GNUNET_OK == res);
  71   res = GNUNET_MQ_nest (mqm, NULL, 0);
  72   GNUNET_assert (GNUNET_OK == res);
  73
  74   GNUNET_assert (strlen (s) * 2 + sizeof (struct MyMessage) == ntohs (mm->header.size));
  75
  76   res = GNUNET_MQ_nest_mh (mqm, &mm->header);
  77   GNUNET_assert (GNUNET_OK == res);
  78   GNUNET_assert (2 * (strlen (s) * 2 + sizeof (struct MyMessage)) == ntohs (mm->header.size));
  79
  80   res = GNUNET_MQ_nest (mqm, (void *) 0xF00BA, 0xFFF0);
  81   GNUNET_assert (GNUNET_OK != res);
  82
  83   GNUNET_MQ_discard (mqm);
  84 }
  85
  86
  87 void
  88 test3 (void)
  89 {
  90   struct GNUNET_MQ_Message *mqm;
  91   struct GNUNET_MessageHeader *mh;
  92
  93   mqm = GNUNET_MQ_msg_header (42);
  94   /* how could the above be checked? */
  95
  96   GNUNET_MQ_discard (mqm);
  97
  98   mqm = GNUNET_MQ_msg_header_extra (mh, 20, 42);
  99   GNUNET_assert (42 == ntohs (mh->type));
 100   GNUNET_assert (sizeof (struct GNUNET_MessageHeader) + 20 == ntohs (mh->size));
 101 }
 102
 103
 104 int
 105 main (int argc, char **argv)
 106 {
 107
 108   GNUNET_log_setup ("test-mq", "INFO", NULL);
 109   test1 ();
 110   test2 ();
 111   test3 ();
 112
 113   return 0;
 114 }
 115