org.eclipsetrader.ui.internal.charts.DataImportJobTest.java Source code

Introduction

Here is the source code for org.eclipsetrader.ui.internal.charts.DataImportJobTest.java

Source

/*
 * Copyright (c) 2004-2011 Marco Maccaferri and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Marco Maccaferri - initial API and implementation
 */

package org.eclipsetrader.ui.internal.charts;

import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;

import junit.framework.TestCase;

import org.eclipse.jface.preference.PreferenceStore;
import org.eclipsetrader.core.feed.FeedIdentifier;
import org.eclipsetrader.core.feed.FeedProperties;
import org.eclipsetrader.core.instruments.ISecurity;
import org.eclipsetrader.core.instruments.Security;
import org.eclipsetrader.ui.internal.UIActivator;

public class DataImportJobTest extends TestCase {

    public void testFilterSecuritiesWithoutFeed() throws Exception {
        ISecurity[] security = new ISecurity[] { new Security("Test1", null),
                new Security("Test2", new FeedIdentifier("TEST", new FeedProperties())), };

        DataImportJob job = new DataImportJob(security, 0, null, null, null);
        ISecurity[] filtered = job.getFilteredSecurities(security);

        assertEquals(1, filtered.length);
    }

    public void testSortFilterSecurities() throws Exception {
        ISecurity[] security = new ISecurity[] {
                new Security("Test2", new FeedIdentifier("TEST", new FeedProperties())),
                new Security("Test1", new FeedIdentifier("TEST", new FeedProperties())), };

        DataImportJob job = new DataImportJob(security, 0, null, null, null);
        ISecurity[] filtered = job.getFilteredSecurities(security);

        assertSame(filtered[0], security[1]);
        assertSame(filtered[1], security[0]);
    }

    public void testGetDefaultYearsStartDate() throws Exception {
        PreferenceStore preferences = new PreferenceStore();
        preferences.setValue(UIActivator.PREFS_INITIAL_BACKFILL_METHOD, 1);
        preferences.setValue(UIActivator.PREFS_INITIAL_BACKFILL_YEARS, 5);

        Calendar c = Calendar.getInstance();
        c.set(Calendar.MILLISECOND, 0);
        c.add(Calendar.YEAR, -preferences.getInt(UIActivator.PREFS_INITIAL_BACKFILL_YEARS));

        DataImportJob job = new DataImportJob(new Security("Test", null), 0, null, null, null);
        job.preferences = preferences;

        assertEquals(c.getTime(), job.getDefaultStartDate());
    }

    public void testGetDefaultStartDate() throws Exception {
        PreferenceStore preferences = new PreferenceStore();
        preferences.setValue(UIActivator.PREFS_INITIAL_BACKFILL_METHOD, 0);
        preferences.setValue(UIActivator.PREFS_INITIAL_BACKFILL_START_DATE, "20010512");

        Date expectedDate = new SimpleDateFormat("yyyyMMdd")
                .parse(preferences.getString(UIActivator.PREFS_INITIAL_BACKFILL_START_DATE));

        DataImportJob job = new DataImportJob(new Security("Test", null), 0, null, null, null);
        job.preferences = preferences;

        assertEquals(expectedDate, job.getDefaultStartDate());
    }
}