t_tc_load.c

Go to the documentation of this file.

/*
** Copyright (C) 2005 by Kevin L. Mitchell <klmitch@mit.edu>
**
** 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 2 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, write to the Free Software
** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
**
** @(#)$Id: t_tc_load.c,v 1.6 2005/06/08 04:38:39 klmitch Exp $
*/
#include "test-harness.h"

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "treeconf_int.h"

RCSTAG("@(#)$Id: t_tc_load.c,v 1.6 2005/06/08 04:38:39 klmitch Exp $");

TEST_PROG(t_tc_load, "Test tc_load() function")
TEST_DEP(t_tc_load, t_treeconf)
TEST_ARG(t_tc_load, "<t_tc_load.conf>")
TEST_ARG(t_tc_load, "<t_tc_load.res>")

#define RES_CHUNK 10

#define RES_BUFSIZE     1024

static const char *prog;

/* load the results file and register the variables it lists */
static char **
load_results(treeconf_ctx_t *ctx, const char *file)
{
  char **results = 0, buf[RES_BUFSIZE], *var, *def, *val = 0, *nl;
  unsigned int err;
  int pos = 0, size = 0;
  FILE *fp;

  /* Open the results file... */
  if (!(fp = fopen(file, "r"))) {
    perror(file);
    exit(1);
  }

  while ((var = fgets(buf, sizeof(buf), fp))) {
    if (*var == '#' || /* skip comment lines... */
      !(nl = strchr(buf, '\n')) || /* look for trailing newline... */
      ((def = strchr(var, ':')) && /* find beginning of default value */
      !(val = strchr(def + 1, ':')))) /* find beginning of expected value */
      continue; /* skip invalid lines... */

    *nl = '\0'; /* clip out newline */

    if (def) { /* if default was specified... */
      *(def++) = '\0'; /* terminate variable name... */
      *(val++) = '\0'; /* and default value */

      if (pos >= size - 1) { /* need some more space to store pointers... */
      if (!(results = (char **)realloc(results, sizeof(char *) *
                               (size += RES_CHUNK)))) {
        fprintf(stderr, "%s: Out of memory\n", prog);
        exit(1);
      }
      }

      /* now allocate memory for the result string; +2 accounts for = and \0 */
      if (!(results[pos] = (char *)malloc(strlen(var) + strlen(val) + 2))) {
      fprintf(stderr, "%s: Out of memory\n", prog);
      exit(1);
      }

      /* next, fill in the string */
      sprintf(results[pos++], "%s=%s", var, val);
    }

    /* Finally, register the variable */
    if ((err = tc_register(ctx, var, def, 0, 0, 0, 0))) {
      fprintf(stderr, "%s: tc_register() failed on %s: %u\n", prog, var, err);
      exit(1);
    }
  }

  fclose(fp); /* done with file... */

  if (!results) { /* make sure we actually loaded some results */
    fprintf(stderr, "%s: No results in list\n", file);
    exit(1);
  }

  results[pos] = 0; /* terminate the array */

  return results;
}

int
main(int argc, char **argv)
{
  treeconf_ctx_t ctx = TREECONF_CTX_INIT;
  char *conf, *result, **results, *t;
  const char *val;
  unsigned int err = 0;

  prog = argv[0]; /* remember program name */

  if (argc < 3) { /* must have two arguments: conf and result */
    fprintf(stderr, "Usage: %s <conf> <res>\n", prog);
    exit(1);
  }

  conf = argv[1];
  result = argv[2];

  results = load_results(&ctx, result); /* load results and register vars */

  /* OK, now attempt to load the configuration file */
  TEST(t_tc_load, tc_load_call, "Test that tc_load() may be called",
       (!(err = tc_load(conf, &ctx))), FATAL(0),
       ("tc_load() successfully loaded configuration \"%s\"", conf),
       ("tc_load() call failed with error %u", err));

  TEST_DECL(t_tc_load, tc_load_verify, "Test that tc_load() loaded expected "
          "values")
  /* Now check that the load attempt actually worked */
  for (; *results; results++) { /* walk through the results */
    if (!(t = strchr(*results, '='))) /* splitting them at the '='... */
      continue; /* shouldn't ever happen, but just go on if it does */
    *(t++) = '\0'; /* split variable name from expected value */

    /* get the variable's value... */
    if ((err = tc_get(&ctx, *results, &val, 0)))
      FAIL(TEST_NAME(tc_load_verify), FATAL(0), "tc_get() on variable \"%s\" "
         "failed with error %u", *results, err);
    else if (strcmp(t, val))
      FAIL(TEST_NAME(tc_load_verify), FATAL(0), "Value mismatch on variable "
         "\"%s\": expected \"%s\", got \"%s\"", *results, t, val);
  }
  PASS(TEST_NAME(tc_load_verify), "tc_load() successfully loaded all "
       "variables");

  return 0;
}

---