TDengine/tools/shell/test/shellTest.cpp

/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), 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.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include <gtest/gtest.h>
#include <iostream>
#include <cstring> 
#include <cctype>
#include "shellAuto.h"
#include "shellInt.h"

TEST(fieldOptionsArea, autoTabTest) {
  printf("hello world  SHELL tab test\n");

  // str false
  const char *s0[] = {
      "create table st(ts ",
      "create table st(ts timestamp, age ",
      "create table st(ts timestamp,  age",
      "create table st(ts timestamp, age int ,  name ",
      "create table st(ts timestamp, age int ,  name  binary(16)",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( ",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int, addr ",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int,addr varbinary",
      "create table st(ts timestamp, age int, name binary(16)) tags(area int  ,  addr varbinary(32)",
      "create table st( ts timestamp, age int, name binary(16)) tags( area int,    addr",
      "create table st  (ts   timestamp ,  age int, name binary(16) , area int,",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int ,addr varbinary",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) level "
      "'high' , no i",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) encode "
      "'simple8b' level 'high', no  in",
  };

  // str true
  const char *s1[] = {
      "create table st(ts timestamp ",
      "create table st(ts timestamp, age int ",
      "create table st(ts timestamp,  age  int  ",
      "create table st(ts timestamp, age int ,  name binary(16) ",
      "create table st(ts timestamp, age int ,  name  binary(16)  ",
      "create table st(ts timestamp, age int ,  name  binary(16) , addr varbinary( 32 ) ",
      "create table st(ts timestamp, age int ,  name  binary(16) ,area int, addr varbinary(32) ",
      "create table st(ts timestamp, age int ,  name  binary(16), area int,addr varbinary(32) ",
      "create table st(ts timestamp, age int, name binary(16) , area int,addr varbinary(32) ",
      "create table st( ts timestamp, age int, name binary(16) ,area int,addr varbinary(32) ",
      "create table st  (ts   timestamp ,  age int, name binary(16), area int,addr varbinary(32) ",
      "create table st  (ts   timestamp ,  age int, name binary(16), area int , addr varbinary(32) compress 'zlib' ",
      "create table st  (ts   timestamp ,  age int, name binary(16),  area int , addr varbinary(32) level 'high' ",
      "create table st  (ts   timestamp ,  age int, name binary(16) , area int , addr varbinary(32) encode 'simple8b' "
      "level 'high'   ",
  };

  // s0 is false
  for (int32_t i = 0; i < sizeof(s0) / sizeof(char *); i++) {
    printf("s0 i=%d fieldOptionsArea %s expect false \n", i, s0[i]);
    ASSERT(fieldOptionsArea((char *)s0[i]) == false);
  }

  // s1 is true
  for (int32_t i = 0; i < sizeof(s1) / sizeof(char *); i++) {
    printf("s1 i=%d fieldOptionsArea %s expect true \n", i, s1[i]);
    ASSERT(fieldOptionsArea((char *)s1[i]) == true);
  }
}

TEST(isCreateFieldsArea, autoTabTest) {
  printf("hellow world  SHELL tab test\n");

  // str false
  const char *s0[] = {
      "create table st(ts )",
      "create table st(ts timestamp, age) ",
      "create table st(ts timestamp,  age)",
      "create table st(ts timestamp, age int ,  name binary(16) )",
      "create table st(ts timestamp, age int ,  name  binary(16))",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( )",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int, addr )",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int,addr varbinary)",
      "create table st(ts timestamp, age int, name binary(16)) tags(area int  ,  addr varbinary(32))",
      "create table st( ts timestamp, age int, name binary(16)) tags( area int,    addr int)",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int,addr varbinary(32) )",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int ,addr varbinary(14))",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) level "
      "'high' )",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) encode "
      "'simple8b' level 'high' )  ",
  };

  // str true
  const char *s1[] = {
      "create table st(ts timestamp ",
      "create table st(ts timestamp, age int ",
      "create table st(ts timestamp,  age  int  ,",
      "create table st(ts timestamp, age int ,  name binary(16), ",
      "create table st(ts timestamp, age int ,  name  binary(16)  ",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int ",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int, addr varbinary(32) ",
      "create table st(ts timestamp, age int ,  name  binary(16) ) tags( area int,addr varbinary(32)",
      "create table st(ts timestamp, age int, name binary(16)) tags(area int,addr varbinary(32) ",
      "create table st( ts timestamp, age int, name binary(16)) tags(area int,addr varbinary(32) ",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int, addr varbinary(32) ",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) compress "
      "'zlib' ",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) level "
      "'high' ",
      "create table st  (ts   timestamp ,  age int, name binary(16) ) tags ( area int , addr varbinary(32) encode "
      "'simple8b' level 'high' ",
  };

  // s0 is false
  for (int32_t i = 0; i < sizeof(s0) / sizeof(char *); i++) {
    printf("s0 i=%d isCreateFieldsArea %s expect false. \n", i, s0[i]);
    ASSERT(isCreateFieldsArea((char *)s0[i]) == false);
  }

  // s1 is true
  for (int32_t i = 0; i < sizeof(s1) / sizeof(char *); i++) {
    printf("s1 i=%d isCreateFieldsArea %s expect true. \n", i, s1[i]);
    ASSERT(isCreateFieldsArea((char *)s1[i]) == true);
  }
}

TEST(shellCountPrefixOnes, checkUtf8) {
  // test ASCII
  assert(shellCountPrefixOnes(0x00) == 1);   // 00000000 → 0 leading 1s → Return 1
  assert(shellCountPrefixOnes(0x41) == 1);   // 01000001 → 0 leading 1s → Return 1
  assert(shellCountPrefixOnes(0x7F) == 1);   // 01111111 → 0 leading 1s → Return 1

  // test two byte characters (such as Latin extended characters)
  assert(shellCountPrefixOnes(0xC2) == 2);   // 11000010 → 2 leading 1s → Return 2
  assert(shellCountPrefixOnes(0xDF) == 2);   // 11011111 → 2 leading 1s → Return 2

  // test three byte characters (such as Chinese)
  assert(shellCountPrefixOnes(0xE0) == 3);   // 11100000 → 3 leading 1s → Return 3
  assert(shellCountPrefixOnes(0xEF) == 3);   // 11101111 → 3 leading 1s → Return 3

  // test four byte characters (such as emojis)
  assert(shellCountPrefixOnes(0xF0) == 4);   // 11110000 → 4 leading 1s → Return 4
  assert(shellCountPrefixOnes(0xF7) == 4);   // 11110111 → 4 leading 1s → Return 4

  // test for illegal UTF-8 first byte (only verifying leading 1 count)
  assert(shellCountPrefixOnes(0xF8) == 5);   // 11111000 → 5 leading 1s → Return 5
  assert(shellCountPrefixOnes(0xFF) == 8);   // 11111111 → 8个 leading 1s → Return 8
}

TEST(shellUtil, trimStr) {
    struct Case { const char* in; const char* expect; } cases[] = {
      {"3.0.8.0 trial ", "3.0.8.0"},    // remove "trial" and trailing spaces
      {"3.0.8.0 trial",  "3.0.8.0"},    // remove "trial"
      {"3.0.8.0   ",     "3.0.8.0"},   // remove trailing spaces
      {"3.0.8.0",        "3.0.8.0"},   // no change
      {"trial",          ""},           // exact match "trial" -> empty
      {"trial ",         ""},           // "trial" followed by a space
      {" trial",         ""},           // leading space + "trial"
      {"3.0.8.0,trial",  "3.0.8.0,"},   // remove ",trial"
    };

    for (const auto& c : cases) {
        char buf[64] = {0};
        strncpy(buf, c.in, sizeof(buf) - 1);
        trimStr(buf, "trial");
        EXPECT_STREQ(c.expect, buf) << "input: '" << c.in << "'";
    }
}


int main(int argc, char **argv) {
  testing::InitGoogleTest(&argc, argv);
  return RUN_ALL_TESTS();
}