src/util/test_mq.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2012 GNUnet e.V.
   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., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, USA.
  19 */
  20
  21 /**
  22  * @file util/test_mq.c
  23  * @brief simple tests for mq
  24  */
  25 #include "platform.h"
  26 #include "gnunet_util_lib.h"
  27
  28
  29 GNUNET_NETWORK_STRUCT_BEGIN
  30
  31 struct MyMessage
  32 {
  33   struct GNUNET_MessageHeader header;
  34   uint32_t x GNUNET_PACKED;
  35 };
  36
  37 GNUNET_NETWORK_STRUCT_END
  38
  39
  40 static void
  41 test1 ()
  42 {
  43   struct GNUNET_MQ_Envelope *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   GNUNET_MQ_discard (mqm);
  55 }
  56
  57
  58 static void
  59 test2 ()
  60 {
  61   struct GNUNET_MQ_Envelope *mqm;
  62   struct GNUNET_MessageHeader *mh;
  63
  64   mqm = GNUNET_MQ_msg_header (42);
  65   /* how could the above be checked? */
  66
  67   GNUNET_MQ_discard (mqm);
  68
  69   mqm = GNUNET_MQ_msg_header_extra (mh, 20, 42);
  70   GNUNET_assert (42 == ntohs (mh->type));
  71   GNUNET_assert (sizeof (struct GNUNET_MessageHeader) + 20 == ntohs (mh->size));
  72   GNUNET_MQ_discard (mqm);
  73 }
  74
  75
  76 int
  77 main (int argc, char **argv)
  78 {
  79   GNUNET_log_setup ("test-mq", "INFO", NULL);
  80   test1 ();
  81   test2 ();
  82   return 0;
  83 }
  84