// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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/url_coding.h"
#include "util/logging.h"

namespace doris {

// Tests encoding/decoding of input.  If expected_encoded is non-empty, the
// encoded string is validated against it.
void test_url(const string& input, const string& expected_encoded, bool hive_compat) {
    std::string intermediate;
    url_encode(input, &intermediate, hive_compat);
    string output;

    if (!expected_encoded.empty()) {
        EXPECT_EQ(intermediate, expected_encoded);
    }

    EXPECT_TRUE(UrlDecode(intermediate, &output, hive_compat));
    EXPECT_EQ(input, output);

    // Convert string to vector and try that also
    vector<uint8_t> input_vector;
    input_vector.resize(input.size());
    memcpy(&input_vector[0], input.c_str(), input.size());
    string intermediate2;
    url_encode(input_vector, &intermediate2, hive_compat);
    EXPECT_EQ(intermediate, intermediate2);
}

void test_base64(const string& input, const string& expected_encoded) {
    string intermediate;
    Base64Encode(input, &intermediate);
    string output;

    if (!expected_encoded.empty()) {
        EXPECT_EQ(intermediate, expected_encoded);
    }

    EXPECT_TRUE(Base64Decode(intermediate, &output));
    EXPECT_EQ(input, output);

    // Convert string to vector and try that also
    vector<uint8_t> input_vector;
    input_vector.resize(input.size());
    memcpy(&input_vector[0], input.c_str(), input.size());
    string intermediate2;
    Base64Encode(input_vector, &intermediate2);
    EXPECT_EQ(intermediate, intermediate2);
}

// Test URL encoding. Check that the values that are put in are the
// same that come out.
TEST(UrlCodingTest, Basic) {
    std::string input = "ABCDEFGHIJKLMNOPQRSTUWXYZ1234567890~!@#$%^&*()<>?,./:\";'{}|[]\\_+-=";
    test_url(input, "", false);
    test_url(input, "", true);
}

TEST(UrlCodingTest, HiveExceptions) {
    test_url(" +", " +", true);
}

TEST(UrlCodingTest, BlankString) {
    test_url("", "", false);
    test_url("", "", true);
}

TEST(UrlCodingTest, PathSeparators) {
    test_url("/home/doris/directory/", "%2Fhome%2Fdoris%2Fdirectory%2F", false);
    test_url("/home/doris/directory/", "%2Fhome%2Fdoris%2Fdirectory%2F", true);
}

TEST(Base64Test, Basic) {
    test_base64("a", "YQ==");
    test_base64("ab", "YWI=");
    test_base64("abc", "YWJj");
    test_base64("abcd", "YWJjZA==");
    test_base64("abcde", "YWJjZGU=");
    test_base64("abcdef", "YWJjZGVm");
}

TEST(HtmlEscapingTest, Basic) {
    string before = "<html><body>&amp";
    stringstream after;
    EscapeForHtml(before, &after);
    EXPECT_EQ(after.str(), "&lt;html&gt;&lt;body&gt;&amp;amp");
}

}

int main(int argc, char** argv) {
    std::string conffile = std::string(getenv("DORIS_HOME")) + "/conf/be.conf";
    if (!doris::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();
}
