// clang-format off
#include "pch.h"
#include "unit_test_helper.h"
#include "it_odbc_helper.h"
#include "version.h"
#include <codecvt>
#ifndef WIN32
#include <climits>
#endif
// clang-format on

class TestSQLGetInfo : public testing::Test {
   public:
    TestSQLGetInfo() {
    }

    void SetUp() {
        ITDriverConnect((SQLTCHAR*)conn_string.c_str(), &m_env, &m_conn, true,
                        true);
    }

    void TearDown() {
        SQLDisconnect(m_conn);
        SQLFreeHandle(SQL_HANDLE_ENV, m_env);
    }

    ~TestSQLGetInfo() {
        // cleanup any pending stuff, but no exceptions allowed
    }

    SQLHENV m_env = SQL_NULL_HENV;
    SQLHDBC m_conn = SQL_NULL_HDBC;
};

// 1 for ver1 >= ver2, 0 for ver1 < ver2, -1 for error
int Ver1GEVer2(std::wstring ver_1_str, std::wstring ver_2_str) {
    auto VersionSplit = [&](std::vector< unsigned long >& output,
                            std::wstring& input, std::wstring delim) {
        try {
            size_t start = 0;
            size_t end = input.find(delim);
            while (end != std::string::npos) {
                output.push_back(std::stoul(input.substr(start, end - start)));
                start = end + delim.length();
                end = input.find(delim, start);
            }
            output.push_back(std::stoul(input.substr(start, end)));
        } catch (...) {
            output.clear();
        }
    };

    std::vector< unsigned long > ver_1;
    std::vector< unsigned long > ver_2;
    VersionSplit(ver_1, ver_1_str, L".");
    VersionSplit(ver_2, ver_2_str, L".");
    if ((ver_1.size() == 0) || (ver_2.size() == 0))
        return -1;

    size_t cnt = ((ver_1.size() < ver_2.size()) ? ver_1.size() : ver_2.size());
    for (size_t i = 0; i < cnt; i++) {
        if (ver_1[i] != ver_2[i])
            return (ver_1[i] >= ver_2[i]) ? 1 : 0;
    }
    if (ver_1.size() != ver_2.size())
        return (ver_1.size() > ver_2.size()) ? 1 : 0;

    // They are identical
    return 1;
}

// Test template for SQLGetInfo
#define TEST_SQL_GET_INFO_STRING(test_name, info_type, expected_value)         \
    TEST_F(TestSQLGetInfo, test_name) {                                        \
        SQLTCHAR info_value_ptr[1024];                                         \
        SQLSMALLINT string_length_ptr;                                         \
        SQLRETURN ret =                                                        \
            SQLGetInfo(m_conn, info_type, info_value_ptr,                      \
                       IT_SIZEOF(info_value_ptr), &string_length_ptr);         \
        LogAnyDiagnostics(SQL_HANDLE_DBC, m_conn, ret);                        \
        EXPECT_EQ(std::wstring(info_value_ptr), std::wstring(expected_value)); \
    }

// Test template for SQLGetInfo
#define TEST_SQL_GET_INFO_VERSION_GE(test_name, info_type, expected_value) \
    TEST_F(TestSQLGetInfo, test_name) {                                    \
        SQLTCHAR info_value_ptr[1024];                                     \
        SQLSMALLINT string_length_ptr;                                     \
        SQLRETURN ret =                                                    \
            SQLGetInfo(m_conn, info_type, info_value_ptr,                  \
                       IT_SIZEOF(info_value_ptr), &string_length_ptr);     \
        LogAnyDiagnostics(SQL_HANDLE_DBC, m_conn, ret);                    \
        EXPECT_EQ(Ver1GEVer2(info_value_ptr, expected_value), 1);          \
    }

// Test template for SQLGetInfo
#define TEST_SQL_GET_INFO_UINT_MASK(test_name, info_type, expected_value) \
    TEST_F(TestSQLGetInfo, test_name) {                                   \
        SQLUINTEGER info_value_ptr;                                       \
        SQLSMALLINT string_length_ptr;                                    \
        SQLRETURN ret =                                                   \
            SQLGetInfo(m_conn, info_type, &info_value_ptr,                \
                       sizeof(info_value_ptr), &string_length_ptr);       \
        LogAnyDiagnostics(SQL_HANDLE_DBC, m_conn, ret);                   \
        EXPECT_EQ((size_t)info_value_ptr, (size_t)expected_value);        \
    }

// Test template for SQLGetInfo
#define TEST_SQL_GET_INFO_UINT16(test_name, info_type, expected_value) \
    TEST_F(TestSQLGetInfo, test_name) {                                \
        SQLUSMALLINT info_value_ptr;                                   \
        SQLSMALLINT string_length_ptr;                                 \
        SQLRETURN ret =                                                \
            SQLGetInfo(m_conn, info_type, &info_value_ptr,             \
                       sizeof(info_value_ptr), &string_length_ptr);    \
        LogAnyDiagnostics(SQL_HANDLE_DBC, m_conn, ret);                \
        EXPECT_EQ(info_value_ptr, expected_value);                     \
    }

/////////////////
// Driver Info //
/////////////////

#if defined(WIN32)
TEST_SQL_GET_INFO_STRING(SQLDriverName, SQL_DRIVER_NAME, L"sqlodbc.dll");
#elif defined(__APPLE__)
TEST_SQL_GET_INFO_STRING(SQLDriverName, SQL_DRIVER_NAME, L"libsqlodbc.dylib");
#endif
TEST_SQL_GET_INFO_STRING(SQLDriverODBCVer, SQL_DRIVER_ODBC_VER, L"03.51");

std::wstring version =
    std::wstring_convert< std::codecvt_utf8_utf16< wchar_t >, wchar_t >{}
        .from_bytes(OPENSEARCHDRIVERVERSION);
TEST_SQL_GET_INFO_STRING(SQLDriverVer, SQL_DRIVER_VER, version);

TEST_SQL_GET_INFO_UINT16(SQLGetDataExtensions, SQL_GETDATA_EXTENSIONS,
                         (SQL_GD_ANY_COLUMN | SQL_GD_ANY_ORDER | SQL_GD_BOUND
                          | SQL_GD_BLOCK));
TEST_SQL_GET_INFO_STRING(SQLSearchPatternEscape, SQL_SEARCH_PATTERN_ESCAPE,
                         L"");

//////////////////////
// Data Source Info //
//////////////////////

TEST_SQL_GET_INFO_UINT16(SQLCursorCommitBehavior, SQL_CURSOR_COMMIT_BEHAVIOR,
                         SQL_CB_CLOSE);
TEST_SQL_GET_INFO_UINT16(SQLTxnCapable, SQL_TXN_CAPABLE, SQL_TC_NONE);
TEST_SQL_GET_INFO_UINT16(SQLConcatNullBehavior, SQL_CONCAT_NULL_BEHAVIOR,
                         SQL_CB_NULL);
TEST_SQL_GET_INFO_STRING(SQLSchemaTerm, SQL_SCHEMA_TERM, L"");
TEST_SQL_GET_INFO_STRING(SQLCatalogTerm, SQL_CATALOG_TERM, L"");

///////////////
// DBMS Info //
///////////////

TEST_SQL_GET_INFO_STRING(SQLDBMSName, SQL_DBMS_NAME, L"OpenSearch");
TEST_SQL_GET_INFO_VERSION_GE(SQLDBMSVer, SQL_DBMS_VER, L"7.1.1");

///////////////////
// Supported SQL //
///////////////////

TEST_SQL_GET_INFO_STRING(SQLColumnAlias, SQL_COLUMN_ALIAS, L"Y");
TEST_SQL_GET_INFO_UINT16(SQLGroupBy, SQL_GROUP_BY,
                         SQL_GB_GROUP_BY_CONTAINS_SELECT);
TEST_SQL_GET_INFO_STRING(SQLIdentifierQuoteChar, SQL_IDENTIFIER_QUOTE_CHAR,
                         L"`");
TEST_SQL_GET_INFO_UINT_MASK(SQLOJCapabilities, SQL_OJ_CAPABILITIES,
                            SQL_OJ_LEFT | SQL_OJ_RIGHT | SQL_OJ_NOT_ORDERED
                                | SQL_OJ_ALL_COMPARISON_OPS);
TEST_SQL_GET_INFO_UINT_MASK(SQLSchemaUsage, SQL_SCHEMA_USAGE, 0);
TEST_SQL_GET_INFO_UINT16(SQLQuotedIdentifierCase, SQL_QUOTED_IDENTIFIER_CASE,
                         SQL_IC_SENSITIVE);
TEST_SQL_GET_INFO_STRING(SQLSpecialCharacters, SQL_SPECIAL_CHARACTERS, L"_");
TEST_SQL_GET_INFO_UINT_MASK(SQLODBCInterfaceConformance,
                            SQL_ODBC_INTERFACE_CONFORMANCE, SQL_OIC_CORE);
TEST_SQL_GET_INFO_UINT_MASK(SQLSQLConformance, SQL_SQL_CONFORMANCE,
                            SQL_SC_SQL92_ENTRY);
TEST_SQL_GET_INFO_UINT_MASK(SQLCatalogUsage, SQL_CATALOG_USAGE, 0);
TEST_SQL_GET_INFO_UINT16(SQLCatalogLocation, SQL_CATALOG_LOCATION, 0);
TEST_SQL_GET_INFO_STRING(SQLCatalogNameSeparator, SQL_CATALOG_NAME_SEPARATOR,
                         L"");
TEST_SQL_GET_INFO_UINT_MASK(SQLSQL92Predicates, SQL_SQL92_PREDICATES,
                            SQL_SP_BETWEEN | SQL_SP_COMPARISON | SQL_SP_IN
                                | SQL_SP_ISNULL | SQL_SP_LIKE);
TEST_SQL_GET_INFO_UINT_MASK(SQLSQL92RelationalJoinOperators,
                            SQL_SQL92_RELATIONAL_JOIN_OPERATORS,
                            SQL_SRJO_CROSS_JOIN | SQL_SRJO_INNER_JOIN
                                | SQL_SRJO_LEFT_OUTER_JOIN
                                | SQL_SRJO_RIGHT_OUTER_JOIN);
TEST_SQL_GET_INFO_UINT_MASK(SQLSQL92ValueExpressions,
                            SQL_SQL92_VALUE_EXPRESSIONS,
                            SQL_SVE_CASE | SQL_SVE_CAST);
TEST_SQL_GET_INFO_UINT_MASK(SQLDatetimeLiterals, SQL_DATETIME_LITERALS, 0);
TEST_SQL_GET_INFO_STRING(SQLOrderByColumnsInSelect,
                         SQL_ORDER_BY_COLUMNS_IN_SELECT, L"Y");
TEST_SQL_GET_INFO_STRING(SQLCatalogName, SQL_CATALOG_NAME, L"N");

////////////////
// Conversion //
////////////////

TEST_SQL_GET_INFO_UINT_MASK(SQLConvertInteger, SQL_CONVERT_INTEGER, SQL_CVT_INTEGER | SQL_CVT_REAL | SQL_CVT_DOUBLE | SQL_CVT_BIT | SQL_CVT_BIGINT | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertSmallint, SQL_CONVERT_SMALLINT, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertTinyint, SQL_CONVERT_TINYINT, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertBit, SQL_CONVERT_BIT, SQL_CVT_INTEGER | SQL_CVT_REAL | SQL_CVT_DOUBLE | SQL_CVT_BIT | SQL_CVT_BIGINT | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertVarchar, SQL_CONVERT_VARCHAR, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertBigint, SQL_CONVERT_BIGINT, SQL_CVT_INTEGER | SQL_CVT_REAL | SQL_CVT_DOUBLE | SQL_CVT_BIT | SQL_CVT_BIGINT | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertDecimal, SQL_CONVERT_DECIMAL, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertDouble, SQL_CONVERT_DOUBLE, SQL_CVT_INTEGER | SQL_CVT_REAL | SQL_CVT_DOUBLE | SQL_CVT_BIT | SQL_CVT_BIGINT | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertFloat, SQL_CONVERT_FLOAT, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertNumeric, SQL_CONVERT_NUMERIC, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertReal, SQL_CONVERT_REAL, SQL_CVT_INTEGER | SQL_CVT_REAL | SQL_CVT_DOUBLE | SQL_CVT_BIT | SQL_CVT_BIGINT | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertDate, SQL_CONVERT_DATE, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertTime, SQL_CONVERT_TIME, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertTimestamp, SQL_CONVERT_TIMESTAMP, SQL_CVT_TIMESTAMP | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertBinary, SQL_CONVERT_BINARY, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertLongvarbinary, SQL_CONVERT_LONGVARBINARY,
                            0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertVarbinary, SQL_CONVERT_VARBINARY, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertChar, SQL_CONVERT_CHAR, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertLongVarchar, SQL_CONVERT_LONGVARCHAR, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertWChar, SQL_CONVERT_WCHAR, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertWLongVarchar, SQL_CONVERT_WLONGVARCHAR,
                            0);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertWVarchar, SQL_CONVERT_WVARCHAR, SQL_CVT_INTEGER | SQL_CVT_REAL | SQL_CVT_DOUBLE | SQL_CVT_BIGINT | SQL_CVT_WVARCHAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLConvertGuid, SQL_CONVERT_GUID, 0);

//////////////////////
// Scalar Functions //
//////////////////////

TEST_SQL_GET_INFO_UINT_MASK(SQLConvertFunctions, SQL_CONVERT_FUNCTIONS,
                            SQL_FN_CVT_CAST);
TEST_SQL_GET_INFO_UINT_MASK(SQLNumericFunctions, SQL_NUMERIC_FUNCTIONS,
                            SQL_FN_NUM_ABS | SQL_FN_NUM_ATAN | SQL_FN_NUM_ATAN2
                                | SQL_FN_NUM_COS | SQL_FN_NUM_COT
                                | SQL_FN_NUM_DEGREES | SQL_FN_NUM_FLOOR
                                | SQL_FN_NUM_LOG | SQL_FN_NUM_LOG10
                                | SQL_FN_NUM_PI | SQL_FN_NUM_POWER
                                | SQL_FN_NUM_RADIANS | SQL_FN_NUM_ROUND
                                | SQL_FN_NUM_SIGN | SQL_FN_NUM_SIN
                                | SQL_FN_NUM_SQRT | SQL_FN_NUM_TAN);
TEST_SQL_GET_INFO_UINT_MASK(SQLStringFunctions, SQL_STRING_FUNCTIONS,
                            SQL_FN_STR_ASCII | SQL_FN_STR_LENGTH
                                | SQL_FN_STR_LTRIM | SQL_FN_STR_REPLACE
                                | SQL_FN_STR_RTRIM | SQL_FN_STR_SUBSTRING);
TEST_SQL_GET_INFO_UINT_MASK(SQLSystemFunctions, SQL_SYSTEM_FUNCTIONS,
                            SQL_FN_SYS_IFNULL);
TEST_SQL_GET_INFO_UINT_MASK(SQLTimedateAddIntervals, SQL_TIMEDATE_ADD_INTERVALS,
                            0);
TEST_SQL_GET_INFO_UINT_MASK(SQLTimedateDiffIntervals,
                            SQL_TIMEDATE_DIFF_INTERVALS, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLTimedateFunctions, SQL_TIMEDATE_FUNCTIONS,
                            SQL_FN_TD_CURDATE | SQL_FN_TD_DAYOFMONTH
                                | SQL_FN_TD_MONTH | SQL_FN_TD_MONTHNAME
                                | SQL_FN_TD_NOW | SQL_FN_TD_YEAR);
TEST_SQL_GET_INFO_UINT_MASK(SQLSQL92DatetimeFunctions,
                            SQL_SQL92_DATETIME_FUNCTIONS, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLSQL92NumericValueFunctions,
                            SQL_SQL92_NUMERIC_VALUE_FUNCTIONS, 0);
TEST_SQL_GET_INFO_UINT_MASK(SQLSQL92StringFunctions, SQL_SQL92_STRING_FUNCTIONS,
                            SQL_SSF_LOWER | SQL_SSF_UPPER);

////////////
// Limits //
////////////

TEST_SQL_GET_INFO_UINT16(SQLMaxIdentifierLen, SQL_MAX_IDENTIFIER_LEN, SHRT_MAX);
TEST_SQL_GET_INFO_UINT16(SQLMaxColumnsInGroupBy, SQL_MAX_COLUMNS_IN_GROUP_BY,
                         0);
TEST_SQL_GET_INFO_UINT16(SQLMaxColumnsInOrderBy, SQL_MAX_COLUMNS_IN_ORDER_BY,
                         0);
TEST_SQL_GET_INFO_UINT16(SQLMaxColumnsInSelect, SQL_MAX_COLUMNS_IN_SELECT, 0);

int main(int argc, char** argv) {
#ifdef __APPLE__
    // Enable malloc logging for detecting memory leaks.
    system("export MallocStackLogging=1");
#endif
    testing::internal::CaptureStdout();
    ::testing::InitGoogleTest(&argc, argv);

    int failures = RUN_ALL_TESTS();

    std::string output = testing::internal::GetCapturedStdout();
    std::cout << output << std::endl;
    std::cout << (failures ? "Not all tests passed." : "All tests passed")
              << std::endl;
    WriteFileIfSpecified(argv, argv + argc, "-fout", output);

#ifdef __APPLE__
    // Disable malloc logging and report memory leaks
    system("unset MallocStackLogging");
    system("leaks itodbc_info > leaks_itodbc_info");
#endif
    return failures;
}