/* Test of vasnprintf() and asnprintf() functions.

   Copyright (C) 2007-2017 Free Software Foundation, Inc.

   This program is free software: you can redistribute it and/or modify

   it under the terms of the GNU General Public License as published by

   the Free Software Foundation; either version 3 of the License, or

   (at your option) any later version.

   This program is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License

   along with this program.  If not, see <http://www.gnu.org/licenses/>.  */

/* Written by Bruno Haible <bruno@clisp.org>, 2007.  */

#include <config.h>

#include "vasnprintf.h"

#include <stdarg.h>

#include <stdlib.h>

#include <string.h>

#include "macros.h"

static void

test_function (char * (*my_asnprintf) (char *, size_t *, const char *, ...))

{

  char buf[8];

  int size;

  for (size = 0; size <= 8; size++)

    {

      size_t length = size;

      char *result = my_asnprintf (NULL, &length, "%d", 12345);

      ASSERT (result != NULL);

      ASSERT (strcmp (result, "12345") == 0);

      ASSERT (length == 5);

      free (result);

    }

  for (size = 0; size <= 8; size++)

    {

      size_t length;

      char *result;

      memcpy (buf, "DEADBEEF", 8);

      length = size;

      result = my_asnprintf (buf, &length, "%d", 12345);

      ASSERT (result != NULL);

      ASSERT (strcmp (result, "12345") == 0);

      ASSERT (length == 5);

      if (size < 6)

        ASSERT (result != buf);

      ASSERT (memcmp (buf + size, &"DEADBEEF"[size], 8 - size) == 0);

      if (result != buf)

        free (result);

    }

}

static char *

my_asnprintf (char *resultbuf, size_t *lengthp, const char *format, ...)

{

  va_list args;

  char *ret;

  va_start (args, format);

  ret = vasnprintf (resultbuf, lengthp, format, args);

  va_end (args);

  return ret;

}

static void

test_vasnprintf ()

{

  test_function (my_asnprintf);

}

static void

test_asnprintf ()

{

  test_function (asnprintf);

}

int

main (int argc, char *argv[])

{

  test_vasnprintf ();

  test_asnprintf ();

  return 0;

}