/*
 * 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.
 */
#if defined(_MSC_VER)
#pragma warning ( disable: 4231 4251 4275 4786 )
#endif

#include "logunit.h"

#include <apr_general.h>
#include <algorithm>
#include <stdlib.h>
#include <locale.h>

void initialize() {
    setlocale(LC_CTYPE, "");
    const char* ctype = setlocale(LC_CTYPE, 0);
    if (ctype == 0) {
        puts("LC_CTYPE: NULL");
    } else {
        printf("LC_CTYPE: %s\n", ctype);
    }
    apr_initialize();
}

extern const char** testlist;

static bool suite_sort(const LogUnit::SuiteList::value_type& lhs, const LogUnit::SuiteList::value_type& rhs) {
    return lhs.first < rhs.first;
}

abts_suite* abts_run_suites(abts_suite* suite) {
    LogUnit::SuiteList sorted(LogUnit::getAllSuites());

#if !defined(_MSC_VER)
    std::sort(sorted.begin(), sorted.end(), suite_sort);
#endif
    
    for(LogUnit::SuiteList::const_iterator iter = sorted.begin();
        iter != sorted.end();
        iter++) {
        //
        //   if there is an explicit testlist or if the suite is not by default disabled
        //         pump suite through filter
        if(testlist || !iter->second->isDisabled()) {
            suite = iter->second->run(suite);
        }
    }
    apr_terminate();
    return suite;
}

using namespace LogUnit;
using namespace std;

TestException::TestException() {}
TestException::TestException(const TestException& src) : std::exception(src) {
}

TestException& TestException::operator=(const TestException& src) {
    exception::operator=(src);
    return *this;
}


AssertException::AssertException(std::string message, int line) : msg(message), lineno(line) {}

AssertException::AssertException(bool expected, const char* actualExpr, int line) : msg(actualExpr), lineno(line) {
   if (expected) {
        msg.append(" was expected to be true, was false.");
   } else {
        msg.append(" was expected to be true, was false.");
   }
}

AssertException::AssertException(const AssertException& src) 
    : std::exception(src), 
      msg(src.msg), 
     lineno(src.lineno) {
}

AssertException::~AssertException() throw() {
}

AssertException& AssertException::operator=(const AssertException& src) {
    exception::operator=(src);
    msg = src.msg;
    lineno = src.lineno;
    return *this;
}

std::string AssertException::getMessage() const {
    return msg;
}

int AssertException::getLine() const {
    return lineno;
}


TestFixture::TestFixture() : tc(0) {}
TestFixture::~TestFixture() {}
void TestFixture::setCase(abts_case* newtc) {
      tc = newtc;
}
void TestFixture::setUp() {}
void TestFixture::tearDown() {}

void TestFixture::assertEquals(const char* expected, 
     const char* actual, 
     const char* expectedExpr,
     const char* actualExpr,
     int lineno) {
    abts_str_equal(tc, expected, actual, lineno);
    if ((expected == 0 || actual != 0) ||
        (expected != 0 || actual == 0) ||
        (expected != 0 && strcmp(expected, actual) != 0)) {
        throw TestException();
    }
}

void TestFixture::assertEquals(const std::string expected, 
     const std::string actual, 
     const char* expectedExpr,
     const char* actualExpr,
     int lineno) {
    abts_str_equal(tc, expected.c_str(), actual.c_str(), lineno);
    if (expected != actual) {
        throw TestException();
    }
}

template<class S>
static void transcode(std::string& dst, const S& src) {
        for(typename S::const_iterator iter = src.begin();
            iter != src.end();
            iter++) {
            if (*iter <= 0x7F) {
                dst.append(1, (char) *iter);
             } else {
                dst.append(1, '?');
            }
        }
}

#if LOG4CXX_LOGCHAR_IS_WCHAR || LOG4CXX_WCHAR_T_API       
void TestFixture::assertEquals(const std::wstring expected, 
    const std::wstring actual, 
    const char* expectedExpr,
    const char* actualExpr,
    int lineno) {
    if (expected != actual) {
        std::string exp, act;
      transcode(exp, expected);
      transcode(act, actual);
        abts_str_equal(tc, exp.c_str(), act.c_str(), lineno);
        throw TestException();
    }
}
#endif        
#if LOG4CXX_LOGCHAR_IS_UNICHAR || LOG4CXX_UNICHAR_API || LOG4CXX_CFSTRING_API
void TestFixture::assertEquals(const std::basic_string<log4cxx::UniChar> expected, 
     const std::basic_string<log4cxx::UniChar> actual,      
     const char* expectedExpr,
     const char* actualExpr,
int lineno) {
    if (expected != actual) {
        std::string exp, act;
      transcode(exp, expected);
      transcode(act, actual);
        abts_str_equal(tc, exp.c_str(), act.c_str(), lineno);
        throw TestException();
    }
}
#endif        


void TestFixture::assertEquals(const int expected, const int actual, int lineno) {
    abts_int_equal(tc, expected, actual, lineno);
    if (expected != actual) {
        throw TestException();
    }
}


LogUnit::TestSuite::TestSuite(const char* fname) : filename(fname), disabled(false) {
#if defined(_WIN32)
    for(size_t i = filename.find('\\');
        i != std::string::npos;
        i = filename.find('\\', i+1)) {
        filename.replace(i, 1, 1, '/');
    }
#endif
}

void LogUnit::TestSuite::addTest(const char*, test_func func) {
   test_funcs.push_back(func);
}

std::string LogUnit::TestSuite::getName() const {
   return filename;
}

void LogUnit::TestSuite::setDisabled(bool newVal) {
    disabled = newVal;
}

bool LogUnit::TestSuite::isDisabled() const {
    return disabled;
}

abts_suite* TestSuite::run(abts_suite* suite) const {
    suite = abts_add_suite(suite, filename.c_str());
    for(TestList::const_iterator iter = test_funcs.begin();
        iter != test_funcs.end();
        iter++) {
        abts_run_test(suite, *iter, NULL);
    }
    return suite;
}


LogUnit::SuiteList& LogUnit::getAllSuites() {
    static LogUnit::SuiteList allSuites;
    return allSuites;
}