yaoh.wu
5 years ago
2 changed files with 107 additions and 10 deletions
@ -0,0 +1,102 @@ |
|||||||
|
package com.fr.design.data; |
||||||
|
|
||||||
|
import com.fr.base.TableData; |
||||||
|
import com.fr.data.impl.storeproc.StoreProcedure; |
||||||
|
import com.fr.design.data.tabledata.wrapper.TableDataFactory; |
||||||
|
import com.fr.design.data.tabledata.wrapper.TableDataWrapper; |
||||||
|
import com.fr.file.ProcedureConfig; |
||||||
|
import com.fr.file.TableDataConfig; |
||||||
|
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.PrepareForTest; |
||||||
|
import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor; |
||||||
|
import org.powermock.modules.junit4.PowerMockRunner; |
||||||
|
|
||||||
|
import java.util.HashMap; |
||||||
|
import java.util.Map; |
||||||
|
|
||||||
|
@RunWith(PowerMockRunner.class) |
||||||
|
@PrepareForTest({TableDataFactory.class, StoreProcedure.class, TableDataConfig.class, ProcedureConfig.class}) |
||||||
|
@SuppressStaticInitializationFor({"com.fr.design.data.tabledata.wrapper.TableDataFactory"}) |
||||||
|
public class DesignTableDataManagerTest { |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testFireDataChange() { |
||||||
|
|
||||||
|
DesignTableDataManager.envChange(); |
||||||
|
|
||||||
|
|
||||||
|
PowerMock.mockStatic(TableDataConfig.class); |
||||||
|
|
||||||
|
TableDataConfig config = EasyMock.mock(TableDataConfig.class); |
||||||
|
TableDataConfig config2 = EasyMock.mock(TableDataConfig.class); |
||||||
|
|
||||||
|
TableData td1 = EasyMock.mock(TableData.class); |
||||||
|
TableData td2 = EasyMock.mock(TableData.class); |
||||||
|
TableData td3 = EasyMock.mock(TableData.class); |
||||||
|
TableData td4 = EasyMock.mock(TableData.class); |
||||||
|
EasyMock.expect(config.getTableData("firstData")).andReturn(td1).once(); |
||||||
|
EasyMock.expect(config.getTableData("secondData")).andReturn(td2).once(); |
||||||
|
EasyMock.expect(config2.getTableData("firstData")).andReturn(td3).once(); |
||||||
|
EasyMock.expect(config2.getTableData("secondData")).andReturn(td4).once(); |
||||||
|
|
||||||
|
EasyMock.expect(TableDataConfig.getInstance()).andReturn(config).once().andReturn(config2).once(); |
||||||
|
|
||||||
|
PowerMock.mockStatic(TableDataFactory.class); |
||||||
|
EasyMock.expect(TableDataFactory.getSortOfChineseNameOfServerData(config)) |
||||||
|
.andReturn(new String[]{"firstData", "secondData"}).once(); |
||||||
|
EasyMock.expect(TableDataFactory.getSortOfChineseNameOfServerData(config2)) |
||||||
|
.andReturn(new String[]{"firstData", "secondData"}).once(); |
||||||
|
|
||||||
|
|
||||||
|
ProcedureConfig proConfig = EasyMock.mock(ProcedureConfig.class); |
||||||
|
StoreProcedure pc1 = PowerMock.createMock(StoreProcedure.class); |
||||||
|
StoreProcedure pc2 = PowerMock.createMock(StoreProcedure.class); |
||||||
|
ProcedureConfig proConfig2 = EasyMock.mock(ProcedureConfig.class); |
||||||
|
StoreProcedure pc3 = PowerMock.createMock(StoreProcedure.class); |
||||||
|
StoreProcedure pc4 = PowerMock.createMock(StoreProcedure.class); |
||||||
|
|
||||||
|
Map<String, StoreProcedure> procedureMap = new HashMap<>(); |
||||||
|
procedureMap.put("firstPRO", pc1); |
||||||
|
procedureMap.put("secondPRO", pc2); |
||||||
|
EasyMock.expect(proConfig.getProcedure("firstPRO")).andReturn(pc1).once(); |
||||||
|
EasyMock.expect(proConfig.getProcedure("secondPRO")).andReturn(pc2).once(); |
||||||
|
|
||||||
|
Map<String, StoreProcedure> procedureMap2 = new HashMap<>(); |
||||||
|
procedureMap2.put("firstPRO", pc3); |
||||||
|
procedureMap2.put("secondPRO", pc4); |
||||||
|
EasyMock.expect(proConfig2.getProcedure("firstPRO")).andReturn(pc3).once(); |
||||||
|
EasyMock.expect(proConfig2.getProcedure("secondPRO")).andReturn(pc4).once(); |
||||||
|
|
||||||
|
EasyMock.expect(proConfig.getProcedures()).andReturn(procedureMap).once(); |
||||||
|
EasyMock.expect(proConfig2.getProcedures()).andReturn(procedureMap2).once(); |
||||||
|
|
||||||
|
PowerMock.mockStatic(ProcedureConfig.class); |
||||||
|
EasyMock.expect(ProcedureConfig.getInstance()).andReturn(proConfig).once().andReturn(proConfig2).once(); |
||||||
|
|
||||||
|
EasyMock.replay(proConfig, config, config2, proConfig2); |
||||||
|
PowerMock.replay(TableDataFactory.class, TableDataConfig.class, ProcedureConfig.class); |
||||||
|
|
||||||
|
|
||||||
|
Map<String, TableDataWrapper> map = DesignTableDataManager.getAllEditingDataSet(null); |
||||||
|
Assert.assertEquals(4, map.size()); |
||||||
|
Assert.assertSame(pc1, map.get("firstPRO").getTableData()); |
||||||
|
Assert.assertSame(pc2, map.get("secondPRO").getTableData()); |
||||||
|
Assert.assertSame(td1, map.get("firstData").getTableData()); |
||||||
|
Assert.assertSame(td2, map.get("secondData").getTableData()); |
||||||
|
|
||||||
|
DesignTableDataManager.fireDSChanged(new HashMap<String, String>()); |
||||||
|
|
||||||
|
Map<String, TableDataWrapper> map2 = DesignTableDataManager.getAllEditingDataSet(null); |
||||||
|
Assert.assertEquals(4, map2.size()); |
||||||
|
Assert.assertSame(pc3, map2.get("firstPRO").getTableData()); |
||||||
|
Assert.assertSame(pc4, map2.get("secondPRO").getTableData()); |
||||||
|
Assert.assertSame(td3, map2.get("firstData").getTableData()); |
||||||
|
Assert.assertSame(td4, map2.get("secondData").getTableData()); |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue