design/designer-base/src/test/java/com/fr/file/FILEChooserPaneTest.java

package com.fr.file;

import com.fr.base.extension.FileExtension;
import com.fr.design.i18n.Toolkit;
import com.fr.file.filter.ChooseFileFilter;
import com.fr.invoke.Reflect;
import com.fr.stable.StringUtils;
import org.easymock.EasyMock;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.easymock.PowerMock;
import org.powermock.core.classloader.annotations.PowerMockIgnore;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;


/**
 * Created by kerry on 2019-10-15
 */
@RunWith(PowerMockRunner.class)
@PowerMockIgnore({"javax.management.*", "javax.swing.*"})
@PrepareForTest({Toolkit.class, FileNodeFILE.class})
public class FILEChooserPaneTest {
    @Test
    public void testAddChooseFileFilter() {
        PowerMock.mockStatic(Toolkit.class);
        EasyMock.expect(Toolkit.i18nText(EasyMock.anyString())).andReturn("test").anyTimes();
        PowerMock.replayAll();
        FILEChooserPane chooserPane = EasyMock.mock(FILEChooserPane.class);
        Reflect.on(chooserPane).set("suffix", ".cpt");
        String result1 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1.cpt", null).get();
        Assert.assertEquals("WorkBook1.cpt", result1);

        ChooseFileFilter chooseFileFilter1 = new ChooseFileFilter(FileExtension.CPT, StringUtils.EMPTY);
        String result2 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1.cpt", chooseFileFilter1).get();
        Assert.assertEquals("WorkBook1.cpt", result2);

        ChooseFileFilter chooseFileFilter2 = new ChooseFileFilter(FileExtension.CPTX, StringUtils.EMPTY);
        String result3 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1.cpt", chooseFileFilter2).get();
        Assert.assertEquals("WorkBook1.cptx", result3);

        ChooseFileFilter chooseFileFilter3 = new ChooseFileFilter(FileExtension.CPT, StringUtils.EMPTY);
        String result4 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1.cptx", chooseFileFilter3).get();
        Assert.assertEquals("WorkBook1.cpt", result4);

        ChooseFileFilter chooseFileFilter5 = new ChooseFileFilter(FileExtension.CPTX, StringUtils.EMPTY);
        String result5 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1.cptx", chooseFileFilter5).get();
        Assert.assertEquals("WorkBook1.cptx", result5);

        ChooseFileFilter chooseFileFilter6 = new ChooseFileFilter(FileExtension.CPT, StringUtils.EMPTY);
        String result6 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1.xls", chooseFileFilter6).get();
        Assert.assertEquals("WorkBook1.xls.cpt", result6);

        ChooseFileFilter chooseFileFilter7 = new ChooseFileFilter(FileExtension.XLS, StringUtils.EMPTY);
        chooseFileFilter7.addExtension(".xlsx");
        String result7 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1", chooseFileFilter7).get();
        Assert.assertEquals("WorkBook1.xls", result7);

        ChooseFileFilter chooseFileFilter8 = new ChooseFileFilter(FileExtension.XLSX, StringUtils.EMPTY);
        chooseFileFilter8.addExtension(".xls");
        String result8 = Reflect.on(chooserPane).call("calProperFileName", "WorkBook1", chooseFileFilter8).get();
        Assert.assertEquals("WorkBook1.xlsx", result8);

    }

}