Merge pull request #11705 from phobos-/msp_vtx

VTX Device over MSP

src/test/Makefile

@@ -462,6 +462,23 @@ rx_spi_expresslrs_telemetry_unittest_DEFINES := \
 		USE_GPS= \
 		USE_MSP_OVER_TELEMETRY= \
 
+vtx_msp_unittest_SRC := \
+        $(USER_DIR)/common/crc.c \
+		$(USER_DIR)/common/streambuf.c \
+        $(USER_DIR)/fc/runtime_config.c \
+		$(USER_DIR)/drivers/vtx_common.c \
+		$(USER_DIR)/drivers/vtx_table.c \
+		$(USER_DIR)/pg/vtx_table.c \
+		$(USER_DIR)/io/vtx_control.c \
+		$(USER_DIR)/io/vtx.c \
+		$(USER_DIR)/io/vtx_msp.c \
+
+vtx_msp_unittest_DEFINES := \
+		USE_VTX_COMMON= \
+		USE_VTX_CONTROL= \
+		USE_VTX_TABLE= \
+		USE_VTX_MSP=
+
 # Please tweak the following variable definitions as needed by your
 # project, except GTEST_HEADERS, which you can use in your own targets
 # but shouldn't modify.

src/test/unit/vtx_msp_unittest.cc

@@ -0,0 +1,116 @@
+/*
+ * This file is part of Cleanflight.
+ *
+ * Cleanflight is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Cleanflight is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with Cleanflight.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <stdint.h>
+#include <stdbool.h>
+
+#include <limits.h>
+
+extern "C" {
+    #include "platform.h"
+
+    #include "io/vtx.h"
+    #include "io/vtx_msp.h"
+    #include "pg/vtx_table.h"
+
+    static uint8_t mspFrame[15];
+    extern mspVtxStatus_e mspVtxStatus;
+}
+
+#include "unittest_macros.h"
+#include "gtest/gtest.h"
+
+//make clean test_vtx_msp_unittest
+TEST(VtxMspUnitTest, TestVtxMspPacket)
+{
+    vtxSettingsConfigMutable()->band = 1;
+    vtxSettingsConfigMutable()->channel = 2;
+    vtxSettingsConfigMutable()->power = 3;
+    vtxSettingsConfigMutable()->freq = 5800;
+    vtxSettingsConfigMutable()->pitModeFreq = 5300;
+    vtxSettingsConfigMutable()->lowPowerDisarm = VTX_LOW_POWER_DISARM_ALWAYS;
+
+    vtxTableConfigMutable()->bands = 4;
+    vtxTableConfigMutable()->channels = 5;
+    vtxTableConfigMutable()->powerLevels = 6;
+
+    uint8_t expectedFrame[15] = {5, 1, 2, 1, 0, 168, 22, 0, 1, 180, 20, 1, 4, 5, 6};
+
+    prepareMspFrame(mspFrame);
+
+    for (int i = 0; i < 15; i++) {
+        EXPECT_EQ(expectedFrame[i], mspFrame[i]);
+    }
+}
+
+TEST(VtxMspUnitTest, TestVtxMspReady)
+{
+    mspVtxStatus = MSP_VTX_STATUS_OFFLINE;
+    vtxTableConfigMutable()->bands = 0;
+    vtxTableConfigMutable()->channels = 0;
+    vtxTableConfigMutable()->powerLevels = 0;
+
+    setMspVtxDeviceStatusReady(0);
+
+    EXPECT_EQ(MSP_VTX_STATUS_OFFLINE, mspVtxStatus);
+
+    vtxTableConfigMutable()->bands = 1;
+    vtxTableConfigMutable()->channels = 1;
+    vtxTableConfigMutable()->powerLevels = 1;
+
+    setMspVtxDeviceStatusReady(1); // testing wrong descriptor
+
+    EXPECT_EQ(MSP_VTX_STATUS_OFFLINE, mspVtxStatus);
+
+    setMspVtxDeviceStatusReady(0);
+
+    EXPECT_EQ(MSP_VTX_STATUS_READY, mspVtxStatus);
+}
+
+// STUBS
+
+extern "C" {
+
+    #include "build/debug.h"
+    #include "fc/rc_modes.h"
+    #include "rx/rx.h"
+    #include "io/serial.h"
+    #include "msp/msp.h"
+
+    uint8_t debugMode = 0;
+    int16_t debug[DEBUG16_VALUE_COUNT];
+
+    uint8_t cliMode = 0;
+    rxRuntimeState_t rxRuntimeState = rxRuntimeState_t();
+    static serialPortConfig_t *findSerialPortConfig_stub_retval;
+
+    bool IS_RC_MODE_ACTIVE(boxId_e) {return false;}
+    void beeperConfirmationBeeps(uint8_t) {}
+    void saveConfigAndNotify(void) {}
+    void crsfRxSendTelemetryData(void) {}
+    void crsfRxWriteTelemetryData(const void *, int) {}
+    bool failsafeIsActive(void) {return false;}
+    const serialPortConfig_t *findSerialPortConfig(serialPortFunction_e) {return findSerialPortConfig_stub_retval;}
+    bool isRangeActive(uint8_t , const channelRange_t *) {return true;}
+    int mspSerialPush(serialPortIdentifier_e, uint8_t, uint8_t *, int, mspDirection_e, mspVersion_e) {return 0;}
+    void tfp_sprintf(char *, char*, ...) {}
+
+    mspDescriptor_t getMspSerialPortDescriptor(const uint8_t ) {return 0;}
+    mspDescriptor_t getMspTelemetryDescriptor(void) {return 0;}
+
+    void mspCmsUpdateStatusString(void) {}
+}