src/test/crsf_native/test_crsf.cpp

   1 #include <cstdint>
   2 #include <unity.h>
   3 #include <iostream>
   4 #include "../msp_native/mock_serial.h"
   5
   6 #include "CRSF.h"
   7
   8 using namespace std;
   9 // Mock out the serial port using a string stream
  10 std::string buf;
  11 StringStream ss(buf);
  12
  13 // Create a CRSF object to test,
  14 // using the StringStream as a mock UART
  15 CRSF crsf(&ss);
  16
  17 GENERIC_CRC8 test_crc(CRSF_CRC_POLY);
  18
  19 void test_device_info(void)
  20 {
  21     uint8_t deviceInformation[DEVICE_INFORMATION_LENGTH];
  22
  23     TEST_ASSERT_EQUAL(22, DEVICE_INFORMATION_PAYLOAD_LENGTH);
  24     TEST_ASSERT_EQUAL(28, DEVICE_INFORMATION_LENGTH);
  25
  26     crsf.GetDeviceInformation(deviceInformation, 0);
  27     crsf.SetExtendedHeaderAndCrc(deviceInformation, CRSF_FRAMETYPE_DEVICE_INFO, DEVICE_INFORMATION_FRAME_SIZE, CRSF_ADDRESS_CRSF_RECEIVER, CRSF_ADDRESS_FLIGHT_CONTROLLER);
  28
  29     crsf_ext_header_t *header = (crsf_ext_header_t *) deviceInformation;
  30
  31     TEST_ASSERT_EQUAL(26, header->frame_size);
  32     TEST_ASSERT_EQUAL(CRSF_FRAMETYPE_DEVICE_INFO, header->type);
  33     TEST_ASSERT_EQUAL(CRSF_ADDRESS_FLIGHT_CONTROLLER, header->dest_addr);
  34     TEST_ASSERT_EQUAL(CRSF_ADDRESS_FLIGHT_CONTROLLER, header->device_addr);
  35     TEST_ASSERT_EQUAL(CRSF_ADDRESS_CRSF_RECEIVER, header->orig_addr);
  36     TEST_ASSERT_EQUAL(DEVICE_INFORMATION_FRAME_SIZE, header->frame_size);
  37
  38     uint8_t *data = deviceInformation + sizeof(crsf_ext_header_t);
  39     uint8_t compare [] = {0x74, 0x65, 0x73, 0x74, 0x69, 0x6e, 0x67, 0x0, 0x45, 0x4c, 0x52, 0x53, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0};
  40
  41     for(int i = 0; i < DEVICE_INFORMATION_PAYLOAD_LENGTH; i++)
  42     {
  43         TEST_ASSERT_EQUAL(compare[i], data[i]);
  44     }
  45
  46     TEST_ASSERT_EQUAL(test_crc.calc(&deviceInformation[2], DEVICE_INFORMATION_LENGTH-3), deviceInformation[DEVICE_INFORMATION_LENGTH - 1]);
  47 }
  48
  49
  50 int main(int argc, char **argv)
  51 {
  52     UNITY_BEGIN();
  53     RUN_TEST(test_device_info);
  54     UNITY_END();
  55
  56     return 0;
  57 }