/*****************************************************************************/
// Haiku Translation Kit Test
// Author: Michael Wilber
// Version:
//
// This is the Test application for BTranslationUtils
//
//
// This application and all source files used in its construction, except 
// where noted, are licensed under the MIT License, and have been written 
// and are:
//
// Copyright (c) 2002 Haiku Project
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense, 
// and/or sell copies of the Software, and to permit persons to whom the 
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included 
// in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
/*****************************************************************************/
#include "TranslationUtilsTest.h"
#include <TranslatorFormats.h>
        // for B_TRANSLATOR_EXT_*
#include <TranslatorRoster.h>
#include <Application.h>
#include <TextView.h>
#include <Menu.h>
#include <MenuItem.h>
#include <Bitmap.h>
#include <BitmapStream.h>
#include <Entry.h>
#include <OS.h>

#include <stdio.h>
#include <string.h>

/* cppunit framework */
#include <cppunit/Test.h>
#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>

/**
 * Default constructor - no work
 */
TranslationUtilsTest::TranslationUtilsTest(std::string name)
        : BTestCase(name)
{
}

/**
 * Default destructor - no work
 */
TranslationUtilsTest::~TranslationUtilsTest()
{
}

CppUnit::Test *
TranslationUtilsTest::Suite()
{
        /* create our suite */
        CppUnit::TestSuite *suite = new CppUnit::TestSuite("TranslationUtils");
                
        /* add suckers */
        suite->addTest(new CppUnit::TestCaller<TranslationUtilsTest>("TranslationUtilsTest::GetBitmap Test", &TranslationUtilsTest::GetBitmapTest));
        suite->addTest(new CppUnit::TestCaller<TranslationUtilsTest>("TranslationUtilsTest::GetPutStyledText Test", &TranslationUtilsTest::GetPutStyledTextTest));
        suite->addTest(new CppUnit::TestCaller<TranslationUtilsTest>("TranslationUtilsTest::GetDefaultSettings Test", &TranslationUtilsTest::GetDefaultSettingsTest));
        suite->addTest(new CppUnit::TestCaller<TranslationUtilsTest>("TranslationUtilsTest::AddTranslationItems Test", &TranslationUtilsTest::AddTranslationItemsTest));

        return suite;
}

void
CheckBitmap(BBitmap *pbits)
{
        CPPUNIT_ASSERT(pbits);
        CPPUNIT_ASSERT(pbits->Bits());
        CPPUNIT_ASSERT(pbits->BitsLength() == 443904);
        CPPUNIT_ASSERT(pbits->BytesPerRow() == 1536);
        CPPUNIT_ASSERT(pbits->Bounds().IntegerWidth() == 383);
        CPPUNIT_ASSERT(pbits->Bounds().IntegerHeight() == 288);
}

void
TranslationUtilsTest::GetBitmapTest()
{
        // File
        NextSubTest();
        BApplication app(
                "application/x-vnd.Haiku-translationkit_translationutilstest");
        BBitmap *pbits = NULL;
        pbits = BTranslationUtils::GetBitmap(
                "resources/kits/translation/image.png");
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
        
        // File (passing it a file that isn't actually there)
        pbits = BTranslationUtils::GetBitmap("/tmp/no-file-here.bmp");
        CPPUNIT_ASSERT(pbits == NULL);
        
        // File (GetBitmapFile)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmapFile(
                "resources/kits/translation/image.png");
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
        
        // File (passing GetBitmapFile a file that isn't there)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmapFile("/tmp/no-file-here.bmp");
        CPPUNIT_ASSERT(pbits == NULL);
        
        // File (entry_ref)
        NextSubTest();
        entry_ref ref;
        BEntry bent(
                "resources/kits/translation/image.png");
        CPPUNIT_ASSERT(bent.InitCheck() == B_OK);
        CPPUNIT_ASSERT(bent.GetRef(&ref) == B_OK);
        pbits = BTranslationUtils::GetBitmap(&ref);
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
        
        // File (NULL entry_ref)
        NextSubTest();
        entry_ref *pref = NULL;
        pbits = BTranslationUtils::GetBitmap(pref);
        CPPUNIT_ASSERT(pbits == NULL);
        
        // Resource
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap("res_image");
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
        
        // Resource (bad resource name)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap("Michael Wilber");
        CPPUNIT_ASSERT(pbits == NULL);
        
        // Resource by Type & Id
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap(
                B_TRANSLATOR_BITMAP, 246);
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
        
        // Resource by Type & Id (wrong resource type)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap(B_TRANSLATOR_TEXT, 246);
        CPPUNIT_ASSERT(pbits == NULL);
        
        // Resource by Type & Id (wrong resource Id)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap(B_TRANSLATOR_BITMAP,
                166);
        CPPUNIT_ASSERT(pbits == NULL);
        
        // Resource by Type & Name
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap(
                B_TRANSLATOR_BITMAP, "res_image");
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
        
        // Resource by Type & Name (wrong resource type)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap(B_TRANSLATOR_TEXT,
                "res_image");
        CPPUNIT_ASSERT(pbits == NULL);
        
        // Resource by Type & Name (wrong resource name)
        NextSubTest();
        pbits = BTranslationUtils::GetBitmap(B_TRANSLATOR_BITMAP,
                "Michael Wilber");
        CPPUNIT_ASSERT(pbits == NULL);
        
        // Stream (open file, translate to BBitmapStream,
        // pass that BBitmapStream to GetBitmap)
        NextSubTest();
        BFile imgfile(
                "resources/kits/translation/image.png",
                B_READ_ONLY);
        CPPUNIT_ASSERT(imgfile.InitCheck() == B_OK);
        BTranslatorRoster *proster = BTranslatorRoster::Default();
        CPPUNIT_ASSERT(proster);
        BBitmapStream stream;
        CPPUNIT_ASSERT(proster->Translate(&imgfile, NULL, NULL,
                &stream, B_TRANSLATOR_BITMAP) == B_OK);
        pbits = BTranslationUtils::GetBitmap(&stream);
        CheckBitmap(pbits);
        delete pbits;
        pbits = NULL;
}

void
TranslationUtilsTest::GetPutStyledTextTest()
{
        // Insert text into view with styles OFF
        NextSubTest();
        BApplication app(
                "application/x-vnd.Haiku-translationkit_translationutilstest");
        BTextView *ptextview = NULL;
        ptextview = new BTextView(BRect(0, 0, 100, 100),
                "utilstest_textview", BRect(0, 0, 100, 100),
                B_FOLLOW_ALL_SIDES, B_WILL_DRAW | B_PULSE_NEEDED);
        CPPUNIT_ASSERT(ptextview);
        ptextview->SetStylable(false);
        BFile stylfile(
                "resources/kits/translation/sentence.stxt",
                B_READ_ONLY);
        CPPUNIT_ASSERT(stylfile.InitCheck() == B_OK);
        CPPUNIT_ASSERT(BTranslationUtils::GetStyledText(&stylfile, ptextview) == B_OK);
        CPPUNIT_ASSERT(ptextview->TextLength() == 77);
        CPPUNIT_ASSERT(strcmp(ptextview->Text(),
                "Ask not what open source can do for you, ask what you can do for open source.") == 0);
        
        // Insert text into view with styles ON
        NextSubTest();
        ptextview->SetText("");
        CPPUNIT_ASSERT(ptextview->TextLength() == 0);
        ptextview->SetStylable(true);
        CPPUNIT_ASSERT(BTranslationUtils::GetStyledText(&stylfile, ptextview) == B_OK);
        CPPUNIT_ASSERT(ptextview->TextLength() == 77);
        CPPUNIT_ASSERT(strcmp(ptextview->Text(),
                "Ask not what open source can do for you, ask what you can do for open source.") == 0);
                
        // Write current contents of the BTextView to a file, and make
        // sure that it matches the source file exactly
        NextSubTest();
        BFile tmpfile("/tmp/out_styled_text.stxt", B_READ_WRITE | B_CREATE_FILE | B_ERASE_FILE);
        CPPUNIT_ASSERT(tmpfile.InitCheck() == B_OK);
        CPPUNIT_ASSERT(BTranslationUtils::PutStyledText(ptextview, &tmpfile) == B_OK);
        off_t stylsize = 0, tmpsize = 0;
        CPPUNIT_ASSERT(stylfile.GetSize(&stylsize) == B_OK);
        CPPUNIT_ASSERT(tmpfile.GetSize(&tmpsize) == B_OK);
        CPPUNIT_ASSERT(stylsize == tmpsize);
        char *stylbuf = NULL, *tmpbuf = NULL;
        stylbuf = new char[stylsize];
        tmpbuf = new char[tmpsize];
        CPPUNIT_ASSERT(stylbuf && tmpbuf);
        CPPUNIT_ASSERT(stylfile.ReadAt(0, stylbuf, stylsize) == stylsize);
        CPPUNIT_ASSERT(tmpfile.ReadAt(0, tmpbuf, tmpsize) == tmpsize);
//      CPPUNIT_ASSERT(memcmp(stylbuf, tmpbuf, stylsize) == 0);
        delete[] stylbuf;
        delete[] tmpbuf;
        stylbuf = NULL;
        tmpbuf = NULL;
        
        // Send NULL pointers to GetStyledText
        NextSubTest();
        CPPUNIT_ASSERT(BTranslationUtils::GetStyledText(NULL, ptextview) == B_BAD_VALUE);
        CPPUNIT_ASSERT(BTranslationUtils::GetStyledText(&stylfile, NULL) == B_BAD_VALUE);
        
        // Send NULL pointers to PutStyledText
        NextSubTest();
        CPPUNIT_ASSERT(BTranslationUtils::PutStyledText(NULL, &tmpfile) == B_BAD_VALUE);
        CPPUNIT_ASSERT(BTranslationUtils::PutStyledText(ptextview, NULL) == B_BAD_VALUE);
}

void
TranslationUtilsTest::GetDefaultSettingsTest()
{
        // Test translator_id version with a BTranslatorRoster * supplied
        NextSubTest();
        BApplication app(
                "application/x-vnd.Haiku-translationkit_translationutilstest");
        BMessage *pmsg = NULL;
        bool bdummy = false;
        translator_id *pids = NULL;
        int32 ncount = 0;
        BTranslatorRoster *proster = BTranslatorRoster::Default();
        CPPUNIT_ASSERT(proster);
        CPPUNIT_ASSERT(proster->GetAllTranslators(&pids, &ncount) == B_OK);
        CPPUNIT_ASSERT(pids && ncount > 0);
        pmsg = BTranslationUtils::GetDefaultSettings(pids[0], proster);
        CPPUNIT_ASSERT(pmsg);
        delete pmsg;
        pmsg = NULL;
        
        // Test translator_id version without a BTranslatorRoster supplied
        NextSubTest();
        pmsg = BTranslationUtils::GetDefaultSettings(pids[0]);
        CPPUNIT_ASSERT(pmsg);
        delete pmsg;
        pmsg = NULL;
        delete[] pids;
        pids = NULL;
        
        // Get settings from the Haiku TGATranslator and ensure that
        // all of its settings are there
//      NextSubTest();
//      pmsg = BTranslationUtils::GetDefaultSettings("TGA Images", 100);
//      CPPUNIT_ASSERT(pmsg);
//      CPPUNIT_ASSERT(pmsg->FindBool(B_TRANSLATOR_EXT_HEADER_ONLY,
//              &bdummy) == B_OK);
//      CPPUNIT_ASSERT(pmsg->FindBool(B_TRANSLATOR_EXT_DATA_ONLY,
//              &bdummy) == B_OK);
//      CPPUNIT_ASSERT(pmsg->FindBool("tga /rle", &bdummy) == B_OK);
//      delete pmsg;
        pmsg = NULL;
        
        // Try to get a translator that isn't there
        NextSubTest();
        pmsg = BTranslationUtils::GetDefaultSettings("Michael Wilber", 0);
        CPPUNIT_ASSERT(pmsg == NULL);
        
        // Try to get a version of a translator that we don't have
        NextSubTest();
        pmsg = BTranslationUtils::GetDefaultSettings("PPM Images", 1);
        CPPUNIT_ASSERT(pmsg == NULL);
}

void
TranslationUtilsTest::AddTranslationItemsTest()
{
        // Make menu of bitmap translators
        NextSubTest();
        BApplication app(
                "application/x-vnd.Haiku-translationkit_translationutilstest");
        BMenu *pmenu = new BMenu("utilstest_menu");
        CPPUNIT_ASSERT(pmenu);
        CPPUNIT_ASSERT(BTranslationUtils::AddTranslationItems(pmenu,
                B_TRANSLATOR_BITMAP) == B_OK);
        CPPUNIT_ASSERT(pmenu->CountItems() > 0);
        
        // Make menu of text translators
        NextSubTest();
        int32 nitem = 0;
        while (pmenu->CountItems())
                delete pmenu->RemoveItem(nitem);
        CPPUNIT_ASSERT(pmenu->CountItems() == 0);
        CPPUNIT_ASSERT(BTranslationUtils::AddTranslationItems(pmenu,
                B_TRANSLATOR_TEXT) == B_OK);
        CPPUNIT_ASSERT(pmenu->CountItems() > 0);
        
        // Make menu of 1 translator
        NextSubTest();
        while (pmenu->CountItems())
                delete pmenu->RemoveItem(nitem);
        CPPUNIT_ASSERT(pmenu->CountItems() == 0);
        BTranslatorRoster *proster = new BTranslatorRoster();
        CPPUNIT_ASSERT(proster);
        CPPUNIT_ASSERT(proster->AddTranslators(
                "/boot/system/add-ons/Translators/PPMTranslator"
                ) == B_OK);
        CPPUNIT_ASSERT(BTranslationUtils::AddTranslationItems(pmenu,
                B_TRANSLATOR_BITMAP, NULL, NULL, NULL, proster) == B_OK);
//      CPPUNIT_ASSERT(pmenu->CountItems() == 1); // FIXME
        delete proster;
        proster = NULL;
        
        delete pmenu;
        pmenu = NULL;
        
        // Bad Input
        NextSubTest();
        CPPUNIT_ASSERT(BTranslationUtils::AddTranslationItems(NULL,
                B_TRANSLATOR_BITMAP) == B_BAD_VALUE);
}