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 }
  55
  56
  57 static void
  58 test2 ()
  59 {
  60   struct GNUNET_MQ_Envelope *mqm;
  61   struct GNUNET_MessageHeader *mh;
  62
  63   mqm = GNUNET_MQ_msg_header (42);
  64   /* how could the above be checked? */
  65
  66   GNUNET_MQ_discard (mqm);
  67
  68   mqm = GNUNET_MQ_msg_header_extra (mh, 20, 42);
  69   GNUNET_assert (42 == ntohs (mh->type));
  70   GNUNET_assert (sizeof (struct GNUNET_MessageHeader) + 20 == ntohs (mh->size));
  71   GNUNET_MQ_discard (mqm);
  72 }
  73
  74
  75 int
  76 main (int argc, char **argv)
  77 {
  78   GNUNET_log_setup ("test-mq", "INFO", NULL);
  79   test1 ();
  80   test2 ();
  81   return 0;
  82 }
  83