/*
 * 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 <taoserror.h>
#include <iostream>
#pragma GCC diagnostic ignored "-Wwrite-strings"

#pragma GCC diagnostic ignored "-Wunused-function"
#pragma GCC diagnostic ignored "-Wunused-variable"
#pragma GCC diagnostic ignored "-Wsign-compare"

#include "../inc/clientInt.h"
#include "taos.h"

namespace {
void showDB(TAOS* pConn) {
  TAOS_RES* pRes = taos_query(pConn, "show databases");
  TAOS_ROW  pRow = NULL;

  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
  int32_t     numOfFields = taos_num_fields(pRes);

  char str[512] = {0};
  while ((pRow = taos_fetch_row(pRes)) != NULL) {
    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
    printf("%s\n", str);
  }
}
}  // namespace

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

TEST(testCase, driverInit_Test) { taos_init(); }

TEST(testCase, connect_Test) {
  TAOS* pConn = taos_connect("localhost", "root", "taosdata", "abc1", 0);
  if (pConn == NULL) {
    printf("failed to connect to server, reason:%s\n", taos_errstr(NULL));
  }
  sleep(3);
  taos_close(pConn);
}

//TEST(testCase, create_user_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "create user abc pass 'abc'");
//  if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
//    printf("failed to create user, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, create_account_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "create account aabc pass 'abc'");
//  if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
//    printf("failed to create user, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, drop_account_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "drop account aabc");
//  if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
//    printf("failed to create user, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, show_user_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "show users");
//  TAOS_ROW  pRow = NULL;
//
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, drop_user_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "drop user abc");
//  if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
//    printf("failed to create user, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, show_db_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "show databases");
//  TAOS_ROW  pRow = NULL;
//
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_close(pConn);
//}
//
//TEST(testCase, create_db_Test) {
  //TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
  //assert(pConn != NULL);

  //TAOS_RES* pRes = taos_query(pConn, "create database abc1 vgroups 2");
  //if (taos_errno(pRes) != 0) {
    //printf("error in create db, reason:%s\n", taos_errstr(pRes));
  //}

  //TAOS_FIELD* pFields = taos_fetch_fields(pRes);
  //ASSERT_TRUE(pFields == NULL);

  //int32_t numOfFields = taos_num_fields(pRes);
  //ASSERT_EQ(numOfFields, 0);

  //taos_free_result(pRes);

  //pRes = taos_query(pConn, "create database abc1 vgroups 4");
  //if (taos_errno(pRes) != 0) {
    //printf("error in create db, reason:%s\n", taos_errstr(pRes));
  //}
  //taos_close(pConn);
//}
//
//TEST(testCase, create_dnode_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "create dnode abc1 port 7000");
//  if (taos_errno(pRes) != 0) {
//    printf("error in create dnode, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "create dnode 1.1.1.1 port 9000");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to create dnode, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  taos_close(pConn);
//}
//
//TEST(testCase, drop_dnode_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "drop dnode 3");
//  if (taos_errno(pRes) != 0) {
//    printf("error in drop dnode, reason:%s\n", taos_errstr(pRes));
//  }
//
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  ASSERT_TRUE(pFields == NULL);
//
//  int32_t numOfFields = taos_num_fields(pRes);
//  ASSERT_EQ(numOfFields, 0);
//
//  pRes = taos_query(pConn, "drop dnode 4");
//  if (taos_errno(pRes) != 0) {
//    printf("error in drop dnode, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, use_db_test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("error in use db, reason:%s\n", taos_errstr(pRes));
//  }
//
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  ASSERT_TRUE(pFields == NULL);
//
//  int32_t numOfFields = taos_num_fields(pRes);
//  ASSERT_EQ(numOfFields, 0);
//
//  taos_close(pConn);
//}
//
// TEST(testCase, drop_db_test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  showDB(pConn);
//
//  TAOS_RES* pRes = taos_query(pConn, "drop database abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to drop db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  showDB(pConn);
//
//  pRes = taos_query(pConn, "create database abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("create to drop db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//  taos_close(pConn);
//}

//TEST(testCase, create_stable_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "create database if not exists abc1 vgroups 2");
//  if (taos_errno(pRes) != 0) {
//    printf("error in create db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "create table if not exists abc1.st1(ts timestamp, k int) tags(a int)");
//  if (taos_errno(pRes) != 0) {
//    printf("error in create stable, reason:%s\n", taos_errstr(pRes));
//  }
//
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  ASSERT_TRUE(pFields == NULL);
//
//  int32_t numOfFields = taos_num_fields(pRes);
//  ASSERT_EQ(numOfFields, 0);
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//  pRes = taos_query(pConn, "create stable if not exists abc1.`123_$^)` (ts timestamp, `abc` int) tags(a int)");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to create super table 123_$^), reason:%s\n", taos_errstr(pRes));
//  }
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  taos_free_result(pRes);
//  pRes = taos_query(pConn, "drop stable `123_$^)`");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to drop super table 123_$^), reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_close(pConn);
//}

TEST(testCase, create_table_Test) {
  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
  assert(pConn != NULL);

  TAOS_RES* pRes = taos_query(pConn, "use abc1");
  taos_free_result(pRes);

  pRes = taos_query(pConn, "create table if not exists tm0(ts timestamp, k int)");
  ASSERT_EQ(taos_errno(pRes), 0);

  taos_free_result(pRes);

  pRes = taos_query(pConn, "create table if not exists tm0(ts timestamp, k blob)");
  ASSERT_NE(taos_errno(pRes), 0);

  taos_free_result(pRes);
  taos_close(pConn);
}

//TEST(testCase, create_ctable_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to use db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "create table tm0 using st1 tags(1)");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to create child table tm0, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}

TEST(testCase, show_stable_Test) {
  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
  assert(pConn != nullptr);

//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to use db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);

  TAOS_RES* pRes = taos_query(pConn, "show abc1.stables");
  if (taos_errno(pRes) != 0) {
    printf("failed to show stables, reason:%s\n", taos_errstr(pRes));
    taos_free_result(pRes);
    ASSERT_TRUE(false);
  }

  TAOS_ROW    pRow = NULL;
  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
  int32_t     numOfFields = taos_num_fields(pRes);

  char str[512] = {0};
  while ((pRow = taos_fetch_row(pRes)) != NULL) {
    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
    printf("%s\n", str);
  }

  taos_free_result(pRes);
  taos_close(pConn);
}
//
//TEST(testCase, show_vgroup_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to use db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "show vgroups");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to show vgroups, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  TAOS_ROW pRow = NULL;
//
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, create_multiple_tables) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  ASSERT_NE(pConn, nullptr);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to use db, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    taos_close(pConn);
//    return;
//  }
//
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "create table t_2 using st1 tags(1)");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to create multiple tables, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  taos_free_result(pRes);
//  pRes = taos_query(pConn, "create table t_3 using st1 tags(2)");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to create multiple tables, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  TAOS_ROW    pRow = NULL;
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_free_result(pRes);
//
//  for (int32_t i = 0; i < 20; ++i) {
//    char sql[512] = {0};
//    snprintf(sql, tListLen(sql),
//             "create table t_x_%d using st1 tags(2) t_x_%d using st1 tags(5) t_x_%d using st1 tags(911)", i,
//             (i + 1) * 30, (i + 2) * 40);
//    TAOS_RES* pres = taos_query(pConn, sql);
//    if (taos_errno(pres) != 0) {
//      printf("failed to create table %d\n, reason:%s", i, taos_errstr(pres));
//    }
//    taos_free_result(pres);
//  }
//
//  taos_close(pConn);
//}
//
//TEST(testCase, show_table_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "show tables");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to show vgroups, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  TAOS_ROW    pRow = NULL;
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, drop_stable_Test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  assert(pConn != NULL);
//
//  TAOS_RES* pRes = taos_query(pConn, "create database abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("error in creating db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "use abc1");
//  if (taos_errno(pRes) != 0) {
//    printf("error in using db, reason:%s\n", taos_errstr(pRes));
//  }
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "drop stable st1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to drop stable, reason:%s\n", taos_errstr(pRes));
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}
//
//TEST(testCase, generated_request_id_test) {
//  SHashObj* phash = taosHashInit(10000, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
//
//  for (int32_t i = 0; i < 50000; ++i) {
//    uint64_t v = generateRequestId();
//    void*    result = taosHashGet(phash, &v, sizeof(v));
//    if (result != nullptr) {
//      printf("0x%lx, index:%d\n", v, i);
//    }
//    assert(result == nullptr);
//    taosHashPut(phash, &v, sizeof(v), NULL, 0);
//  }
//
//  taosHashCleanup(phash);
//}
//
TEST(testCase, create_topic_Test) {
  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
  assert(pConn != NULL);

  TAOS_RES* pRes = taos_query(pConn, "use abc1");
  if (taos_errno(pRes) != 0) {
    printf("error in use db, reason:%s\n", taos_errstr(pRes));
  }
  taos_free_result(pRes);

  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
  ASSERT_TRUE(pFields == nullptr);

  int32_t numOfFields = taos_num_fields(pRes);
  ASSERT_EQ(numOfFields, 0);

  taos_free_result(pRes);

  char* sql = "select * from tu";
  pRes = taos_create_topic(pConn, "test_topic_1", sql, strlen(sql));
  taos_free_result(pRes);
  taos_close(pConn);
}

//TEST(testCase, insert_test) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  ASSERT_NE(pConn, nullptr);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "insert into t_2 values(now, 1)");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to create multiple tables, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}

//TEST(testCase, projection_query_tables) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  ASSERT_NE(pConn, nullptr);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  taos_free_result(pRes);
//
////  pRes = taos_query(pConn, "create stable st1 (ts timestamp, k int) tags(a int)");
////  if (taos_errno(pRes) != 0) {
////    printf("failed to create table tu, reason:%s\n", taos_errstr(pRes));
////  }
////  taos_free_result(pRes);
////
////  pRes = taos_query(pConn, "create table tu using st1 tags(1)");
////  if (taos_errno(pRes) != 0) {
////    printf("failed to create table tu, reason:%s\n", taos_errstr(pRes));
////  }
////  taos_free_result(pRes);
////
////  for(int32_t i = 0; i < 100; ++i) {
////    char sql[512] = {0};
////    sprintf(sql, "insert into tu values(now+%da, %d)", i, i);
////    TAOS_RES* p = taos_query(pConn, sql);
////    if (taos_errno(p) != 0) {
////      printf("failed to insert data, reason:%s\n", taos_errstr(p));
////    }
////
////    taos_free_result(p);
////  }
//
//  pRes = taos_query(pConn, "select * from tu");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to select from table, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  TAOS_ROW    pRow = NULL;
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}

//TEST(testCase, projection_query_stables) {
//  TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
//  ASSERT_NE(pConn, nullptr);
//
//  TAOS_RES* pRes = taos_query(pConn, "use abc1");
//  taos_free_result(pRes);
//
//  pRes = taos_query(pConn, "select ts,k from m1");
//  if (taos_errno(pRes) != 0) {
//    printf("failed to select from table, reason:%s\n", taos_errstr(pRes));
//    taos_free_result(pRes);
//    ASSERT_TRUE(false);
//  }
//
//  TAOS_ROW    pRow = NULL;
//  TAOS_FIELD* pFields = taos_fetch_fields(pRes);
//  int32_t     numOfFields = taos_num_fields(pRes);
//
//  char str[512] = {0};
//  while ((pRow = taos_fetch_row(pRes)) != NULL) {
//    int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
//    printf("%s\n", str);
//  }
//
//  taos_free_result(pRes);
//  taos_close(pConn);
//}