#include "StdAfx.h"
#include "UnitTestFrameWork.h"
#include "TestP4BridgeServerUtf8.h"
 
#include "..\p4bridge\P4BridgeClient.h"
#include "..\p4bridge\P4BridgeServer.h"
 
#include <conio.h>
 
CREATE_TEST_SUITE(TestP4BridgeServerUtf8)
 
TestP4BridgeServerUtf8::TestP4BridgeServerUtf8(void)
{
    UnitTestSuite::RegisterTest(ServerConnectionTest, "ServerConnectionTest");
    UnitTestSuite::RegisterTest(TestNonUnicodeClientToUnicodeServer, "TestNonUnicodeClientToUnicodeServer");
    UnitTestSuite::RegisterTest(TestUntaggedCommand, "TestUntaggedCommand");
    UnitTestSuite::RegisterTest(TestUnicodeUserName, "TestUnicodeUserName");
    UnitTestSuite::RegisterTest(TestTaggedCommand, "TestTaggedCommand");
    UnitTestSuite::RegisterTest(TestTextOutCommand, "TestTextOutCommand");
    UnitTestSuite::RegisterTest(TestBinaryOutCommand, "TestBinaryOutCommand");
    UnitTestSuite::RegisterTest(TestErrorOutCommand, "TestErrorOutCommand");
}
 
 
TestP4BridgeServerUtf8::~TestP4BridgeServerUtf8(void)
{
}
 
char unitTestDir8[MAX_PATH];
char unitTestZip8[MAX_PATH];
char * TestDir8 = "c:\\MyTestDir";
char * TestZip8 = "c:\\MyTestDir\\u.exe";
char * rcp_cmd8 = "p4d -r C:/MyTestDir -jr checkpoint.1";
char * udb_cmd8 = "p4d -r C:/MyTestDir -xu";
char * p4d_cmd8 = "p4d -p6666 -IdUnitTestServer -rC:/MyTestDir";
//char * p4d_xi_cmd8 = "p4d -xi";
 
void * pi8 = NULL;
 
bool TestP4BridgeServerUtf8::Setup()
{
    // remove the test directory if it exists
    UnitTestSuite::rmDir( TestDir8 ) ;
 
    GetCurrentDirectory(sizeof(unitTestDir8), unitTestDir8);
 
    strcpy( unitTestZip8, unitTestDir8);
    strcat( unitTestZip8, "\\u.exe");
 
    if (!CreateDirectory( TestDir8, NULL)) return false;
 
    if (!CopyFile(unitTestZip8, TestZip8, false)) return false;
 
    if (!SetCurrentDirectory(TestDir8)) return false;
 
    pi8= UnitTestSuite::RunProgram("u", TestDir8, true, true);
    if (!pi8) 
	{
		SetCurrentDirectory(unitTestDir8);
		return false;
	}
 
    delete pi8;
 
    pi8 = UnitTestSuite::RunProgram(rcp_cmd8, TestDir8, true, true);
    if (!pi8) 
	{
		SetCurrentDirectory(unitTestDir8);
		return false;
	}
 
    delete pi8;
 
    pi8 = UnitTestSuite::RunProgram(udb_cmd8, TestDir8, true, true);
    if (!pi8) 
	{
		SetCurrentDirectory(unitTestDir8);
		return false;
	}
 
    delete pi8;
 
    //server deployed by u.ex is already in Unicode mode
    //pi8 = UnitTestSuite::RunProgram(p4d_xi_cmd8, TestDir8, false, true);
    //if (!pi8) return false;
 
    //delete pi8;
 
    pi8 = UnitTestSuite::RunProgram(p4d_cmd8, TestDir8, false, false);
    if (!pi8) 
	{
		SetCurrentDirectory(unitTestDir8);
		return false;
	}
 
//    _getch();
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TearDown(char* testName)
{
    if (pi8)
        UnitTestSuite::EndProcess( (LPPROCESS_INFORMATION) pi8 );
 
    SetCurrentDirectory(unitTestDir8);
 
    UnitTestSuite::rmDir( TestDir8 ) ;
 
	p4base::PrintMemoryState(testName);
 
    return true;
}
 
bool TestP4BridgeServerUtf8::ServerConnectionTest()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    delete ps;
 
	return true;
}
 
bool TestP4BridgeServerUtf8::TestNonUnicodeClientToUnicodeServer()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "admin_space");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    
	ASSERT_TRUE(ps->unicodeServer());
 
    char* params[1];
    params[0] = "//depot/mycode/*";
 
    ASSERT_FALSE(ps->run_command("files", 5, 0, params, 1))
 
    P4ClientError * out = ps->get_ui(5)->GetErrorResults();
 
    ASSERT_STRING_STARTS_WITH(out->Message, "Unicode server permits only unicode enabled clients.")
   
    delete ps;
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TestUntaggedCommand()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "admin_space");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    char* params[1];
    params[0] = "//depot/mycode/*";
 
    ASSERT_TRUE(ps->run_command("files", 7, 0, params, 1))
 
    P4ClientInfoMsg * out = ps->get_ui(7)->GetInfoResults();
 
    ASSERT_STRING_EQUAL(out->Message, "//depot/MyCode/ReadMe.txt#1 - add change 1 (text)")
	ASSERT_NOT_NULL(out->Next)
    ASSERT_STRING_EQUAL(out->Next->Message, "//depot/MyCode/Silly.bmp#1 - add change 1 (binary)")
	ASSERT_NOT_NULL(out->Next->Next)
    ASSERT_STRING_EQUAL(out->Next->Next->Message, "//depot/MyCode/\xD0\x9F\xD1\x8E\xD0\xBF.txt#1 - add change 3 (utf16)")
 
    delete ps;
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TestUnicodeUserName()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    //Алексей = "\xD0\x90\xD0\xbb\xD0\xB5\xD0\xBA\xD1\x81\xD0\xB5\xD0\xB9\0" IN utf-8
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "\xD0\x90\xD0\xBB\xD0\xB5\xD0\xBA\xD1\x81\xD0\xB5\xD0\xB9\0", "pass", "\xD0\x90\xD0\xbb\xD0\xB5\xD0\xBA\xD1\x81\xD0\xB5\xD0\xB9\0");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    char* params[1];
    params[0] = "//depot/mycode/*";
 
    ASSERT_TRUE(ps->run_command("files", 7, 0, params, 1))
 
    P4ClientInfoMsg * out = ps->get_ui(7)->GetInfoResults();
 
	ASSERT_STRING_EQUAL(out->Message, "//depot/MyCode/ReadMe.txt#1 - add change 1 (text)")
	ASSERT_NOT_NULL(out->Next)
	ASSERT_STRING_EQUAL(out->Next->Message, "//depot/MyCode/Silly.bmp#1 - add change 1 (binary)")
	ASSERT_NOT_NULL(out->Next->Next)
    ASSERT_STRING_EQUAL(out->Next->Next->Message, "//depot/MyCode/\xD0\x9F\xD1\x8E\xD0\xBF.txt#1 - add change 3 (utf16)")
 
    delete ps;
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TestTaggedCommand()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "admin_space");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    char* params[1];
    params[0] = "//depot/mycode/*";
 
    ASSERT_TRUE(ps->run_command("files", 7, 1, params, 1))
 
    StrDictListIterator * out = ps->get_ui(7)->GetTaggedOutput();
 
    ASSERT_NOT_NULL(out);
 
    int itemCnt = 0;
    while (StrDictList * pItem = out->GetNextItem())
    {
        int entryCnt = 0;
 
        while (KeyValuePair * pEntry = out->GetNextEntry())
        {
            if ((itemCnt == 0) && (strcmp(pEntry->key, "depotFile") == 0))
                ASSERT_STRING_EQUAL(pEntry->value, "//depot/MyCode/ReadMe.txt")
            if ((itemCnt == 1) && (strcmp(pEntry->key, "depotFile") == 0))
                ASSERT_STRING_EQUAL(pEntry->value, "//depot/MyCode/Silly.bmp")
            if ((itemCnt == 2) && (strcmp(pEntry->key, "depotFile") == 0))
                ASSERT_STRING_EQUAL(pEntry->value, "//depot/MyCode/\xD0\x9F\xD1\x8E\xD0\xBF.txt")
            entryCnt++;
        }
        ASSERT_NOT_EQUAL(entryCnt, 0);
        itemCnt++;
    }
    ASSERT_EQUAL(itemCnt, 3);
 
    delete out;
 
	delete ps;
 
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TestTextOutCommand()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "admin_space");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    char* params[1];
    params[0] = "//depot/MyCode/ReadMe.txt";
 
    ASSERT_TRUE(ps->run_command("print", 7, 1, params, 1))
 
    StrBuf * out = ps->get_ui(7)->GetTextResults();
 
    ASSERT_NOT_NULL(out);
 
    ASSERT_STRING_EQUAL(out->Text(), "Don't Read This!\n\nIt's Secret!")
 
    delete ps;
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TestBinaryOutCommand()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "admin_space");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    char* params[1];
    params[0] = "//depot/MyCode/Silly.bmp";
 
    ASSERT_TRUE(ps->run_command("print", 3, 1, params, 1))
 
    int cnt = ps->get_ui(3)->GetBinaryResultsCount();
 
    ASSERT_EQUAL(cnt, 3126)
 
    void * out = ps->get_ui(3)->GetBinaryResults();
 
    ASSERT_NOT_NULL(out);
    ASSERT_EQUAL((*(((unsigned char*)out) + 1)), 0x4d)
 
    delete ps;
 
    return true;
}
 
bool TestP4BridgeServerUtf8::TestErrorOutCommand()
{
    P4ClientError* connectionError = NULL;
    // create a new server
    P4BridgeServer * ps = new P4BridgeServer("localhost:6666", "admin", "", "admin_space");
    ASSERT_NOT_NULL(ps);
 
    // connect and see if the api returned an error. 
    if( !ps->connected( &connectionError ) )
    {
        char buff[256];
        sprintf(buff, "Connection error: %s", *connectionError);
        // Abort if the connect did not succeed
        ASSERT_FAIL(buff);
    }
 
    ASSERT_TRUE(ps->unicodeServer());
    ps->set_charset("utf8", "utf16le");
 
    char* params[1];
    params[0] = "//depot/MyCode/Billy.bmp";
 
    // run a command against a nonexistent file
    // Should fail
    ASSERT_FALSE(ps->run_command("rent", 88, 1, params, 1))
 
    P4ClientError * out = ps->get_ui(88)->GetErrorResults();
 
    ASSERT_NOT_NULL(out);
 
	ASSERT_EQUAL(out->ErrorCode, 805379098);
    ASSERT_NULL(out->Next)
 
    delete ps;
 
    return true;
}