/*
 * Copyright (C) EdgeTX
 *
 * Based on code named
 *   opentx - https://github.com/opentx/opentx
 *   th9x - http://code.google.com/p/th9x
 *   er9x - http://code.google.com/p/er9x
 *   gruvin9x - http://code.google.com/p/gruvin9x
 *
 * License GPLv2: http://www.gnu.org/licenses/gpl-2.0.html
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * This program 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.
 */

#include "gtests.h"

#include "storage/yaml/yaml_tree_walker.h"
#include "storage/yaml/yaml_parser.h"
#include "storage/yaml/yaml_datastructs.h"
#include "storage/yaml/yaml_bits.h"

static const char _radio_config[] =
    "potsConfig: \n"
    "   P1:\n"
    "      type: without_detent\n"
    "      name: \"\"\n"
    "   P2:\n"
    "      type: multipos_switch\n"
    "      name: \"\"\n"
    "   P3:\n"
    "      type: with_detent\n"
    "      name: \"\"\n"
    "   SL1:\n"
    "      type: slider\n"
    "      name: \"\"\n"
    "   SL2:\n"
    "      type: slider\n"
    "      name: \"\"\n"
    "switchConfig: \n"
    "   SA:\n"
    "      type: 3pos\n"
    "      name: \"\"\n"
    "   SB:\n"
    "      type: 3pos\n"
    "      name: \"\"\n"
    "   SC:\n"
    "      type: 3pos\n"
    "      name: \"\"\n"
    "   SD:\n"
    "      type: 3pos\n"
    "      name: \"\"\n"
    "   SE:\n"
    "      type: 3pos\n"
    "      name: \"\"\n"
    "   SF:\n"
    "      type: 2pos\n"
    "      name: \"\"\n"
    "   SG:\n"
    "      type: 3pos\n"
    "      name: \"\"\n"
    "   SH:\n"
    "      type: toggle\n"
    "      name: \"\"\n";

static void loadRadioYamlStr(const char* str)
{
  YamlTreeWalker tree;
  tree.reset(get_radiodata_nodes(), (uint8_t*)&g_eeGeneral);

  YamlParser yp;
  yp.init(YamlTreeWalker::get_parser_calls(), &tree);

  size_t len = strlen(str);
  yp.parse(str, len);
}

TEST(Sources, getSourceString)
{
  loadRadioYamlStr(_radio_config);

#if defined(IMU)
  EXPECT_STREQ(getSourceString(MIXSRC_TILT_X), "TltX");
  EXPECT_STREQ(getSourceString(MIXSRC_TILT_Y), "TltY");
#endif

#if defined(PCBHORUS)
  EXPECT_STREQ(getSourceString(MIXSRC_SPACEMOUSE_A), "smA");
  EXPECT_STREQ(getSourceString(MIXSRC_SPACEMOUSE_B), "smB");
  EXPECT_STREQ(getSourceString(MIXSRC_SPACEMOUSE_C), "smC");
  EXPECT_STREQ(getSourceString(MIXSRC_SPACEMOUSE_D), "smD");
  EXPECT_STREQ(getSourceString(MIXSRC_SPACEMOUSE_E), "smE");
  EXPECT_STREQ(getSourceString(MIXSRC_SPACEMOUSE_F), "smF");
#endif

  EXPECT_STREQ(getSourceString(MIXSRC_MIN), STR_MENU_MIN);
  EXPECT_STREQ(getSourceString(MIXSRC_MAX), STR_MENU_MAX);

#if defined(HELI)
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_HELI), "CYC1");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_HELI + 1), "CYC2");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_HELI + 2), "CYC3");
#else
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_HELI), "[C1]");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_HELI + 1), "[C2]");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_HELI + 2), "[C3]");
#endif
#if defined(SURFACE_RADIO)
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_TRIM), CHAR_TRIM "ST");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_TRIM + 1), CHAR_TRIM "TH");
#else
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_TRIM), CHAR_TRIM "Rud");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_TRIM + 1), CHAR_TRIM "Ele");
  EXPECT_STREQ(getSourceString(MIXSRC_FIRST_TRIM + 2), CHAR_TRIM "Thr");
#endif
}