// Copyright (c) 2017, Baidu.com, Inc. All Rights Reserved

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

#include <stdlib.h>
#include <stdio.h>
#include <iostream>

#include <gtest/gtest.h>
#include "util/debug-util.h"

using namespace std;

namespace impala {

string RecursionStack(int level) {
    if (level == 0) {
        return GetStackTrace();
    }

    return RecursionStack(level - 1);
}

TEST(DebugUtil, StackDump) {
    cout << "Stack: " << endl << GetStackTrace() << endl;
    cout << "Stack Recursion: " << endl << RecursionStack(5) << endl;
}

TEST(DebugUtil, QueryIdParsing) {
    TUniqueId id;
    EXPECT_FALSE(ParseId("abcd", &id));
    EXPECT_FALSE(ParseId("abcdabcdabcdabcdabcdabcdabcdabcda", &id));
    EXPECT_FALSE(ParseId("zbcdabcdabcdabcd:abcdabcdabcdabcd", &id));
    EXPECT_FALSE(ParseId("~bcdabcdabcdabcd:abcdabcdabcdabcd", &id));
    EXPECT_FALSE(ParseId("abcdabcdabcdabcd:!bcdabcdabcdabcd", &id));

    EXPECT_TRUE(ParseId("abcdabcdabcdabcd:abcdabcdabcdabcd", &id));
    EXPECT_EQ(id.hi, 0xabcdabcdabcdabcd);
    EXPECT_EQ(id.lo, 0xabcdabcdabcdabcd);

    EXPECT_TRUE(ParseId("abcdabcdabcdabcd:1234abcdabcd5678", &id));
    EXPECT_EQ(id.hi, 0xabcdabcdabcdabcd);
    EXPECT_EQ(id.lo, 0x1234abcdabcd5678);

    EXPECT_TRUE(ParseId("cdabcdabcdabcd:1234abcdabcd5678", &id));
    EXPECT_EQ(id.hi, 0xcdabcdabcdabcd);
    EXPECT_EQ(id.lo, 0x1234abcdabcd5678);

    EXPECT_TRUE(ParseId("cdabcdabcdabcd:abcdabcd5678", &id));
    EXPECT_EQ(id.hi, 0xcdabcdabcdabcd);
    EXPECT_EQ(id.lo, 0xabcdabcd5678);
}

}

int main(int argc, char** argv) {
    std::string conffile = std::string(getenv("PALO_HOME")) + "/conf/be.conf";
    if (!palo::config::init(conffile.c_str(), false)) {
        fprintf(stderr, "error read config file. \n");
        return -1;
    }
    init_glog("be-test");
    ::testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}