tests/lib/timestamp-test.c

   1 /* SPDX-License-Identifier: GPL-2.0-only */
   2
   3 #include "../lib/timestamp.c"
   4 #include <commonlib/bsd/helpers.h>
   5 #include <tests/test.h>
   6 #include "stubs/timestamp.h"
   7
   8 /* Timestamp region definition */
   9 #define TIMESTAMP_REGION_SIZE (1 * KiB)
  10 TEST_REGION(timestamp, TIMESTAMP_REGION_SIZE);
  11
  12 void test_timestamp_init(void **state)
  13 {
  14         timestamp_init(1000);
  15
  16         assert_non_null(glob_ts_table);
  17 }
  18
  19 void test_timestamp_add(void **state)
  20 {
  21         const int base_multipler = 2000;
  22         const int timestamp_base = 1000;
  23         struct timestamp_entry *entry;
  24         int i;
  25
  26         timestamp_init(timestamp_base);
  27
  28         timestamp_add(TS_ROMSTAGE_START, base_multipler);
  29
  30         assert_int_equal(1, glob_ts_table->num_entries);
  31
  32         entry = &glob_ts_table->entries[0];
  33         assert_int_equal(1, entry->entry_id);
  34         assert_int_equal(base_multipler - timestamp_base, /* Added timestamp reduced by base */
  35                          entry->entry_stamp);
  36
  37         /* Add few timestamps to check if all of them will be added properly */
  38         for (i = 1; i < 10; ++i)
  39                 timestamp_add(i + 1, base_multipler * (i + 1));
  40
  41         assert_int_equal(10, glob_ts_table->num_entries);
  42
  43         for (i = 0; i < 10; ++i) {
  44                 entry = &glob_ts_table->entries[i];
  45                 assert_int_equal(i + 1, entry->entry_id);
  46                 assert_int_equal(base_multipler * (i + 1) - timestamp_base, entry->entry_stamp);
  47         }
  48 }
  49
  50 void test_timestamp_add_now(void **state)
  51 {
  52         const int base_multipler = 2000;
  53         const int timestamp_base = 1000;
  54         struct timestamp_entry *entry;
  55
  56         /* Initialize with base timestamp of 1000.
  57          * This value will be subtracted from each timestamp
  58          * when adding it.
  59          */
  60         timestamp_init(timestamp_base);
  61
  62         dummy_timestamp_set(base_multipler);
  63
  64         timestamp_add_now(TS_ROMSTAGE_START);
  65
  66         assert_int_equal(1, glob_ts_table->num_entries);
  67
  68         entry = &glob_ts_table->entries[0];
  69
  70         assert_int_equal(1, entry->entry_id);
  71         assert_int_equal(base_multipler - timestamp_base, /* Added timestamp reduced by base */
  72                          entry->entry_stamp);
  73 }
  74
  75 void test_timestamp_rescale_table(void **state)
  76 {
  77         const int base_multipler = 1000;
  78         int i;
  79
  80         timestamp_init(0);
  81
  82         /* Add few timestamps to check if all of them will be rescaled properly */
  83         for (i = 1; i <= 10; ++i)
  84                 timestamp_add(i, base_multipler * i);
  85
  86         /* Check if all entries were added to table */
  87         assert_int_equal(10, glob_ts_table->num_entries);
  88
  89         timestamp_rescale_table(2, 4);
  90
  91         /* Check if there is the same number of entries */
  92         assert_int_equal(10, glob_ts_table->num_entries);
  93
  94         for (i = 0; i < glob_ts_table->num_entries; ++i)
  95                 assert_int_equal(base_multipler * (i + 1) / 4 * 2,
  96                                  glob_ts_table->entries[i].entry_stamp);
  97 }
  98
  99 void test_get_us_since_boot(void **state)
 100 {
 101         const int base_multipler = 10000;
 102         const int timestamp_base = 1000;
 103         const int freq_base = 100;
 104
 105         timestamp_init(timestamp_base);
 106         dummy_timestamp_set(base_multipler);
 107         dummy_timestamp_tick_freq_mhz_set(freq_base);
 108         /* There is a need to update this field manually, because cbmem hooks are not used. */
 109         glob_ts_table->tick_freq_mhz = freq_base;
 110
 111         assert_int_equal((base_multipler - timestamp_base) / freq_base, get_us_since_boot());
 112 }
 113
 114 int setup_timestamp_and_freq(void **state)
 115 {
 116         dummy_timestamp_set(0);
 117         dummy_timestamp_tick_freq_mhz_set(1);
 118
 119         return 0;
 120 }
 121
 122 int main(void)
 123 {
 124         const struct CMUnitTest tests[] = {
 125                 cmocka_unit_test_setup(test_timestamp_init, setup_timestamp_and_freq),
 126                 cmocka_unit_test_setup(test_timestamp_add, setup_timestamp_and_freq),
 127                 cmocka_unit_test_setup(test_timestamp_add_now, setup_timestamp_and_freq),
 128                 cmocka_unit_test_setup(test_timestamp_rescale_table, setup_timestamp_and_freq),
 129                 cmocka_unit_test_setup(test_get_us_since_boot, setup_timestamp_and_freq),
 130         };
 131
 132 #if CONFIG(COLLECT_TIMESTAMPS)
 133         return cb_run_group_tests(tests, NULL, NULL);
 134 #else
 135         return 0;
 136 #endif
 137 }