Examples with SearchEngineService io.gravitee.rest.api.service.search.SearchEngineService used on opensource projects

Example 1 with SearchEngineService

use of io.gravitee.rest.api.service.search.SearchEngineService in project gravitee-management-rest-api by gravitee-io.

the class PageService_ImportDirectoryTest method shouldImportDirectory.

@Test
public void shouldImportDirectory() throws Exception {
    // We mock the validateSafeContent method because the fetcher keeps sending the same json descriptor which is
    // not a swagger valid document (and modify the fetcher mock to produce valid desc is overkill)
    when(pageService.validateSafeContent(any(), any())).thenReturn(new ArrayList<>());
    PageSourceEntity pageSource = new PageSourceEntity();
    pageSource.setType("type");
    pageSource.setConfiguration(mapper.readTree("{}"));
    ImportPageEntity pageEntity = new ImportPageEntity();
    pageEntity.setSource(pageSource);
    pageEntity.setPublished(true);
    FetcherPlugin fetcherPlugin = mock(FetcherPlugin.class);
    when(fetcherPlugin.clazz()).thenReturn("io.gravitee.rest.api.service.PageService_ImportDirectoryMockFetcher");
    when(fetcherPlugin.configuration()).thenReturn(PageService_MockFilesFetcherConfiguration.class);
    when(fetcherPluginManager.get(any())).thenReturn(fetcherPlugin);
    Class<PageService_ImportDirectoryMockFetcher> mockFetcherClass = PageService_ImportDirectoryMockFetcher.class;
    when(fetcherPlugin.fetcher()).thenReturn(mockFetcherClass);
    PageService_MockFilesFetcherConfiguration fetcherConfiguration = new PageService_MockFilesFetcherConfiguration();
    when(fetcherConfigurationFactory.create(eq(PageService_MockFilesFetcherConfiguration.class), anyString())).thenReturn(fetcherConfiguration);
    AutowireCapableBeanFactory mockAutowireCapableBeanFactory = mock(AutowireCapableBeanFactory.class);
    when(applicationContext.getAutowireCapableBeanFactory()).thenReturn(mockAutowireCapableBeanFactory);
    Page newPage = mock(Page.class);
    PageSource ps = new PageSource();
    ps.setType(pageSource.getType());
    ps.setConfiguration(pageSource.getConfiguration());
    when(newPage.getId()).thenReturn(UuidString.generateRandom());
    when(newPage.isPublished()).thenReturn(Boolean.TRUE);
    when(newPage.getSource()).thenReturn(ps);
    when(newPage.getType()).thenReturn("MARKDOWN");
    when(newPage.getReferenceType()).thenReturn(PageReferenceType.ENVIRONMENT);
    when(newPage.getReferenceId()).thenReturn("envId");
    when(newPage.getVisibility()).thenReturn("PUBLIC");
    when(pageRepository.create(any())).thenReturn(newPage);
    when(graviteeDescriptorService.descriptorName()).thenReturn(".gravitee.json");
    List<PageEntity> pageEntities = pageService.importFiles(pageEntity, GraviteeContext.getCurrentEnvironment());
    assertNotNull(pageEntities);
    assertEquals(8, pageEntities.size());
    verify(searchEngineService, times(8)).index(any(), eq(false));
    // //////////////////////
    // check Folder creation
    // //////////////////////
    verify(pageRepository, times(3)).create(argThat(pageToCreate -> "FOLDER".equals(pageToCreate.getType())));
    // /src
    verify(pageRepository).create(argThat(pageToCreate -> "src".equals(pageToCreate.getName()) && "FOLDER".equals(pageToCreate.getType()) && null == pageToCreate.getParentId()));
    // /src/doc
    verify(pageRepository).create(argThat(pageToCreate -> "doc".equals(pageToCreate.getName()) && "FOLDER".equals(pageToCreate.getType()) && null != pageToCreate.getParentId()));
    // /src/folder.with.dot/
    verify(pageRepository).create(argThat(pageToCreate -> "folder.with.dot".equals(pageToCreate.getName()) && "FOLDER".equals(pageToCreate.getType()) && null != pageToCreate.getParentId()));
    // //////////////////////
    // verify files creation
    // //////////////////////
    verify(pageRepository, times(5)).create(argThat(pageToCreate -> pageToCreate.getType() != null && !"FOLDER".equals(pageToCreate.getType())));
    // /src/doc/m1.md
    verify(pageRepository).create(argThat(pageToCreate -> "m1".equals(pageToCreate.getName()) && "MARKDOWN".equals(pageToCreate.getType()) && null != pageToCreate.getParentId()));
    // /swagger.json
    verify(pageRepository).create(argThat(pageToCreate -> "swagger".equals(pageToCreate.getName()) && "SWAGGER".equals(pageToCreate.getType()) && null == pageToCreate.getParentId()));
    // /src/doc/sub.m11.md
    verify(pageRepository).create(argThat(pageToCreate -> "sub.m11".equals(pageToCreate.getName()) && "MARKDOWN".equals(pageToCreate.getType()) && null != pageToCreate.getParentId()));
    // /src/doc/m2.yaml
    verify(pageRepository).create(argThat(pageToCreate -> "m2".equals(pageToCreate.getName()) && "SWAGGER".equals(pageToCreate.getType()) && null != pageToCreate.getParentId()));
    // /src/folder.with.dot/m2.MD
    verify(pageRepository).create(argThat(pageToCreate -> "m2".equals(pageToCreate.getName()) && "MARKDOWN".equals(pageToCreate.getType()) && null != pageToCreate.getParentId()));
    verify(pageRevisionService, times(5)).create(any());
}

Example 2 with SearchEngineService

use of io.gravitee.rest.api.service.search.SearchEngineService in project gravitee-management-rest-api by gravitee-io.

the class ApiService_UpdateTest method shouldNotDuplicateLabels.

@Test
public void shouldNotDuplicateLabels() throws TechnicalException {
    prepareUpdate();
    when(existingApi.getLabels()).thenReturn(asList("label1", "label1"));
    when(api.getDefinition()).thenReturn("{\"id\": \"" + API_ID + "\",\"name\": \"" + API_NAME + "\",\"proxy\": {\"context_path\": \"/old\"} ,\"labels\": [\"public\"]}");
    final ApiEntity apiEntity = apiService.update(API_ID, existingApi);
    verify(apiRepository).update(argThat(api -> api.getLabels().size() == 1));
    assertNotNull(apiEntity);
    verify(searchEngineService, times(1)).index(any(), eq(false));
}