src/test/fhss_native/test_fhss.cpp

   1 #include <cstdint>
   2 #include <FHSS.h>
   3 #include <unity.h>
   4 #include <set>
   5
   6 void test_fhss_first(void)
   7 {
   8     FHSSrandomiseFHSSsequence(0x01020304L);
   9     TEST_ASSERT_EQUAL(GetInitialFreq(), FHSSfreqs[sync_channel]);
  10 }
  11
  12 void test_fhss_assignment(void)
  13 {
  14     FHSSrandomiseFHSSsequence(0x01020304L);
  15
  16     const uint32_t numFhss = FHSSgetChannelCount();
  17     uint32_t initFreq = GetInitialFreq();
  18
  19     uint32_t freq = initFreq;
  20     for (unsigned int i = 0; i < 512; i++) {
  21         if ((i % numFhss) == 0) {
  22             TEST_ASSERT_EQUAL(initFreq, freq);
  23         } else {
  24             TEST_ASSERT_NOT_EQUAL(initFreq, freq);
  25         }
  26         freq = FHSSgetNextFreq();
  27     }
  28 }
  29
  30 void test_fhss_unique(void)
  31 {
  32     FHSSrandomiseFHSSsequence(0x01020304L);
  33
  34     const uint32_t numFhss = FHSSgetChannelCount();
  35     std::set<uint32_t> freqs;
  36
  37     for (unsigned int i = 0; i < 256; i++) {
  38         uint32_t freq = FHSSgetNextFreq();
  39
  40         if ((i % numFhss) == 0) {
  41             freqs.clear();
  42             freqs.insert(freq);
  43         } else {
  44             bool inserted = freqs.insert(freq).second;
  45             TEST_ASSERT_TRUE_MESSAGE(inserted, "Should only see a frequency one time per number initial value");
  46         }
  47     }
  48 }
  49
  50 void test_fhss_same(void)
  51 {
  52     FHSSrandomiseFHSSsequence(0x01020304L);
  53
  54     const uint32_t numFhss = FHSSgetSequenceCount();
  55
  56     uint32_t fhss[numFhss];
  57
  58     for (unsigned int i = 0; i < FHSSgetSequenceCount(); i++) {
  59         uint32_t freq = FHSSgetNextFreq();
  60         fhss[i] = freq;
  61     }
  62
  63     FHSSrandomiseFHSSsequence(0x01020304L);
  64
  65     for (unsigned int i = 0; i < FHSSgetSequenceCount(); i++) {
  66         uint32_t freq = FHSSgetNextFreq();
  67         TEST_ASSERT_EQUAL(fhss[i],freq);
  68     }
  69 }
  70
  71 int main(int argc, char **argv)
  72 {
  73     UNITY_BEGIN();
  74     RUN_TEST(test_fhss_first);
  75     RUN_TEST(test_fhss_assignment);
  76     RUN_TEST(test_fhss_unique);
  77     RUN_TEST(test_fhss_same);
  78     UNITY_END();
  79
  80     return 0;
  81 }