ucw/table-test-2.c

   1 /*
   2  *      Unit tests of table printer
   3  *
   4  *      (c) 2014 Robert Kessl <robert.kessl@economia.cz>
   5  */
   6
   7 #include <ucw/lib.h>
   8 #include <ucw/table.h>
   9 #include <ucw/table-types.h>
  10 #include <ucw/opt.h>
  11 #include <stdio.h>
  12
  13 enum test_table_cols {
  14   TEST_COL0_SIZE, TEST_COL1_TS
  15 };
  16
  17 static struct table_template test_tbl = {
  18   TBL_COLUMNS {
  19     [TEST_COL0_SIZE] = TBL_COL_SIZE_FMT("size", 15, XT_SIZE_FMT_UNIT(XT_SIZE_UNIT_BYTE)),
  20     [TEST_COL1_TS] = TBL_COL_TIMESTAMP("ts", 20),
  21     TBL_COL_END
  22   },
  23   TBL_FMT_HUMAN_READABLE,
  24 };
  25
  26 static void do_test(void)
  27 {
  28   struct fastbuf *out;
  29   out = bfdopen_shared(1, 4096);
  30   struct table *tbl = table_init(&test_tbl);
  31   table_start(tbl, out);
  32
  33   u64 test_time = 1403685533;
  34   s64 test_size = 4LU*(1024LU * 1024LU * 1024LU);
  35
  36   table_col_size(tbl, TEST_COL0_SIZE, test_size);
  37   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
  38   table_end_row(tbl);
  39
  40   tbl->column_order[TEST_COL0_SIZE].fmt = XT_SIZE_FMT_UNIT(XT_SIZE_UNIT_KILOBYTE);
  41   table_col_size(tbl, TEST_COL0_SIZE, test_size);
  42   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
  43   table_end_row(tbl);
  44
  45   tbl->column_order[TEST_COL0_SIZE].fmt = XT_SIZE_FMT_UNIT(XT_SIZE_UNIT_MEGABYTE);
  46   table_col_size(tbl, TEST_COL0_SIZE, test_size);
  47   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
  48   table_end_row(tbl);
  49
  50   tbl->column_order[TEST_COL0_SIZE].fmt = XT_SIZE_FMT_UNIT(XT_SIZE_UNIT_GIGABYTE);
  51   tbl->column_order[TEST_COL1_TS].fmt = XT_TIMESTAMP_FMT_DATETIME;
  52   table_col_size(tbl, TEST_COL0_SIZE, test_size);
  53   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
  54   table_end_row(tbl);
  55
  56   test_size = test_size * 1024LU;
  57   tbl->column_order[TEST_COL0_SIZE].fmt = XT_SIZE_FMT_UNIT(XT_SIZE_UNIT_TERABYTE);
  58   tbl->column_order[TEST_COL1_TS].fmt = XT_TIMESTAMP_FMT_DATETIME;
  59   table_col_size(tbl, TEST_COL0_SIZE, test_size);
  60   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
  61   table_end_row(tbl);
  62
  63   table_end(tbl);
  64   table_cleanup(tbl);
  65
  66   bclose(out);
  67 }
  68
  69 static struct table_template test_tbl2 = {
  70   TBL_COLUMNS {
  71     [TEST_COL0_SIZE] = TBL_COL_SIZE_FMT("size", 15, XT_SIZE_FMT_UNIT(XT_SIZE_UNIT_BYTE)),
  72     [TEST_COL1_TS] = TBL_COL_TIMESTAMP("ts", 20),
  73     TBL_COL_END
  74   },
  75   TBL_FMT_HUMAN_READABLE,
  76 };
  77
  78 static void do_test2(void)
  79 {
  80   struct fastbuf *out;
  81   out = bfdopen_shared(1, 4096);
  82   struct table *tbl = table_init(&test_tbl2);
  83   table_set_col_order_by_name(tbl, "");
  84   const char *err = table_set_option_value(tbl, "cols", "size[KB],size[MB],size[GB],size[TB],size[auto],ts[datetime],ts[timestamp]");
  85   if(err) {
  86     opt_failure("err in table_set_option_value: '%s'.", err);
  87     abort();
  88   }
  89   table_start(tbl, out);
  90
  91   u64 test_time = 1403685533;
  92   s64 test_size = 4LU*(1024LU * 1024LU);
  93
  94   table_col_size(tbl, TEST_COL0_SIZE, test_size);
  95   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
  96   table_end_row(tbl);
  97
  98   test_size = test_size * 1024LU;
  99
 100   table_col_size(tbl, TEST_COL0_SIZE, test_size);
 101   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
 102   table_end_row(tbl);
 103
 104   test_size = test_size * 1024LU;
 105
 106   table_col_size(tbl, TEST_COL0_SIZE, test_size);
 107   table_col_timestamp(tbl, TEST_COL1_TS, test_time);
 108   table_end_row(tbl);
 109
 110   table_end(tbl);
 111   table_cleanup(tbl);
 112
 113   bclose(out);
 114 }
 115
 116 int main(int argc UNUSED, char **argv UNUSED)
 117 {
 118   do_test();
 119   do_test2();
 120
 121   return 0;
 122 }
 123