com.archimatetool.editor.diagram.ImageExportProviderTests.java Source code

Introduction

Here is the source code for com.archimatetool.editor.diagram.ImageExportProviderTests.java

Source

/**
 * This program and the accompanying materials
 * are made available under the terms of the License
 * which accompanies this distribution in the file LICENSE.txt
 */
package com.archimatetool.editor.diagram;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;

import java.io.File;

import junit.framework.JUnit4TestAdapter;

import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.FreeformLayer;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.jface.preference.IPreferenceStore;
import org.eclipse.swt.graphics.ImageData;
import org.eclipse.swt.widgets.Shell;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.archimatetool.editor.diagram.IImageExportProvider.IExportDialogAdapter;
import com.archimatetool.editor.preferences.Preferences;
import com.archimatetool.tests.TestUtils;

/**
 * ImageExportProviderTests
 * 
 * @author Phillip Beauvoir
 */
public class ImageExportProviderTests {

    public static junit.framework.Test suite() {
        return new JUnit4TestAdapter(ImageExportProviderTests.class);
    }

    private ImageExportProvider provider;
    private Shell shell;

    private IFigure rootFigure;

    @Before
    public void onceOnceBeforeEachTest() {
        provider = new ImageExportProvider();
        shell = new Shell();

        // Set prefs to defaults
        IPreferenceStore store = Preferences.STORE;
        store.setValue(ImageExportProvider.PREFS_IMAGE_SCALE, 0);

        rootFigure = new FreeformLayer();
        rootFigure.setBounds(new Rectangle(0, 0, 500, 500));
    }

    @After
    public void onceOnceAfterEachTest() {
        shell.dispose();
    }

    @Test
    public void testInit() {
        provider.init(mock(IExportDialogAdapter.class), shell, rootFigure);
        assertTrue(shell.getChildren().length > 0);
    }

    @Test
    public void testBMPExport() throws Exception {
        testExport(ImageExportProvider.BMP_IMAGE_EXPORT_PROVIDER);
    }

    @Test
    public void testJPEGExport() throws Exception {
        testExport(ImageExportProvider.JPEG_IMAGE_EXPORT_PROVIDER);
    }

    @Test
    public void testPNGExport() throws Exception {
        testExport(ImageExportProvider.PNG_IMAGE_EXPORT_PROVIDER);
    }

    private void testExport(String formatID) throws Exception {
        IFigure childFigure = new Figure();
        childFigure.setBounds(new Rectangle(200, 200, 128, 52));
        rootFigure.add(childFigure);

        provider.init(mock(IExportDialogAdapter.class), shell, rootFigure);

        File tmp = TestUtils.createTempFile(null);

        provider.export(formatID, tmp);

        assertTrue(tmp.exists());

        ImageData imageData = new ImageData(tmp.getPath());
        assertEquals(148, imageData.width); // 128 width + 10 margins
        assertEquals(72, imageData.height); // 52 height + 10 margins

        // Test it by setting the scale to 200%
        provider.fScaleSpinner.setSelection(200);
        provider.export(formatID, tmp);
        assertTrue(tmp.exists());

        imageData = new ImageData(tmp.getPath());
        assertEquals(276, imageData.width); // 256 width + 10 margins
        assertEquals(124, imageData.height); // 104 height + 10 margins
    }

    @Test
    public void testSavePreferences() {
        provider.init(mock(IExportDialogAdapter.class), shell, rootFigure);

        provider.fScaleSpinner.setSelection(345);

        provider.savePreferences();

        IPreferenceStore store = Preferences.STORE;

        assertEquals(345, store.getInt(ImageExportProvider.PREFS_IMAGE_SCALE));
    }

    @Test
    public void testPreferencesWereLoaded() {
        IPreferenceStore store = Preferences.STORE;
        store.setValue(ImageExportProvider.PREFS_IMAGE_SCALE, 123);

        provider.init(mock(IExportDialogAdapter.class), shell, rootFigure);

        assertEquals(123, provider.fScaleSpinner.getSelection());
    }

}