/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
/*
 * test_logger.c
 *
 * Tests Logger Objects
 *
 */

#include "testutil.h"
#include "testutil_nss.h"

static void *plContext = NULL;

static char *levels[] = {
        "None",
        "Fatal Error",
        "Error",
        "Warning",
        "Debug",
        "Trace"
};

static
PKIX_Error *testLoggerCallback(
        PKIX_Logger *logger,
        PKIX_PL_String *message,
        PKIX_UInt32 logLevel,
        PKIX_ERRORCLASS logComponent,
        void *plContext)
{
        char *comp = NULL;
        char *msg = NULL;
        char result[100];
        static int callCount = 0;

        PKIX_TEST_STD_VARS();

        msg = PKIX_String2ASCII(message, plContext);
        PR_snprintf(result, 100, "Logging %s (%s): %s",
                levels[logLevel], PKIX_ERRORCLASSNAMES[logComponent], msg);
        subTest(result);

        callCount++;
        if (callCount > 1) {
                testError("Incorrect number of Logger Callback <expect 1>");
        }

cleanup:

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Free(msg, plContext));
        PKIX_TEST_RETURN();
}

static
PKIX_Error *testLoggerCallback2(
        PKIX_Logger *logger,
        PKIX_PL_String *message,
        PKIX_UInt32 logLevel,
        PKIX_ERRORCLASS logComponent,
        void *plContext)
{
        char *comp = NULL;
        char *msg = NULL;
        char result[100];

        PKIX_TEST_STD_VARS();

        msg = PKIX_String2ASCII(message, plContext);
        PR_snprintf(result, 100, "Logging %s (%s): %s",
                levels[logLevel], PKIX_ERRORCLASSNAMES[logComponent], msg);
        subTest(result);

cleanup:
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Free(msg, plContext));
        PKIX_TEST_RETURN();
}

static void
createLogger(PKIX_Logger **logger,
        PKIX_PL_Object *context,
        PKIX_Logger_LogCallback cb)
{
        PKIX_TEST_STD_VARS();

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_Create
                (cb, context, logger, plContext));

cleanup:

        PKIX_TEST_RETURN();
}

static void
testContextCallback(PKIX_Logger *logger, PKIX_Logger *logger2)
{
        PKIX_Logger_LogCallback cb = NULL;
        PKIX_PL_Object *context = NULL;
        PKIX_Boolean cmpResult = PKIX_FALSE;
        PKIX_UInt32 length;

        PKIX_TEST_STD_VARS();

        subTest("PKIX_Logger_GetLoggerContext");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_GetLoggerContext
                (logger2, &context, plContext));

        testEqualsHelper
                ((PKIX_PL_Object *)logger, context, PKIX_TRUE, plContext);

        subTest("PKIX_Logger_GetLogCallback");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_GetLogCallback
                (logger, &cb, plContext));

        if (cb != testLoggerCallback) {
                testError("Incorrect Logger Callback returned");
        }

cleanup:

        PKIX_TEST_DECREF_AC(context);
        PKIX_TEST_RETURN();
}

static void
testComponent(PKIX_Logger *logger)
{
        PKIX_ERRORCLASS compName = (PKIX_ERRORCLASS)NULL;
        PKIX_ERRORCLASS compNameReturn = (PKIX_ERRORCLASS)NULL;
        PKIX_Boolean cmpResult = PKIX_FALSE;
        PKIX_TEST_STD_VARS();

        subTest("PKIX_Logger_GetLoggingComponent");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_GetLoggingComponent
                (logger, &compName, plContext));

        if (compName != (PKIX_ERRORCLASS)NULL) {
                testError("Incorrect Logger Component returned. expect <NULL>");
        }

        subTest("PKIX_Logger_SetLoggingComponent");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_SetLoggingComponent
                (logger, PKIX_LIST_ERROR, plContext));

        subTest("PKIX_Logger_GetLoggingComponent");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_GetLoggingComponent
                (logger, &compNameReturn, plContext));

        if (compNameReturn != PKIX_LIST_ERROR) {
                testError("Incorrect Logger Component returned.");
        }

cleanup:

        PKIX_TEST_RETURN();
}

static void
testMaxLoggingLevel(PKIX_Logger *logger)
{
        PKIX_UInt32 level = 0;
        PKIX_TEST_STD_VARS();

        subTest("PKIX_Logger_GetMaxLoggingLevel");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_GetMaxLoggingLevel
                (logger, &level, plContext));

        if (level != 0) {
                testError("Incorrect Logger MaxLoggingLevel returned");
        }

        subTest("PKIX_Logger_SetMaxLoggingLevel");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_SetMaxLoggingLevel
                (logger, 3, plContext));

        subTest("PKIX_Logger_GetMaxLoggingLevel");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_GetMaxLoggingLevel
                (logger, &level, plContext));

        if (level != 3) {
                testError("Incorrect Logger MaxLoggingLevel returned");
        }

cleanup:

        PKIX_TEST_RETURN();
}

static void
testLogger(PKIX_Logger *logger, PKIX_Logger *logger2)
{
        PKIX_List *loggerList = NULL;
        PKIX_List *checkList = NULL;
        PKIX_UInt32 length;
        PKIX_Boolean cmpResult = PKIX_FALSE;
        char *expectedAscii = "[\n"
                "\tLogger: \n"
                "\tContext:          (null)\n"
                "\tMaximum Level:    3\n"
                "\tComponent Name:   LIST\n"
                "]\n";


        PKIX_TEST_STD_VARS();

        subTest("PKIX_GetLoggers");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_GetLoggers(&loggerList, plContext));
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_GetLength
                (loggerList, &length, plContext));
        if (length != 0){
                testError("Incorrect Logger List returned");
        }
        PKIX_TEST_DECREF_BC(loggerList);

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_Create(&loggerList, plContext));
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_AppendItem
                (loggerList, (PKIX_PL_Object *) logger, plContext));

        subTest("PKIX_SetLoggers");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_SetLoggers(loggerList, plContext));

        subTest("PKIX_Logger_SetLoggingComponent");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_SetLoggingComponent
                (logger2, PKIX_MUTEX_ERROR, plContext));

        subTest("PKIX_Logger_SetMaxLoggingLevel");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_Logger_SetMaxLoggingLevel
                (logger2, 5, plContext));

        subTest("PKIX_AddLogger");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_AddLogger(logger2, plContext));

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_Create(&checkList, plContext));
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_AppendItem
                (checkList, (PKIX_PL_Object *) logger, plContext));
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_AppendItem
                (checkList, (PKIX_PL_Object *) logger2, plContext));

        PKIX_TEST_DECREF_BC(loggerList);

        subTest("PKIX_GetLoggers");
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_GetLoggers(&loggerList, plContext));
        PKIX_TEST_EXPECT_NO_ERROR(PKIX_List_GetLength
                (loggerList, &length, plContext));

        subTest("pkix_Loggers_Equals");
        testEqualsHelper
                ((PKIX_PL_Object *) loggerList,
                (PKIX_PL_Object *) checkList,
                PKIX_TRUE,
                plContext);

        subTest("pkix_Loggers_Duplicate");
        testDuplicateHelper((PKIX_PL_Object *)logger, plContext);

        subTest("pkix_Loggers_Hashcode");
        testHashcodeHelper((PKIX_PL_Object *) logger,
                           (PKIX_PL_Object *) logger,
                           PKIX_TRUE,
                           plContext);

        subTest("pkix_Loggers_ToString");
        testToStringHelper((PKIX_PL_Object *) logger, expectedAscii, plContext);

        subTest("PKIX Logger Callback");
        subTest("Expect to have ***Fatal Error (List): Null argument*** once");
        PKIX_TEST_EXPECT_ERROR(PKIX_List_AppendItem
                (NULL, (PKIX_PL_Object *) NULL, plContext));

cleanup:

        PKIX_TEST_DECREF_AC(loggerList);
        PKIX_TEST_DECREF_AC(checkList);
        PKIX_TEST_RETURN();
}

static void
testDestroy(PKIX_Logger *logger)
{
        PKIX_TEST_STD_VARS();

        PKIX_TEST_DECREF_BC(logger);

cleanup:

        PKIX_TEST_RETURN();
}

int test_logger(int argc, char *argv[]) {

        PKIX_Logger *logger, *logger2;
        PKIX_UInt32 actualMinorVersion;
        PKIX_UInt32 j = 0;

        PKIX_TEST_STD_VARS();

        startTests("Loggers");

        PKIX_TEST_EXPECT_NO_ERROR(
            PKIX_PL_NssContext_Create(0, PKIX_FALSE, NULL, &plContext));

        subTest("PKIX_Logger_Create");
        createLogger(&logger, NULL, testLoggerCallback);
        createLogger(&logger2, (PKIX_PL_Object *)logger, testLoggerCallback2);

        subTest("Logger Context and Callback");
        testContextCallback(logger, logger2);

        subTest("Logger Component");
        testComponent(logger);

        subTest("Logger MaxLoggingLevel");
        testMaxLoggingLevel(logger);

        subTest("Logger List operations");
        testLogger(logger, logger2);

        subTest("PKIX_Logger_Destroy");
        testDestroy(logger);
        testDestroy(logger2);

cleanup:

        PKIX_Shutdown(plContext);

        PKIX_TEST_RETURN();

        endTests("Loggers");

        return (0);

}