Search in sources :

Example 6 with RecordCategory

use of org.alfresco.rest.rm.community.model.recordcategory.RecordCategory in project records-management by Alfresco.

the class ScheduleRecordLoadersUnitTest method testUploadRecordsWithNotExistentPreconfiguredSinglePath.

@Test
public void testUploadRecordsWithNotExistentPreconfiguredSinglePath() throws Exception {
    int maxActiveLoaders = 8;
    int recordsNumber = 4;
    String configuredPath1 = "/e1/e2/e3";
    String paths = configuredPath1;
    scheduleRecordLoaders.setUploadRecords(true);
    scheduleRecordLoaders.setMaxActiveLoaders(maxActiveLoaders);
    scheduleRecordLoaders.setRecordsNumber(recordsNumber);
    scheduleRecordLoaders.setRecordFolderPaths(paths);
    // file plan should be always there
    FolderData mockedFilePlanContainer = mock(FolderData.class);
    when(mockedFilePlanContainer.getId()).thenReturn("filePlanId");
    when(mockedFilePlanContainer.getContext()).thenReturn(FILEPLAN_CONTEXT);
    when(mockedFilePlanContainer.getPath()).thenReturn(RECORD_CONTAINER_PATH);
    when(mockedFileFolderService.getFolder(FILEPLAN_CONTEXT, RECORD_CONTAINER_PATH)).thenReturn(mockedFilePlanContainer);
    when(mockedRestApiFactory.getFilePlansAPI()).thenReturn(mockedFilePlanAPI);
    when(mockedRestApiFactory.getRecordCategoryAPI()).thenReturn(mockedRecordCategoryAPI);
    FilePlan mockedFilePlan = mock(FilePlan.class);
    when(mockedFilePlan.getId()).thenReturn("filePlanId");
    when(mockedFilePlanAPI.getFilePlan("filePlanId")).thenReturn(mockedFilePlan);
    // e1 root category
    RecordCategory mockedE1FilePlanComponent = mock(RecordCategory.class);
    when(mockedE1FilePlanComponent.getId()).thenReturn("e1Id");
    when(mockedRecordCategoryAPI.getRecordCategory("e1Id")).thenReturn(mockedE1FilePlanComponent);
    when(mockedFilePlanAPI.createRootRecordCategory(any(RecordCategory.class), eq("filePlanId"))).thenReturn(mockedE1FilePlanComponent);
    String e1Path = RECORD_CONTAINER_PATH + "/e1";
    FolderData mockedE1 = mock(FolderData.class);
    when(mockedE1.getId()).thenReturn("e1Id");
    when(mockedE1.getContext()).thenReturn(RECORD_CATEGORY_CONTEXT);
    when(mockedE1.getPath()).thenReturn(e1Path);
    when(mockedFileFolderService.getFolder(RECORD_CATEGORY_CONTEXT, e1Path)).thenReturn(null).thenReturn(mockedE1);
    when(mockedFileFolderService.getFolder("e1Id")).thenReturn(mockedE1);
    // e2 child category
    RecordCategory mockedE2childRecordCategory = mock(RecordCategory.class);
    when(mockedE2childRecordCategory.getId()).thenReturn("e2Id");
    RecordCategoryChild mockedE2RecordCategoryChild = mock(RecordCategoryChild.class);
    when(mockedE2RecordCategoryChild.getId()).thenReturn("e2Id");
    when(mockedRecordCategoryAPI.getRecordCategory("e2Id")).thenReturn(mockedE2childRecordCategory);
    when(mockedRecordCategoryAPI.createRecordCategoryChild(any(RecordCategoryChild.class), eq("e1Id"))).thenReturn(mockedE2RecordCategoryChild);
    String e2Path = RECORD_CONTAINER_PATH + "/e1/e2";
    FolderData mockedE2 = mock(FolderData.class);
    when(mockedE2.getId()).thenReturn("e2Id");
    when(mockedE2.getContext()).thenReturn(RECORD_CATEGORY_CONTEXT);
    when(mockedE2.getPath()).thenReturn(e2Path);
    when(mockedFileFolderService.getFolder(RECORD_CATEGORY_CONTEXT, e2Path)).thenReturn(null).thenReturn(mockedE2);
    when(mockedFileFolderService.getFolder("e2Id")).thenReturn(mockedE2);
    // e3 record folder
    RecordCategoryChild mockedE3RecordFolder = mock(RecordCategoryChild.class);
    when(mockedE3RecordFolder.getId()).thenReturn("e3Id");
    String e3Path = RECORD_CONTAINER_PATH + "/e1/e2/e3";
    FolderData mockedE3 = mock(FolderData.class);
    when(mockedE3.getId()).thenReturn("e3Id");
    when(mockedE3.getContext()).thenReturn(RECORD_FOLDER_CONTEXT);
    when(mockedE3.getPath()).thenReturn(e3Path);
    when(mockedFileFolderService.getFolder(RECORD_FOLDER_CONTEXT, e3Path)).thenReturn(null).thenReturn(null).thenReturn(mockedE3);
    when(mockedFileFolderService.getFolder("e3Id")).thenReturn(mockedE3);
    when(mockedRecordCategoryAPI.createRecordCategoryChild(any(RecordCategoryChild.class), eq("e2Id"))).thenReturn(mockedE3RecordFolder);
    when(mockedApplicationContext.getBean("restAPIFactory", RestAPIFactory.class)).thenReturn(mockedRestApiFactory);
    EventResult result = scheduleRecordLoaders.processEvent(null, new StopWatch());
    verify(mockedFileFolderService, never()).getFoldersByCounts(any(String.class), any(Long.class), any(Long.class), any(Long.class), any(Long.class), any(Long.class), any(Long.class), any(Integer.class), any(Integer.class));
    verify(mockedFileFolderService, times(3)).createNewFolder(any(String.class), any(String.class), any(String.class));
    verify(mockedFileFolderService, times(3)).incrementFolderCount(any(String.class), any(String.class), eq(1L));
    verify(mockedFileFolderService, times(3)).getFolder(any(String.class));
    verify(mockedSessionService, times(recordsNumber)).startSession(any(DBObject.class));
    assertEquals(true, result.isSuccess());
    assertEquals("Raised further " + (recordsNumber) + " events and rescheduled self.", result.getData());
    assertEquals(recordsNumber + 1, result.getNextEvents().size());
    for (int i = 0; i < recordsNumber; i++) {
        Event event = result.getNextEvents().get(i);
        assertEquals(EVENT_LOAD_RECORD, event.getName());
        DBObject dataObj = (DBObject) event.getData();
        assertNotNull(dataObj);
        assertEquals(RECORD_FOLDER_CONTEXT, (String) dataObj.get(FIELD_CONTEXT));
        assertEquals(LOAD_RECORD_OPERATION, dataObj.get(FIELD_LOAD_OPERATION));
    }
    assertEquals(EVENT_SCHEDULE_RECORD_LOADERS, result.getNextEvents().get(recordsNumber).getName());
}
Also used : EventResult(org.alfresco.bm.event.EventResult) FilePlan(org.alfresco.rest.rm.community.model.fileplan.FilePlan) RecordCategoryChild(org.alfresco.rest.rm.community.model.recordcategory.RecordCategoryChild) FolderData(org.alfresco.bm.cm.FolderData) Event(org.alfresco.bm.event.Event) RecordCategory(org.alfresco.rest.rm.community.model.recordcategory.RecordCategory) DBObject(com.mongodb.DBObject) StopWatch(org.apache.commons.lang3.time.StopWatch) Test(org.junit.Test)

Example 7 with RecordCategory

use of org.alfresco.rest.rm.community.model.recordcategory.RecordCategory in project records-management by Alfresco.

the class RMBaseEventProcessor method createRootRecordCategoryWithFixedName.

/**
 * Helper method for creating a root record category with a specified name.
 *
 * @param folder - container that will contain created root record category
 * @param name - the name that created root record category will have
 * @return created root record category with a specified name
 * @throws Exception
 */
public FolderData createRootRecordCategoryWithFixedName(FolderData folder, String name) throws Exception {
    FolderData createdFolder = null;
    String folderPath = folder.getPath();
    String newfilePlanComponentTitle = "title: " + name;
    // Build root category properties
    RecordCategory recordCategoryModel = RecordCategory.builder().name(name).nodeType(RECORD_CATEGORY_TYPE).properties(RecordCategoryProperties.builder().title(newfilePlanComponentTitle).description(EMPTY).build()).build();
    FilePlanAPI filePlansAPI = getRestAPIFactory().getFilePlansAPI();
    RecordCategory rootCategory = filePlansAPI.createRootRecordCategory(recordCategoryModel, folder.getId());
    String newRootCategoryId = rootCategory.getId();
    fileFolderService.createNewFolder(newRootCategoryId, RECORD_CATEGORY_CONTEXT, folderPath + "/" + name);
    // Increment counts
    fileFolderService.incrementFolderCount(folder.getContext(), folderPath, 1);
    createdFolder = fileFolderService.getFolder(newRootCategoryId);
    return createdFolder;
}
Also used : FilePlanAPI(org.alfresco.rest.rm.community.requests.gscore.api.FilePlanAPI) FolderData(org.alfresco.bm.cm.FolderData) RecordCategory(org.alfresco.rest.rm.community.model.recordcategory.RecordCategory)

Aggregations

RecordCategory (org.alfresco.rest.rm.community.model.recordcategory.RecordCategory)7 FolderData (org.alfresco.bm.cm.FolderData)6 DBObject (com.mongodb.DBObject)5 Event (org.alfresco.bm.event.Event)5 EventResult (org.alfresco.bm.event.EventResult)5 StopWatch (org.apache.commons.lang3.time.StopWatch)5 Test (org.junit.Test)5 RecordCategoryChild (org.alfresco.rest.rm.community.model.recordcategory.RecordCategoryChild)4 LoadSingleComponentUnitTest (org.alfresco.bm.dataload.LoadSingleComponentUnitTest)3 FilePlan (org.alfresco.rest.rm.community.model.fileplan.FilePlan)3 UserModel (org.alfresco.utility.model.UserModel)3 FilePlanAPI (org.alfresco.rest.rm.community.requests.gscore.api.FilePlanAPI)2