/*

* Copyright (c) 2008-2010 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:

*

*/

package com.nokia.mj.impl.installer.storagehandler;

import com.nokia.mj.impl.installer.utils.InstallerMain;

import com.nokia.mj.impl.installer.utils.Log;

import com.nokia.mj.impl.storage.StorageAttribute;

import com.nokia.mj.impl.storage.StorageEntry;

import com.nokia.mj.impl.storage.StorageNames;

import com.nokia.mj.impl.storage.StorageSession;

import com.nokia.mj.impl.utils.Attribute;

import com.nokia.mj.impl.utils.Uid;

import com.nokia.mj.impl.utils.Version;

import java.util.Enumeration;

import java.util.Hashtable;

import java.util.Vector;

import j2meunit.framework.Test;

import j2meunit.framework.TestCase;

import j2meunit.framework.TestMethod;

import j2meunit.framework.TestSuite;

/**

 * StorageHandler unit tests.

 */

public class StorageHandlerTest extends TestCase implements InstallerMain

{

    private static final int MAX_SUITE_COUNT = 8;

    private StorageHandler iStorageHandler = null;

    // Begin j2meunit test framework setup

    public void installerMain(String[] args)

    {

        TestSuite suite = new TestSuite(this.getClass().getName());

        suite.addTest(new StorageHandlerTest("testSuiteComparison", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testSuiteComparison();

            }

        }));

        suite.addTest(new StorageHandlerTest("testCommitTwice", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testCommitTwice();

            }

        }));

        suite.addTest(new StorageHandlerTest("testRollbackTwice", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testRollbackTwice();

            }

        }));

        suite.addTest(new StorageHandlerTest("testCommitRollback", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testCommitRollback();

            }

        }));

        suite.addTest(new StorageHandlerTest("testRollbackCommit", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testRollbackCommit();

            }

        }));

        suite.addTest(new StorageHandlerTest("testWriteReadByUidRemove", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testWriteReadByUidRemove();

            }

        }));

        suite.addTest(new StorageHandlerTest("testWriteReadByNameAndVendorRemove", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testWriteReadByNameAndVendorRemove();

            }

        }));

        suite.addTest(new StorageHandlerTest("testWriteCommitReadByUidRemoveCommit", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testWriteCommitReadByUidRemoveCommit();

            }

        }));

        suite.addTest(new StorageHandlerTest("testWriteCommitReadByNameAndVendorRemoveCommit", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testWriteCommitReadByNameAndVendorRemoveCommit();

            }

        }));

        suite.addTest(new StorageHandlerTest("testWriteRollbackReadByUid", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testWriteRollbackReadByUid();

            }

        }));

        suite.addTest(new StorageHandlerTest("testWriteRollbackReadByNameAndVendor", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testWriteRollbackReadByNameAndVendor();

            }

        }));

        suite.addTest(new StorageHandlerTest("testAllocateUid", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testAllocateUid();

            }

        }));

        suite.addTest(new StorageHandlerTest("testGetSuiteUids", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testGetSuiteUids();

            }

        }));

        suite.addTest(new StorageHandlerTest("testGetSuiteUidFromAppUid", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testGetSuiteUidFromAppUid();

            }

        }));

        suite.addTest(new StorageHandlerTest("testGetSuiteUidFromSuiteUid", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testGetSuiteUidFromSuiteUid();

            }

        }));

        suite.addTest(new StorageHandlerTest("testGetAppUids", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testGetAppUids();

            }

        }));

        suite.addTest(new StorageHandlerTest("testInstallPreinstall", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testInstallPreinstall();

            }

        }));

        suite.addTest(new StorageHandlerTest("testPreinstallInstall", new TestMethod()

        {

            public void run(TestCase tc)

            {

                ((StorageHandlerTest)tc).testPreinstallInstall();

            }

        }));

        com.nokia.mj.impl.utils.OmjTestRunner.run(suite);

    }

    public StorageHandlerTest()

    {

    }

    public StorageHandlerTest(String aTestName, TestMethod aTestMethod)

    {

        super(aTestName, aTestMethod);

    }

    public void assertFalse(String aMsg, boolean aCondition)

    {

        assertTrue(aMsg, !aCondition);

    }

    // End j2meunit test framework setup

    protected void setUp()

    {

        //System.out.println("----" + getName() + " setUp----");

        SuiteInfo[] suites = new SuiteInfo[MAX_SUITE_COUNT];

        for (int i = 0; i < suites.length; i++)

        {

            suites[i] = createSuiteInfo(i);

            cleanSuiteFromStorage(suites[i]);

        }

        iStorageHandler = new StorageHandler();

    }

    protected void tearDown()

    {

        iStorageHandler.close();

        iStorageHandler = null;

        //System.out.println("----" + getName() + " tearDown----");

    }

    public void testSuiteComparison()

    {

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suite2 = createSuiteInfo();

        // Test code for verifying that suite comparison works.

        assertTrue("expected suite to be equal", equals(suite, suite));

        assertTrue("expected suite2 to be equal", equals(suite, suite2));

        suite2.setAttribute("MIDlet-Version", "1.1");

        assertFalse("expected suite2 to not be equal", equals(suite, suite2));

    }

    public void testCommitTwice()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        tryCommitSession();

        tryCommitSessionFails();

    }

    public void testRollbackTwice()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        tryRollbackSession();

        tryRollbackSessionFails();

    }

    public void testCommitRollback()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        tryCommitSession();

        tryRollbackSessionFails();

    }

    public void testRollbackCommit()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        tryRollbackSession();

        tryCommitSessionFails();

    }

    public void testWriteReadByUidRemove()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        suite = tryReadSuite(suiteOrig.getUid());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        tryRemoveSuite(suiteOrig);

        tryReadNoSuite(suiteOrig.getUid());

        tryRollbackSession();

        // After rollback we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        tryReadNoSuite(suiteOrig.getUid());

        iStorageHandler.close();

    }

    public void testWriteReadByNameAndVendorRemove()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        suite = tryReadSuite(suiteOrig.getName(), suiteOrig.getVendor());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        tryRemoveSuite(suiteOrig);

        tryReadNoSuite(suiteOrig.getUid());

        tryRollbackSession();

        // After rollback we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        tryReadNoSuite(suiteOrig.getUid());

        iStorageHandler.close();

    }

    public void testWriteCommitReadByUidRemoveCommit()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        suite = tryReadSuite(suiteOrig.getUid());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        tryCommitSession();

        // After commit we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        suite = tryReadSuite(suiteOrig.getUid());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        iStorageHandler.close();

        iStorageHandler = new StorageHandler();

        tryStartSession();

        tryRemoveSuite(suiteOrig);

        tryReadNoSuite(suiteOrig.getUid());

        tryCommitSession();

        // After commit we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        tryReadNoSuite(suiteOrig.getUid());

        iStorageHandler.close();

    }

    public void testWriteCommitReadByNameAndVendorRemoveCommit()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        suite = tryReadSuite(suiteOrig.getName(), suiteOrig.getVendor());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        tryCommitSession();

        // After commit we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        suite = tryReadSuite(suiteOrig.getName(), suiteOrig.getVendor());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        iStorageHandler.close();

        iStorageHandler = new StorageHandler();

        tryStartSession();

        tryRemoveSuite(suiteOrig);

        tryReadNoSuite(suiteOrig.getUid());

        tryCommitSession();

        // After commit we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        tryReadNoSuite(suiteOrig.getUid());

        iStorageHandler.close();

    }

    public void testWriteRollbackReadByUid()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        suite = tryReadSuite(suiteOrig.getUid());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        tryRollbackSession();

        // After rollback we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        tryReadNoSuite(suiteOrig.getUid());

        iStorageHandler.close();

    }

    public void testWriteRollbackReadByNameAndVendor()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        SuiteInfo suiteOrig = suite;

        tryWriteSuite(suite);

        suite = tryReadSuite(suiteOrig.getName(), suiteOrig.getVendor());

        assertTrue("expected suite from storage to be equal with original",

                   equals(suite, suiteOrig));

        tryRollbackSession();

        // After rollback we have to get a new StorageHandler instance.

        iStorageHandler = new StorageHandler();

        tryReadNoSuite(suiteOrig.getUid());

        iStorageHandler.close();

    }

    public void testAllocateUid()

    {

        Uid uid = iStorageHandler.allocateUid("");

        assertTrue("expected uid to be not null", uid != null);

    }

    public void testGetSuiteUids()

    {

        tryStartSession();

        SuiteInfo[] suites = new SuiteInfo[MAX_SUITE_COUNT];

        for (int i = 0; i < suites.length; i++)

        {

            suites[i] = createSuiteInfo(i);

            tryWriteSuite(suites[i]);

        }

        try

        {

            Uid[] uids = iStorageHandler.getSuiteUids();

            assertTrue("expected uids to be not null", uids != null);

                       uids.length == suites.length);

        }

        catch (Exception ex)

        {

            ex.printStackTrace();

            assertTrue("Exception at getSuiteUids: " + ex.toString(), false);

        }

        tryRollbackSession();

        try

        {

            StorageHandler storageHandler = new StorageHandler();

            Uid[] uids = storageHandler.getSuiteUids();

            storageHandler.close();

            assertTrue("expected uids to be null", uids == null);

        }

        catch (Exception ex)

        {

            ex.printStackTrace();

            assertTrue("Exception at getSuiteUids after rollback: " +

                       ex.toString(), false);

        }

    }

    public void testGetSuiteUidFromAppUid()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        tryWriteSuite(suite);

        Uid[] appUids = suite.getApplicationUids();

        assertTrue("expected appUids not to be null", appUids != null);

        for (int i = 0; i < appUids.length; i++)

        {

            Uid uid = iStorageHandler.getSuiteUid(appUids[i]);

            assertTrue("expected uids to be equal", suite.getUid().equals(uid));

        }

        tryRollbackSession();

    }

    public void testGetSuiteUidFromSuiteUid()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        tryWriteSuite(suite);

        Uid uid = iStorageHandler.getSuiteUid(suite.getUid());

        assertTrue("expected uids to be equal", suite.getUid().equals(uid));

        tryRollbackSession();

    }

    public void testGetAppUids()

    {

        tryStartSession();

        SuiteInfo suite = createSuiteInfo();

        tryWriteSuite(suite);

        Uid[] appUidsOrig = suite.getApplicationUids();

        Uid[] appUids = iStorageHandler.getAppUids(suite.getUid());

        assertTrue("expected appUids not to be null", appUids != null);

        assertTrue("appUids length does not match",

                   appUidsOrig.length == appUids.length);

        boolean uidFound = false;

        for (int i = 0; i < appUids.length; i++)

        {

            for (int j = 0; j < appUidsOrig.length; j++)

            {

                if (appUids[i].equals(appUidsOrig[j]))

                {

                    // Found matching uid.

                    uidFound = true;

                    break;

                }

            }