#include "StdAfx.h"
#include "UnitTestFrameWork.h"
#include "TestP4BridgeServerLogging.h"
 
#include "..\p4bridge\P4BridgeClient.h"
#include "..\p4bridge\P4BridgeServer.h"
 
#include <strtable.h>
#include <strarray.h>
 
CREATE_TEST_SUITE(TestP4BridgeServerLogging)
 
TestP4BridgeServerLogging::TestP4BridgeServerLogging(void)
{
    UnitTestSuite::RegisterTest(LogMessageTest, "LogMessageTest");
    UnitTestSuite::RegisterTest(BadLogFnPtrTest, "BadLogFnPtrTest");
}
 
TestP4BridgeServerLogging::~TestP4BridgeServerLogging(void)
{
}
 
bool TestP4BridgeServerLogging::Setup()
{
    return true;
}
 
bool TestP4BridgeServerLogging::TearDown(char* testName)
{
	p4base::PrintMemoryState(testName);
    return true;
}
 
int _stdcall TestP4BridgeServerLogging::LogCallback(int level, const char* file, int line, const char* message)
{
	//int len = strlen(message);
	const char* msg = message;
 
	//int cnt = 0; 
 
	//while ((cnt++ < len) && (*msg++ != '>'));
 
	//// skip over the space
	//cnt++;
	//msg++;
 
	//if (cnt++ >= len) 
	//{
	//	return 0;
	//}
 
    if ((level == 3) && (strncmp( msg, "Info", 4 ) == 0))
        return 1;
 
    if ((level == 0) && (strncmp( msg, "Fatal:1", 7 ) == 0))
        return 1;
 
    if ((level == 42) && (strncmp( msg, "Debug:42", 8 ) == 0))
        return 1;
 
    return 0;
}
 
bool TestP4BridgeServerLogging::LogMessageTest()
{
    P4BridgeServer::SetLogCallFn(LogCallback);
 
    ASSERT_TRUE(LOG_INFO("Info"));
 
    ASSERT_TRUE(LOG_FATAL1("Fatal:%s", "1"));
 
    ASSERT_TRUE(LOG_DEBUG2(42, "Debug:%c%c", '4', '2'));
 
    return true;
}
 
bool TestP4BridgeServerLogging::BadLogFnPtrTest()
{
    P4BridgeServer::SetLogCallFn(0);
 
    ASSERT_FALSE(LOG_INFO("Info"));
 
    P4BridgeServer::SetLogCallFn((LogCallbackFn*) 0xFFFFFFFF);
 
    ASSERT_FALSE(LOG_INFO("Info"));
 
    P4BridgeServer::SetLogCallFn((LogCallbackFn*) 0x123456789ACDEF0);
 
    ASSERT_FALSE(LOG_INFO("Info"));
 
    return true;
}