testsuite/testsuite.h

   1 /*
   2  * Copyright (C) 2012-2013  ProFUSION embedded systems
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  17  */
  18
  19 #pragma once
  20
  21 #include <stdbool.h>
  22 #include <stdarg.h>
  23
  24 #include <shared/macro.h>
  25
  26 struct test;
  27 typedef int (*testfunc)(const struct test *t);
  28
  29 enum test_config {
  30         /*
  31          * Where's the roots dir for this test. It will LD_PRELOAD path.so in
  32          * order to trap calls to functions using paths.
  33          */
  34         TC_ROOTFS = 0,
  35
  36         /*
  37          * What's the desired string to be returned by `uname -r`. It will
  38          * trap calls to uname(3P) by LD_PRELOAD'ing uname.so and then filling
  39          * in the information in u.release.
  40          */
  41         TC_UNAME_R,
  42
  43         /*
  44          * Fake calls to init_module(2), returning return-code and setting
  45          * errno to err-code. Set this variable with the following format:
  46          *
  47          *        modname:return-code:err-code
  48          *
  49          * When this variable is used, all calls to init_module() are trapped
  50          * and by default the return code is 0. In other words, they fake
  51          * "success" for all modules, except the ones in the list above, for
  52          * which the return codes are used.
  53          */
  54         TC_INIT_MODULE_RETCODES,
  55
  56         /*
  57          * Fake calls to delete_module(2), returning return-code and setting
  58          * errno to err-code. Set this variable with the following format:
  59          *
  60          *        modname:return-code:err-code
  61          *
  62          * When this variable is used, all calls to init_module() are trapped
  63          * and by default the return code is 0. In other words, they fake
  64          * "success" for all modules, except the ones in the list above, for
  65          * which the return codes are used.
  66          */
  67         TC_DELETE_MODULE_RETCODES,
  68
  69         _TC_LAST,
  70 };
  71
  72 #define S_TC_ROOTFS "TESTSUITE_ROOTFS"
  73 #define S_TC_UNAME_R "TESTSUITE_UNAME_R"
  74 #define S_TC_INIT_MODULE_RETCODES "TESTSUITE_INIT_MODULE_RETCODES"
  75 #define S_TC_DELETE_MODULE_RETCODES "TESTSUITE_DELETE_MODULE_RETCODES"
  76
  77 struct keyval {
  78         const char *key;
  79         const char *val;
  80 };
  81
  82 struct test {
  83         const char *name;
  84         const char *description;
  85         struct {
  86                 /* File with correct stdout */
  87                 const char *out;
  88                 /* File with correct stderr */
  89                 const char *err;
  90
  91                 /*
  92                  * Vector with pair of files
  93                  * key = correct file
  94                  * val = file to check
  95                  */
  96                 const struct keyval *files;
  97         } output;
  98         /* comma-separated list of loaded modules at the end of the test */
  99         const char *modules_loaded;
 100         testfunc func;
 101         const char *config[_TC_LAST];
 102         const char *path;
 103         const struct keyval *env_vars;
 104         bool need_spawn;
 105         bool expected_fail;
 106 };
 107
 108
 109 const struct test *test_find(const struct test *tests[], const char *name);
 110 int test_init(int argc, char *const argv[], const struct test *tests[]);
 111 int test_spawn_prog(const char *prog, const char *const args[]);
 112
 113 int test_run(const struct test *t);
 114
 115 #define TS_EXPORT __attribute__ ((visibility("default")))
 116
 117 #define _LOG(prefix, fmt, ...) printf("TESTSUITE: " prefix fmt, ## __VA_ARGS__)
 118 #define LOG(fmt, ...) _LOG("", fmt, ## __VA_ARGS__)
 119 #define WARN(fmt, ...) _LOG("WARN: ", fmt, ## __VA_ARGS__)
 120 #define ERR(fmt, ...) _LOG("ERR: ", fmt, ## __VA_ARGS__)
 121
 122 #define assert_return(expr, r)                                          \
 123         do {                                                            \
 124                 if ((!(expr))) {                                        \
 125                         ERR("Failed assertion: " #expr "%s:%d %s",                      \
 126                             __FILE__, __LINE__, __PRETTY_FUNCTION__);   \
 127                         return (r);                                     \
 128                 }                                                       \
 129         } while (false)
 130
 131
 132 /* Test definitions */
 133 #define DEFINE_TEST(_name, ...) \
 134         const struct test s##_name = { \
 135                 .name = #_name, \
 136                 .func = _name, \
 137                 ## __VA_ARGS__ \
 138         }
 139
 140 #define TESTSUITE_MAIN(_tests) \
 141         int main(int argc, char *argv[])                        \
 142         {                                                       \
 143                 const struct test *t;                           \
 144                 int arg;                                        \
 145                 size_t i;                                       \
 146                                                                 \
 147                 arg = test_init(argc, argv, tests);             \
 148                 if (arg == 0)                                   \
 149                         return 0;                               \
 150                                                                 \
 151                 if (arg < argc) {                               \
 152                         t = test_find(tests, argv[arg]);        \
 153                         if (t == NULL) {                        \
 154                                 fprintf(stderr, "could not find test %s\n", argv[arg]);\
 155                                 exit(EXIT_FAILURE);             \
 156                         }                                       \
 157                                                                 \
 158                         return test_run(t);                     \
 159                 }                                               \
 160                                                                 \
 161                 for (i = 0; tests[i] != NULL; i++) {            \
 162                         if (test_run(tests[i]) != 0)            \
 163                                 exit(EXIT_FAILURE);             \
 164                 }                                               \
 165                                                                 \
 166                 exit(EXIT_SUCCESS);                             \
 167         }                                                       \
 168
 169 #ifdef noreturn
 170 # define __noreturn noreturn
 171 #elif __STDC_VERSION__ >= 201112L
 172 # define __noreturn _Noreturn
 173 #else
 174 # define __noreturn __attribute__((noreturn))
 175 #endif