Search in sources :

Example 1 with FileDiffState

use of org.smartdata.model.FileDiffState in project SSM by Intel-bigdata.

the class TestFileDiffDao method testBatchUpdateAndQuery.

@Test
public void testBatchUpdateAndQuery() {
    FileDiff[] fileDiffs = new FileDiff[2];
    fileDiffs[0] = new FileDiff();
    fileDiffs[0].setDiffId(1);
    fileDiffs[0].setParameters(new HashMap<String, String>());
    fileDiffs[0].setSrc("test");
    fileDiffs[0].setState(FileDiffState.RUNNING);
    fileDiffs[0].setDiffType(FileDiffType.APPEND);
    fileDiffs[0].setCreateTime(1);
    fileDiffs[1] = new FileDiff();
    fileDiffs[1].setDiffId(2);
    fileDiffs[1].setParameters(new HashMap<String, String>());
    fileDiffs[1].setSrc("src");
    fileDiffs[1].setState(FileDiffState.PENDING);
    fileDiffs[1].setDiffType(FileDiffType.APPEND);
    fileDiffs[1].setCreateTime(1);
    fileDiffDao.insert(fileDiffs);
    List<FileDiff> fileInfoList = fileDiffDao.getAll();
    for (int i = 0; i < 2; i++) {
        Assert.assertTrue(fileInfoList.get(i).equals(fileDiffs[i]));
    }
    // update
    List<Long> dids = new ArrayList<>();
    dids.add(1L);
    dids.add(2L);
    List<String> parameters = new ArrayList<>();
    parameters.add(fileDiffs[0].getParametersJsonString());
    parameters.add(fileDiffs[1].getParametersJsonString());
    List<FileDiffState> fileDiffStates = new ArrayList<>();
    fileDiffStates.add(FileDiffState.APPLIED);
    fileDiffStates.add(fileDiffs[1].getState());
    fileDiffDao.batchUpdate(dids, fileDiffStates, parameters);
    fileInfoList = fileDiffDao.getAll();
    Assert.assertTrue(fileInfoList.get(0).getState().equals(FileDiffState.APPLIED));
    fileDiffDao.batchUpdate(dids, FileDiffState.MERGED);
    Assert.assertTrue(fileDiffDao.getAll().get(0).getState().equals(FileDiffState.MERGED));
}
Also used : FileDiffState(org.smartdata.model.FileDiffState) ArrayList(java.util.ArrayList) FileDiff(org.smartdata.model.FileDiff) Test(org.junit.Test)

Aggregations

ArrayList (java.util.ArrayList)1 Test (org.junit.Test)1 FileDiff (org.smartdata.model.FileDiff)1 FileDiffState (org.smartdata.model.FileDiffState)1