/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:

*

*/

#include <stdio.h>

#include <string>

#include <x509certchain.h>

#include <memory>

#include "javastorage.h"

#include "javastorageexception.h"

#include "javastoragenames.h"

#include "javauid.h"

#include "javaprocessconstants.h"

#include "javasymbianoslayer.h"

#include "javaregistry.h"

#include "javaregistryentry.h"

#include "logger.h"

#include "mjavaattribute.h"

#include "storagetestutils.h"

#include "javacommonutils.h"

using namespace std;

using namespace Java;

using namespace java::storage;

using namespace java::util;

const JavaStorageApplicationEntry_t createEntry(const wstring& aUid,

        const wstring& aName,

        const wstring& aValue);

const JavaStorageApplicationEntry_t createValidCertsEntry(

    const wstring& aUid,

    const wstring& aCerts);

TEST_GROUP(TestRegistryEntry)

{

    JavaStorage* js;

    JavaStorageTestUtils* jtu;

    CActiveScheduler* newScheduler;

    TEST_SETUP()

    {

        newScheduler = new CActiveScheduler();

        CActiveScheduler::Install(newScheduler);

        js = JavaStorage::createInstance();

        jtu = new JavaStorageTestUtils();

    }

    TEST_TEARDOWN()

    {

        try

        {

            js->rollbackTransaction();

            js->close();

            delete js;

            js = NULL;

        }

        catch (...)

        {

            // No can do

        }

        delete js;

        js = 0;

        delete jtu;

        jtu = 0;

        delete newScheduler;

        newScheduler = NULL;

    }

};

/**

 * Test CJavaRegistryEntry::Uid() method.

 * 1. Application uid.

 * 2. Application suite uid.

 * NOTE: Invalid Uid cannot be tested as it cannot be queried.

 */

TEST(TestRegistryEntry, TestUid)

{

    LOG(EJavaStorage, EInfo, "+TestUid");

    JavaStorageApplicationEntry_t midlet1;

    JavaStorageEntry attr;

    Uid suite1Uid(L"[e22a4577]");

    attr.setEntry(PACKAGE_ID, suite1Uid.toString());

    midlet1.insert(attr);

    Uid midlet1Uid(L"[e22a4588]");

    attr.setEntry(ID, midlet1Uid.toString());

    midlet1.insert(attr);

    js->open();

    js->startTransaction();

    CHECK(jtu->populate(*js, APPLICATION_TABLE, midlet1));

    JavaStorageApplicationEntry_t suite1;

    attr.setEntry(PACKAGE_NAME, L"MySuite");

    suite1.insert(attr);

    attr.setEntry(ID, suite1Uid.toString());

    suite1.insert(attr);

    attr.setEntry(MEDIA_ID, L"-124614446");

    suite1.insert(attr);

    CHECK(jtu->populate(*js, APPLICATION_PACKAGE_TABLE, suite1));

    // Session must be committed before next use of Registry otherwise

    // it is locked.

    js->commitTransaction();

    auto_ptr<CJavaRegistry> registry(CJavaRegistry::NewL());

    // 1. Application uid.

    TUid midletUid;

    uidToTUid(midlet1Uid, midletUid);

    CJavaRegistryEntry* entry = registry->RegistryEntryL(midletUid);

    CHECK(entry != NULL);

    CHECK(entry->Uid().iUid == midletUid.iUid);

    delete entry;

    entry = NULL;

    // 2. Application suite uid.

    TUid suiteUid;

    uidToTUid(suite1Uid, suiteUid);

    entry = registry->RegistryEntryL(suiteUid);

    CHECK(entry != NULL);

    CHECK(entry->Uid().iUid == suiteUid.iUid);

    delete entry;

    entry = NULL;

    // Clean

    js->startTransaction();

    CHECK(jtu->remove(*js, APPLICATION_TABLE, midlet1));

    CHECK(jtu->remove(*js, APPLICATION_PACKAGE_TABLE, suite1));

    js->commitTransaction();

    js->close();

    LOG(EJavaStorage, EInfo, "-TestUid");

}

/**

 * Test CJavaRegistryEntry::Name() method.

 * 1. Application no name defined

 * 2. Application suite no name defined

 * 3. Application name.

 * 4. Application suite name.

 */

TEST(TestRegistryEntry, TestName)

{

    LOG(EJavaStorage, EInfo, "+TestName");

    JavaStorageApplicationEntry_t midlet1;

    JavaStorageEntry attr;

    Uid suite1Uid(L"[e2234577]");

    attr.setEntry(PACKAGE_ID, suite1Uid.toString());

    midlet1.insert(attr);

    Uid midlet1Uid(L"[e2234588]");

    attr.setEntry(ID, midlet1Uid.toString());

    midlet1.insert(attr);

    js->open();

    js->startTransaction();

    CHECK(jtu->populate(*js, APPLICATION_TABLE, midlet1));

    JavaStorageApplicationEntry_t suite1;

    attr.setEntry(ID, suite1Uid.toString());

    suite1.insert(attr);

    attr.setEntry(MEDIA_ID, L"-124614446");

    suite1.insert(attr);

    CHECK(jtu->populate(*js, APPLICATION_PACKAGE_TABLE, suite1));

    js->commitTransaction();

    auto_ptr<CJavaRegistry> registry(CJavaRegistry::NewL());

    // 1. Application uid.

    TUid midletUid;

    uidToTUid(midlet1Uid, midletUid);

    CJavaRegistryEntry* entry = registry->RegistryEntryL(midletUid);

    CHECK(entry != NULL);

    // 1. Application no name defined

    CHECK(entry->Name() == KNullDesC);

    // 2. Application suite no name defined

    TUid suiteUid;

    uidToTUid(suite1Uid, suiteUid);

    delete entry;

    entry = NULL;

    entry = registry->RegistryEntryL(suiteUid);

    CHECK(entry != NULL);

    CHECK(entry->Name() == KNullDesC);

    delete entry;

    entry = NULL;

    JavaStorageApplicationEntry_t midlet2;

    Uid suite2Uid(L"[e2224577]");

    attr.setEntry(PACKAGE_ID, suite2Uid.toString());

    midlet2.insert(attr);

    Uid midlet2Uid(L"[e2224588]");

    attr.setEntry(ID, midlet2Uid.toString());

    midlet2.insert(attr);

    wstring midlet2Name = L"MIDlet2";

    attr.setEntry(NAME, midlet2Name);

    midlet2.insert(attr);

    js->open();

    js->startTransaction();

    CHECK(jtu->populate(*js, APPLICATION_TABLE, midlet2));

    wstring suite2Name = L"MySuite2";

    JavaStorageApplicationEntry_t suite2;

    attr.setEntry(PACKAGE_NAME, suite2Name);

    suite2.insert(attr);

    attr.setEntry(ID, suite2Uid.toString());

    suite2.insert(attr);

    attr.setEntry(MEDIA_ID, L"-124614446");

    suite2.insert(attr);

    CHECK(jtu->populate(*js, APPLICATION_PACKAGE_TABLE, suite2));

    // Session must be committed before next use of Registry otherwise

    // it is locked.

    js->commitTransaction();

    // 3. Application name.

    uidToTUid(midlet2Uid, midletUid);

    entry = registry->RegistryEntryL(midletUid);

    CHECK(entry != NULL);

    const TDesC16& name = entry->Name();

    HBufC16* refName = wstringToBuf(midlet2Name);

    CHECK(name.Compare(refName->Des()) == 0);

    delete refName;

    refName = NULL;

    delete entry;

    entry = NULL;

    // 4. Application suite name.

    uidToTUid(suite2Uid, suiteUid);

    entry = registry->RegistryEntryL(suiteUid);

    CHECK(entry != NULL);

    const TDesC16& name2 = entry->Name();

    refName = wstringToBuf(suite2Name);

    CHECK(name2.Compare(refName->Des()) == 0);

    delete refName;

    refName = NULL;

    delete entry;

    entry = NULL;

    registry.reset(0);

    js->startTransaction();

    CHECK(jtu->remove(*js, APPLICATION_TABLE, midlet1));

    CHECK(jtu->remove(*js, APPLICATION_PACKAGE_TABLE, suite1));

    CHECK(jtu->remove(*js, APPLICATION_TABLE, midlet2));

    CHECK(jtu->remove(*js, APPLICATION_PACKAGE_TABLE, suite2));

    js->commitTransaction();

    js->close();

    LOG(EJavaStorage, EInfo, "-TestName");

}

/**

 * Test CJavaRegistryEntry::Type() method.

 * 1. Application type.

 * 2. Application suite type.

 */

TEST(TestRegistryEntry, TestType)

{

    LOG(EJavaStorage, EInfo, "+TestType");

    JavaStorageApplicationEntry_t midlet1;

    JavaStorageEntry attr;

    Uid suite1Uid(L"[ea234577]");

    attr.setEntry(PACKAGE_ID, suite1Uid.toString());

    midlet1.insert(attr);

    Uid midlet1Uid(L"[ea234588]");

    attr.setEntry(ID, midlet1Uid.toString());

    midlet1.insert(attr);

    js->open();

    js->startTransaction();

    CHECK(jtu->populate(*js, APPLICATION_TABLE, midlet1));

    JavaStorageApplicationEntry_t suite1;

    attr.setEntry(ID, suite1Uid.toString());

    suite1.insert(attr);

    attr.setEntry(MEDIA_ID, L"-124614446");

    suite1.insert(attr);

    CHECK(jtu->populate(*js, APPLICATION_PACKAGE_TABLE, suite1));

    // Session must be committed before next use of Registry otherwise

    // it is locked.

    js->commitTransaction();

    auto_ptr<CJavaRegistry> registry(CJavaRegistry::NewL());

    // 1. Application uid.

    TUid midletUid;

    uidToTUid(midlet1Uid, midletUid);

    CJavaRegistryEntry* entry = registry->RegistryEntryL(midletUid);

    CHECK(entry != NULL);

    CHECK(entry->Type() == EMidp2Midlet);

    delete entry;

    entry = NULL;

    TUid suiteUid;

    uidToTUid(suite1Uid, suiteUid);

    entry = registry->RegistryEntryL(suiteUid);

    CHECK(entry != NULL);

    CHECK(entry->Type() == EMidp2MidletSuite);

    delete entry;

    entry = NULL;

    js->startTransaction();

    CHECK(jtu->remove(*js, APPLICATION_TABLE, midlet1));

    CHECK(jtu->remove(*js, APPLICATION_PACKAGE_TABLE, suite1));

    js->commitTransaction();

    js->close();

    registry.reset(0);

    LOG(EJavaStorage, EInfo, "-TestType");

}

/**

 * Test CJavaRegistryEntry::Drive() method.

 * 1. Get application suite drive.

 * 2. Get application drive.

 */

TEST(TestRegistryEntry, TestDrive)

{

    LOG(EJavaStorage, EInfo, "+TestDrive");

    JavaStorageApplicationEntry_t midlet1;

    JavaStorageEntry attr;

    Uid suite1Uid(L"[e01a4577]");

    attr.setEntry(PACKAGE_ID, suite1Uid.toString());

    midlet1.insert(attr);

    Uid midlet1Uid(L"[e01a4588]");

    attr.setEntry(ID, midlet1Uid.toString());

    midlet1.insert(attr);

    js->open();

    js->startTransaction();

    CHECK(jtu->populate(*js, APPLICATION_TABLE, midlet1));

    JavaStorageApplicationEntry_t suite1;

    attr.setEntry(ID, suite1Uid.toString());

    suite1.insert(attr);

    attr.setEntry(ROOT_PATH, L"e:\\private\\MIDlets\\[e01a4577]");

    suite1.insert(attr);

    attr.setEntry(MEDIA_ID, L"-124614446");

    suite1.insert(attr);

    CHECK(jtu->populate(*js, APPLICATION_PACKAGE_TABLE, suite1));

    // Session must be committed before next use of Registry otherwise

    // it is locked.

    js->commitTransaction();

    auto_ptr<CJavaRegistry> registry(CJavaRegistry::NewL());

    // 1. Application uid.

    TUid midletUid;

    uidToTUid(midlet1Uid, midletUid);

    CJavaRegistryEntry* entry = registry->RegistryEntryL(midletUid);

    CHECK(entry != NULL);

    CHECK(entry->Drive() == 4); // 4 represents e:

    delete entry;

    entry = NULL;

    TUid suiteUid;

    uidToTUid(suite1Uid, suiteUid);

    entry = registry->RegistryEntryL(suiteUid);

    CHECK(entry != NULL);

    CHECK(entry->Drive() == 4);  // 4 represents c:

    delete entry;

    entry = NULL;

    js->startTransaction();

    CHECK(jtu->remove(*js, APPLICATION_TABLE, midlet1));

    CHECK(jtu->remove(*js, APPLICATION_PACKAGE_TABLE, suite1));

    js->commitTransaction();

    js->close();

    registry.reset(0);

    LOG(EJavaStorage, EInfo, "-TestDrive");

}

/**

 * Test CJavaRegistryEntry::AddributeL() method.

 * 1. Test no attributes in DB. NULL is returned.

 * 2. Test empty name. Leaves with KErrArgument.

 * 3. Test non-existing attribute. NULL is returned.

 * 4. Test existing application attribute.

 * 5. Test existing application suite attribute.

 * 6. Test trusted attribute.

 * 7. Test attribute VALUE not set i.e. empty.

 *    NULL is returned.

 */

TEST(TestRegistryEntry, TestAttribute)

{

    LOG(EJavaStorage, EInfo, "+TestAttribute");

    JavaStorageApplicationEntry_t midlet1;

    JavaStorageEntry attr;

    Uid suite1Uid(L"[ebbb4577]");

    attr.setEntry(PACKAGE_ID, suite1Uid.toString());

    midlet1.insert(attr);

    Uid midlet1Uid(L"[ebbb4588]");

    attr.setEntry(ID, midlet1Uid.toString());

    midlet1.insert(attr);

    js->open();

    js->startTransaction();

    CHECK(jtu->populate(*js, APPLICATION_TABLE, midlet1));

    JavaStorageApplicationEntry_t suite1;

    attr.setEntry(ID, suite1Uid.toString());

    suite1.insert(attr);

    attr.setEntry(ROOT_PATH, L"e:\\private\\MIDlets\\[ebbb4577]");

    suite1.insert(attr);

    attr.setEntry(MEDIA_ID, L"-124614446");

    suite1.insert(attr);

    CHECK(jtu->populate(*js, APPLICATION_PACKAGE_TABLE, suite1));

    js->commitTransaction();

    // 1. Test no attributes in DB.

    TUid midletUid;

    uidToTUid(midlet1Uid, midletUid);

    auto_ptr<CJavaRegistry> registry(CJavaRegistry::NewL());

    CJavaRegistryEntry* entry = registry->RegistryEntryL(midletUid);

    CHECK(entry != NULL);

    // 1. non-existing attribute

    _LIT(KAttrName, "TestClass");

    const MJavaAttribute* regAttr = entry->AttributeL(KAttrName);

    CHECK(regAttr == NULL);

    delete entry;

    entry = NULL;

    // Populate attributes

    JavaStorageApplicationEntry_t suiteAttribute;

    attr.setEntry(ID, suite1Uid.toString());

    suiteAttribute.insert(attr);

    attr.setEntry(NAME, L"MIDlet-1");

    suiteAttribute.insert(attr);

    attr.setEntry(VALUE, L"");

    suiteAttribute.insert(attr);

    attr.setEntry(TRUSTED, L"0", JavaStorageEntry::INT);