/* 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_object.c

 *

 * Test Object Allocation, toString, Equals, Destruction

 *

 */

#include "testutil.h"

#include "testutil_nss.h"

static void *plContext = NULL;

static PKIX_Error *

destructor(

        /* ARGSUSED */ PKIX_PL_Object *object,

        /* ARGSUSED */ void *plContext)

{

        (void) printf("\tUser defined destructor called\n");

        return (NULL);

}

static PKIX_Error*

toStringCallback(

        PKIX_PL_Object *obj,

        PKIX_PL_String **pString,

        /* ARGSUSED */ void* plContext) {

        PKIX_Error *errorResult;

        PKIX_UInt32 type;

        char *format = "(addr: %x, type: %d)";

        PKIX_PL_String *formatString = NULL;

        errorResult = PKIX_PL_String_Create(

                                        PKIX_ESCASCII,

                                        format,

                                        PL_strlen(format),

                                        &formatString,

                                        plContext);

        if (errorResult) testError("PKIX_PL_String_Create failed");

        if (pString == plContext)

                testError("Null String");

        type = (unsigned int)0;

        (void) PKIX_PL_Object_GetType(obj, &type, plContext);

        errorResult = PKIX_PL_Sprintf(pString, plContext,

                                    formatString,

                                    (int)obj, type);

        if (errorResult) testError("PKIX_PL_Sprintf failed");

        errorResult = PKIX_PL_Object_DecRef((PKIX_PL_Object*)formatString,

                                        plContext);

        if (errorResult) testError("PKIX_PL_Object_DecRef failed");

        return (NULL);

}

static PKIX_Error *

comparator(

        PKIX_PL_Object *first,

        PKIX_PL_Object *second,

        PKIX_Int32 *pValue,

        /* ARGSUSED */ void *plContext)

{

        if (*(char *)first > *(char *)second)

                *pValue = 1;

        else if (*(char *)first < *(char *)second)

                *pValue = -1;

        else

                *pValue = 0;

        return (NULL);

}

static PKIX_Error *

hashcodeCallback(

        PKIX_PL_Object *object,

        PKIX_UInt32 *pValue,

        /* ARGSUSED */ void *plContext)

{

        *pValue = 123456789;

        return (NULL);

}

static PKIX_Error*

equalsCallback(

        PKIX_PL_Object *first,

        PKIX_PL_Object *second,

        PKIX_Boolean *result,

        void* plContext) {

        PKIX_UInt32 firstType = 0, secondType = 0;

        if ((first == plContext)||(second == plContext))

                testError("Null Object");

        (void) PKIX_PL_Object_GetType(first, &firstType, plContext);

        (void) PKIX_PL_Object_GetType(second, &secondType, plContext);

        *result = (firstType == secondType)?PKIX_TRUE:PKIX_FALSE;

        return (NULL);

}

static void

createObjects(

        PKIX_PL_Object **obj,

        PKIX_PL_Object **obj2,

        PKIX_PL_Object **obj3,

        PKIX_PL_Object **obj4)

{

        PKIX_TEST_STD_VARS();

#ifdef PKIX_USER_OBJECT_TYPE

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_RegisterType

                    (1000, /* type */

                    "thousand", /* description */

                    NULL, /* destructor */

                    NULL, /* equals */

                    (PKIX_PL_HashcodeCallback)hashcodeCallback,

                    NULL,  /* toString */

                    NULL,  /* Comparator */

                    NULL,

                    plContext));

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_Alloc

                    (1000, /* type */

                    12, /* size */

                    obj,

                    plContext));

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_RegisterType

                    (2000, /* type */

                    "two thousand" /* description */,

                    (PKIX_PL_DestructorCallback)destructor,

                    (PKIX_PL_EqualsCallback)equalsCallback,

                    NULL, /* hashcode */

                    (PKIX_PL_ToStringCallback)toStringCallback,

                    (PKIX_PL_ComparatorCallback)comparator,

                    NULL,

                    plContext));

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_Alloc

                    (2000, /* type */

                    1, /* size */

                    obj2,

                    plContext));

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_Alloc

                    (2000, /* type */

                    1, /* size */

                    obj4,

                    plContext));

        *obj3 = *obj;

        PKIX_TEST_EXPECT_NO_ERROR(PKIX_PL_Object_IncRef(*obj3, plContext));

cleanup:

#endif /* PKIX_USER_OBJECT_TYPE */

        PKIX_TEST_RETURN();

}

static void

testGetType(

            PKIX_PL_Object *obj,

            PKIX_PL_Object *obj2,

            PKIX_PL_Object *obj3)

{

        PKIX_UInt32 testType;

        PKIX_TEST_STD_VARS();

        PKIX_TEST_EXPECT_NO_ERROR

                (PKIX_PL_Object_GetType(obj, &testType, plContext));

        if (testType != 1000)

                testError("Object 1 returned the wrong type");

        PKIX_TEST_EXPECT_NO_ERROR

                (PKIX_PL_Object_GetType(obj2, &testType, plContext));

        if (testType != 2000)

                testError("Object 2 returned the wrong type");

        PKIX_TEST_EXPECT_NO_ERROR

                (PKIX_PL_Object_GetType(obj3, &testType, plContext));

        if (testType != 1000)

                testError("Object 3 returned the wrong type");

cleanup:

        PKIX_TEST_RETURN();

}

static void

testCompare(

            PKIX_PL_Object *obj2,

            PKIX_PL_Object *obj4)

{

        PKIX_Int32 cmpResult;

        PKIX_TEST_STD_VARS();

        *(char *)obj2 = 0x20;

        *(char *)obj4 = 0x10;

        PKIX_TEST_EXPECT_NO_ERROR

                (PKIX_PL_Object_Compare(obj2, obj4, &cmpResult, plContext));

        if (cmpResult <= 0) testError("Invalid Result from Object Compare");

        PKIX_TEST_EXPECT_NO_ERROR

                (PKIX_PL_Object_Compare(obj4, obj2, &cmpResult, plContext));

        if (cmpResult >= 0) testError("Invalid Result from Object Compare");

        PKIX_TEST_EXPECT_NO_ERROR

                (PKIX_PL_Object_Compare(obj4, obj4, &cmpResult, plContext));

        *(char *)obj2 = 0x10;

        if (cmpResult != 0) testError("Invalid Result from Object Compare");

cleanup:

        PKIX_TEST_RETURN();

}

static void

testDestroy(

            PKIX_PL_Object *obj,

            PKIX_PL_Object *obj2,

            PKIX_PL_Object *obj3,

            PKIX_PL_Object *obj4)

{

        PKIX_TEST_STD_VARS();

        PKIX_TEST_DECREF_BC(obj);

        PKIX_TEST_DECREF_BC(obj2);

        PKIX_TEST_DECREF_BC(obj3);

        PKIX_TEST_DECREF_BC(obj4);

cleanup:

        PKIX_TEST_RETURN();

}

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

#ifdef PKIX_USER_OBJECT_TYPE

        PKIX_PL_Object *obj, *obj2, *obj3, *obj4;

        PKIX_UInt32 actualMinorVersion;

        PKIX_UInt32 j = 0;

        PKIX_TEST_STD_VARS();

        startTests("Objects");

        PKIX_TEST_EXPECT_NO_ERROR(

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

        subTest("PKIX_PL_Object_Create");

        createObjects(&obj, &obj2, &obj3, &obj4);

        PKIX_TEST_EQ_HASH_TOSTR_DUP(obj, obj3, obj2, NULL, Object, PKIX_FALSE);

        subTest("PKIX_PL_Object_GetType");

        testGetType(obj, obj2, obj3);

        subTest("PKIX_PL_Object_Compare");

        testCompare(obj2, obj4);

        subTest("PKIX_PL_Object_Destroy");

        testDestroy(obj, obj2, obj3, obj4);

cleanup:

        PKIX_Shutdown(plContext);

        endTests("Objects");

#endif /* PKIX_USER_OBJECT_TYPE */

        return (0);

}