Search in sources :

Example 1 with FileResultStorageEngine

use of com.yahoo.elide.async.service.storageengine.FileResultStorageEngine in project elide by yahoo.

the class GraphQLTableExportOperationTest method setupMocks.

@BeforeEach
public void setupMocks(@TempDir Path tempDir) {
    dataStore = new HashMapDataStore(DefaultClassScanner.getInstance(), new HashSet<>(Arrays.asList(TableExport.class.getPackage(), ArtifactGroup.class.getPackage())));
    Map<String, Class<? extends Check>> map = new HashMap<>();
    map.put(AsyncAPIInlineChecks.AsyncAPIOwner.PRINCIPAL_IS_OWNER, AsyncAPIInlineChecks.AsyncAPIOwner.class);
    map.put(AsyncAPIInlineChecks.AsyncAPIAdmin.PRINCIPAL_IS_ADMIN, AsyncAPIInlineChecks.AsyncAPIAdmin.class);
    map.put(AsyncAPIInlineChecks.AsyncAPIStatusValue.VALUE_IS_CANCELLED, AsyncAPIInlineChecks.AsyncAPIStatusValue.class);
    map.put(AsyncAPIInlineChecks.AsyncAPIStatusQueuedValue.VALUE_IS_QUEUED, AsyncAPIInlineChecks.AsyncAPIStatusQueuedValue.class);
    elide = new Elide(new ElideSettingsBuilder(dataStore).withEntityDictionary(EntityDictionary.builder().checks(map).build()).withAuditLogger(new Slf4jLogger()).withExportApiPath("/export").build());
    elide.doScans();
    user = mock(User.class);
    requestScope = mock(RequestScope.class);
    asyncExecutorService = mock(AsyncExecutorService.class);
    engine = new FileResultStorageEngine(tempDir.toString(), false);
    when(asyncExecutorService.getElide()).thenReturn(elide);
    when(requestScope.getApiVersion()).thenReturn(NO_VERSION);
    when(requestScope.getUser()).thenReturn(user);
    when(requestScope.getElideSettings()).thenReturn(elide.getElideSettings());
    when(requestScope.getBaseUrlEndPoint()).thenReturn("https://elide.io");
}
Also used : FileResultStorageEngine(com.yahoo.elide.async.service.storageengine.FileResultStorageEngine) User(com.yahoo.elide.core.security.User) HashMap(java.util.HashMap) Slf4jLogger(com.yahoo.elide.core.audit.Slf4jLogger) Check(com.yahoo.elide.core.security.checks.Check) AsyncAPIInlineChecks(com.yahoo.elide.async.models.security.AsyncAPIInlineChecks) RequestScope(com.yahoo.elide.core.RequestScope) ElideSettingsBuilder(com.yahoo.elide.ElideSettingsBuilder) AsyncExecutorService(com.yahoo.elide.async.service.AsyncExecutorService) HashMapDataStore(com.yahoo.elide.core.datastore.inmemory.HashMapDataStore) Elide(com.yahoo.elide.Elide) HashSet(java.util.HashSet) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 2 with FileResultStorageEngine

use of com.yahoo.elide.async.service.storageengine.FileResultStorageEngine in project elide by yahoo.

the class JsonAPITableExportOperationTest method setupMocks.

@BeforeEach
public void setupMocks(@TempDir Path tempDir) {
    dataStore = new HashMapDataStore(DefaultClassScanner.getInstance(), new HashSet<>(Arrays.asList(TableExport.class.getPackage(), ArtifactGroup.class.getPackage())));
    Map<String, Class<? extends Check>> map = new HashMap<>();
    map.put(AsyncAPIInlineChecks.AsyncAPIOwner.PRINCIPAL_IS_OWNER, AsyncAPIInlineChecks.AsyncAPIOwner.class);
    map.put(AsyncAPIInlineChecks.AsyncAPIAdmin.PRINCIPAL_IS_ADMIN, AsyncAPIInlineChecks.AsyncAPIAdmin.class);
    map.put(AsyncAPIInlineChecks.AsyncAPIStatusValue.VALUE_IS_CANCELLED, AsyncAPIInlineChecks.AsyncAPIStatusValue.class);
    map.put(AsyncAPIInlineChecks.AsyncAPIStatusQueuedValue.VALUE_IS_QUEUED, AsyncAPIInlineChecks.AsyncAPIStatusQueuedValue.class);
    elide = new Elide(new ElideSettingsBuilder(dataStore).withEntityDictionary(EntityDictionary.builder().checks(map).build()).withAuditLogger(new Slf4jLogger()).withExportApiPath("/export").build());
    elide.doScans();
    user = mock(User.class);
    requestScope = mock(RequestScope.class);
    asyncExecutorService = mock(AsyncExecutorService.class);
    engine = new FileResultStorageEngine(tempDir.toString(), true);
    when(asyncExecutorService.getElide()).thenReturn(elide);
    when(requestScope.getApiVersion()).thenReturn(NO_VERSION);
    when(requestScope.getUser()).thenReturn(user);
    when(requestScope.getElideSettings()).thenReturn(elide.getElideSettings());
    when(requestScope.getBaseUrlEndPoint()).thenReturn("https://elide.io");
}
Also used : FileResultStorageEngine(com.yahoo.elide.async.service.storageengine.FileResultStorageEngine) User(com.yahoo.elide.core.security.User) HashMap(java.util.HashMap) Slf4jLogger(com.yahoo.elide.core.audit.Slf4jLogger) Check(com.yahoo.elide.core.security.checks.Check) AsyncAPIInlineChecks(com.yahoo.elide.async.models.security.AsyncAPIInlineChecks) RequestScope(com.yahoo.elide.core.RequestScope) ElideSettingsBuilder(com.yahoo.elide.ElideSettingsBuilder) AsyncExecutorService(com.yahoo.elide.async.service.AsyncExecutorService) HashMapDataStore(com.yahoo.elide.core.datastore.inmemory.HashMapDataStore) Elide(com.yahoo.elide.Elide) HashSet(java.util.HashSet) BeforeEach(org.junit.jupiter.api.BeforeEach)

Aggregations

Elide (com.yahoo.elide.Elide)2 ElideSettingsBuilder (com.yahoo.elide.ElideSettingsBuilder)2 AsyncAPIInlineChecks (com.yahoo.elide.async.models.security.AsyncAPIInlineChecks)2 AsyncExecutorService (com.yahoo.elide.async.service.AsyncExecutorService)2 FileResultStorageEngine (com.yahoo.elide.async.service.storageengine.FileResultStorageEngine)2 RequestScope (com.yahoo.elide.core.RequestScope)2 Slf4jLogger (com.yahoo.elide.core.audit.Slf4jLogger)2 HashMapDataStore (com.yahoo.elide.core.datastore.inmemory.HashMapDataStore)2 User (com.yahoo.elide.core.security.User)2 Check (com.yahoo.elide.core.security.checks.Check)2 HashMap (java.util.HashMap)2 HashSet (java.util.HashSet)2 BeforeEach (org.junit.jupiter.api.BeforeEach)2