/*
 * 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 <string>

#include <log4cxx/helpers/charsetdecoder.h>
#include <log4cxx/helpers/exception.h>
#include <log4cxx/helpers/fileinputstream.h>
#include <log4cxx/helpers/inputstreamreader.h>
#include <log4cxx/helpers/pool.h>
#include <log4cxx/helpers/transcoder.h>
#include <log4cxx/helpers/loglog.h>
#include <log4cxx/logstring.h>

#include "logunit.h"

#define LOG4CXX_TEST 1
#include <log4cxx/private/log4cxx_private.h>

//
// If there is no support for wchar_t logging then
// there is not a consistent way to get the test characters compared.
//
#if LOG4CXX_WCHAR_T_API

using namespace log4cxx;
using namespace log4cxx::helpers;

/**
 * Tests support for decoding specification.
 *
 * The lib provides multiple different decoders and decides which to use by default on compile time.
 * This test uses the same checks like {@link CharsetDecoder#createDefaultDecoder} to decide which
 * checks to run actually, so that some input file with some encoded text according to the decoder
 * in use is read and the contents compared to some witness. Because of different decoders not all
 * files have the same content, e.g. the ASCII-only decoder can't deal with Unicode chars obviously.
 *
 * This test is based on encodingtest and uses that witnesses, especially the hard coded strings for
 * greeting and pi. We only combine it into one in the former mentioned order, divided by a space.
 *
 * @see LOGCXX-369
 * @see LOGCXX-399
 */
LOGUNIT_CLASS(DecodingTest)
{
	LOGUNIT_TEST_SUITE(DecodingTest);
#if LOG4CXX_CHARSET_USASCII
	LOGUNIT_TEST(testASCII);
#elif LOG4CXX_CHARSET_ISO88591 || defined(_WIN32_WCE)
	LOGUNIT_TEST(testLatin1);
#elif LOG4CXX_CHARSET_UTF8
	LOGUNIT_TEST(testUtf8);
#elif LOG4CXX_LOGCHAR_IS_WCHAR && LOG4CXX_HAS_MBSRTOWCS
	LOGUNIT_TEST(testUtf8_to_wchar);
	LOGUNIT_TEST(testUtf16);
	LOGUNIT_TEST(testUtf16LE);
	LOGUNIT_TEST(testUtf16BE);
#else
	// LocaleCharsetDecoder, so it's difficult to provide a file working for e.g. windows-1252
	// as well as something completely different.
	LOGUNIT_TEST(testASCII);
#endif
	LOGUNIT_TEST_SUITE_END();
public:
	/**
	 * Test us-ascii decoding.
	 */
	void testASCII()
	{
		const wchar_t witness[] = { L'A', 0x003F, 0x003F, 0x003F, 0x003F, 0x003F, 0x0020, 0x003F, 0 };

		testImpl(LOG4CXX_STR("ascii.txt"), witness);

	}

	/**
	 * Test iso-8859-1 decoding.
	 */
	void testLatin1()
	{
		const wchar_t witness[] = { L'A', 0x003F, 0x003F, 0x003F, 0x003F, 0x003F, 0x0020, 0x00B9, 0 };

		testImpl(LOG4CXX_STR("latin1.txt"), witness);
	}

	/**
	 * Test utf-8 decoding.
	 */
	void testUtf8()
	{
		const wchar_t witness[] = { L'A', 0x0605, 0x0530, 0x986, 0x4E03, 0x0400, 0x0020, 0x00B9, 0 };

		testImpl(LOG4CXX_STR("UTF-8.txt"), witness);
	}

	/**
	 * Test utf-8 decoding.
	 */
	void testUtf8_to_wchar()
	{
		const wchar_t witness[] = { L'A', 0x0605, 0x0530, 0x986, 0x4E03, 0x0400, 0x0020, 0x00B9, 0 };

		try
		{
			std::locale::global(std::locale("en_US.UTF-8"));
			testImpl(LOG4CXX_STR("UTF-8.txt"), witness);
		}
		catch (std::runtime_error& ex)
		{
			LogString msg;
			Transcoder::decode(ex.what(), msg);
			msg.append(LOG4CXX_STR(": "));
			msg.append(LOG4CXX_STR("en_US.UTF-8"));
			LogLog::warn(msg);
		}
	}

	/**
	 * Test utf-16 decoding.
	 */
	void testUtf16()
	{
		const wchar_t witness[] = { L'A', 0x0605, 0x0530, 0x986, 0x4E03, 0x0400, 0x0020, 0x00B9, 0 };

		try
		{
			std::locale::global(std::locale("en_US.UTF-16"));
			testImpl(LOG4CXX_STR("UTF-16.txt"), witness);
		}
		catch (std::runtime_error& ex)
		{
			LogString msg;
			Transcoder::decode(ex.what(), msg);
			msg.append(LOG4CXX_STR(": "));
			msg.append(LOG4CXX_STR("en_US.UTF-16"));
			LogLog::warn(msg);
		}
	}

	/**
	 * Test utf-16be decoding.
	 */
	void testUtf16BE()
	{
		const wchar_t witness[] = { L'A', 0x0605, 0x0530, 0x986, 0x4E03, 0x0400, 0x0020, 0x00B9, 0 };
		try
		{
			std::locale::global(std::locale("en_US.UTF-16BE"));
			testImpl(LOG4CXX_STR("UTF-16BE.txt"), witness);
		}
		catch (std::runtime_error& ex)
		{
			LogString msg;
			Transcoder::decode(ex.what(), msg);
			msg.append(LOG4CXX_STR(": "));
			msg.append(LOG4CXX_STR("en_US.UTF-16BE"));
			LogLog::warn(msg);
		}
	}

	/**
	 * Test utf16-le decoding.
	 */
	void testUtf16LE()
	{
		const wchar_t witness[] = { L'A', 0x0605, 0x0530, 0x986, 0x4E03, 0x0400, 0x0020, 0x00B9, 0 };
		try
		{
			std::locale::global(std::locale("en_US.UTF-16LE"));
			testImpl(LOG4CXX_STR("UTF-16LE.txt"), witness);
		}
		catch (std::exception& ex)
		{
			LogString msg;
			Transcoder::decode(ex.what(), msg);
			msg.append(LOG4CXX_STR(": "));
			msg.append(LOG4CXX_STR("en_US.UTF-16LE"));
			LogLog::warn(msg);
		}
	}

private:
	void testImpl(
		const LogString & fileName,
		const wchar_t*   witness)
	{
		CharsetDecoderPtr decoder(CharsetDecoder::getDefaultDecoder());
		LogString         lsContent;
		std::wstring      wsContent;
		LogString         path(LOG4CXX_STR("input/decoding/") + fileName);
		Pool              pool;

		FileInputStreamPtr   fis(     new FileInputStream(path));
		InputStreamReaderPtr isReader(new InputStreamReader(fis, decoder));

		lsContent.assign(isReader->read(pool));
		Transcoder::encode(lsContent, wsContent);

		LOGUNIT_ASSERT_EQUAL((std::wstring) witness, wsContent);
	}
};

LOGUNIT_TEST_SUITE_REGISTRATION(DecodingTest);

#endif