test: Add test for partial export (#29728)
## Description Tests for partial export flow #### PR fixes following issue(s) <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit - **Tests** - Introduced a new test suite for partial export functionality, ensuring resources like datasources and actions can be exported correctly under various configurations. <!-- end of auto-generated comment: release notes by coderabbit.ai -->
This commit is contained in:
Anagh Hegde
GitHub
parent
ad256ed64a
commit
71cd59fc4c
|
|
@ -0,0 +1,346 @@
|
||||||
|
package com.appsmith.server.services;
|
||||||
|
|
||||||
|
import com.appsmith.external.models.ActionConfiguration;
|
||||||
|
import com.appsmith.external.models.ActionDTO;
|
||||||
|
import com.appsmith.external.models.DBAuth;
|
||||||
|
import com.appsmith.external.models.Datasource;
|
||||||
|
import com.appsmith.external.models.DatasourceConfiguration;
|
||||||
|
import com.appsmith.external.models.DatasourceStorageDTO;
|
||||||
|
import com.appsmith.external.models.DefaultResources;
|
||||||
|
import com.appsmith.external.models.Property;
|
||||||
|
import com.appsmith.server.applications.base.ApplicationService;
|
||||||
|
import com.appsmith.server.datasources.base.DatasourceService;
|
||||||
|
import com.appsmith.server.domains.Application;
|
||||||
|
import com.appsmith.server.domains.GitApplicationMetadata;
|
||||||
|
import com.appsmith.server.domains.Plugin;
|
||||||
|
import com.appsmith.server.domains.User;
|
||||||
|
import com.appsmith.server.domains.Workspace;
|
||||||
|
import com.appsmith.server.dtos.ApplicationJson;
|
||||||
|
import com.appsmith.server.dtos.PageDTO;
|
||||||
|
import com.appsmith.server.dtos.PartialExportFileDTO;
|
||||||
|
import com.appsmith.server.exports.internal.PartialExportService;
|
||||||
|
import com.appsmith.server.helpers.MockPluginExecutor;
|
||||||
|
import com.appsmith.server.helpers.PluginExecutorHelper;
|
||||||
|
import com.appsmith.server.newpages.base.NewPageService;
|
||||||
|
import com.appsmith.server.plugins.base.PluginService;
|
||||||
|
import com.appsmith.server.repositories.ApplicationRepository;
|
||||||
|
import com.appsmith.server.repositories.CacheableRepositoryHelper;
|
||||||
|
import com.appsmith.server.repositories.PermissionGroupRepository;
|
||||||
|
import com.appsmith.server.repositories.PluginRepository;
|
||||||
|
import com.appsmith.server.repositories.ThemeRepository;
|
||||||
|
import com.appsmith.server.solutions.EnvironmentPermission;
|
||||||
|
import com.appsmith.server.solutions.PagePermission;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.junit.jupiter.api.BeforeEach;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
import org.junit.jupiter.api.extension.ExtendWith;
|
||||||
|
import org.mockito.Mockito;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.boot.test.context.SpringBootTest;
|
||||||
|
import org.springframework.boot.test.mock.mockito.MockBean;
|
||||||
|
import org.springframework.boot.test.mock.mockito.SpyBean;
|
||||||
|
import org.springframework.http.HttpMethod;
|
||||||
|
import org.springframework.security.test.context.support.WithUserDetails;
|
||||||
|
import org.springframework.test.context.junit.jupiter.SpringExtension;
|
||||||
|
import reactor.core.publisher.Flux;
|
||||||
|
import reactor.core.publisher.Mono;
|
||||||
|
import reactor.test.StepVerifier;
|
||||||
|
|
||||||
|
import java.time.Instant;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Set;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
|
||||||
|
@Slf4j
|
||||||
|
@ExtendWith(SpringExtension.class)
|
||||||
|
@SpringBootTest
|
||||||
|
public class PartialExportServiceTest {
|
||||||
|
|
||||||
|
private static final Map<String, Datasource> datasourceMap = new HashMap<>();
|
||||||
|
private static Plugin installedPlugin;
|
||||||
|
private static String workspaceId;
|
||||||
|
private static String defaultEnvironmentId;
|
||||||
|
private static String testAppId;
|
||||||
|
private static Datasource jsDatasource;
|
||||||
|
private static Plugin installedJsPlugin;
|
||||||
|
private static Boolean isSetupDone = false;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private ApplicationService applicationService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private NewPageService newPageService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private LayoutActionService layoutActionService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private LayoutCollectionService layoutCollectionService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private SessionUserService sessionUserService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private PartialExportService partialExportService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
ApplicationPageService applicationPageService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
PluginRepository pluginRepository;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
ApplicationRepository applicationRepository;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
DatasourceService datasourceService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
WorkspaceService workspaceService;
|
||||||
|
|
||||||
|
@MockBean
|
||||||
|
PluginExecutorHelper pluginExecutorHelper;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
ThemeRepository themeRepository;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
PermissionGroupRepository permissionGroupRepository;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
PermissionGroupService permissionGroupService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
EnvironmentPermission environmentPermission;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
PagePermission pagePermission;
|
||||||
|
|
||||||
|
@SpyBean
|
||||||
|
PluginService pluginService;
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
CacheableRepositoryHelper cacheableRepositoryHelper;
|
||||||
|
|
||||||
|
@BeforeEach
|
||||||
|
public void setup() {
|
||||||
|
Mockito.when(pluginExecutorHelper.getPluginExecutor(Mockito.any()))
|
||||||
|
.thenReturn(Mono.just(new MockPluginExecutor()));
|
||||||
|
|
||||||
|
if (Boolean.TRUE.equals(isSetupDone)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
User currentUser = sessionUserService.getCurrentUser().block();
|
||||||
|
Set<String> beforeCreatingWorkspace =
|
||||||
|
cacheableRepositoryHelper.getPermissionGroupsOfUser(currentUser).block();
|
||||||
|
log.info("Permission Groups for User before creating workspace: {}", beforeCreatingWorkspace);
|
||||||
|
installedPlugin = pluginRepository.findByPackageName("installed-plugin").block();
|
||||||
|
Workspace workspace = new Workspace();
|
||||||
|
workspace.setName("Import-Export-Test-Workspace");
|
||||||
|
Workspace savedWorkspace = workspaceService.create(workspace).block();
|
||||||
|
workspaceId = savedWorkspace.getId();
|
||||||
|
defaultEnvironmentId = workspaceService
|
||||||
|
.getDefaultEnvironmentId(workspaceId, environmentPermission.getExecutePermission())
|
||||||
|
.block();
|
||||||
|
Set<String> afterCreatingWorkspace =
|
||||||
|
cacheableRepositoryHelper.getPermissionGroupsOfUser(currentUser).block();
|
||||||
|
log.info("Permission Groups for User after creating workspace: {}", afterCreatingWorkspace);
|
||||||
|
|
||||||
|
log.info("Workspace ID: {}", workspaceId);
|
||||||
|
log.info("Workspace Role Ids: {}", workspace.getDefaultPermissionGroups());
|
||||||
|
log.info("Policy for created Workspace: {}", workspace.getPolicies());
|
||||||
|
log.info("Current User ID: {}", currentUser.getId());
|
||||||
|
|
||||||
|
Application testApplication = new Application();
|
||||||
|
testApplication.setName("Export-Application-Test-Application");
|
||||||
|
testApplication.setWorkspaceId(workspaceId);
|
||||||
|
testApplication.setUpdatedAt(Instant.now());
|
||||||
|
testApplication.setLastDeployedAt(Instant.now());
|
||||||
|
testApplication.setModifiedBy("some-user");
|
||||||
|
testApplication.setGitApplicationMetadata(new GitApplicationMetadata());
|
||||||
|
|
||||||
|
Application savedApplication = applicationPageService
|
||||||
|
.createApplication(testApplication, workspaceId)
|
||||||
|
.block();
|
||||||
|
testAppId = savedApplication.getId();
|
||||||
|
|
||||||
|
Datasource ds1 = new Datasource();
|
||||||
|
ds1.setName("DS1");
|
||||||
|
ds1.setWorkspaceId(workspaceId);
|
||||||
|
ds1.setPluginId(installedPlugin.getId());
|
||||||
|
final DatasourceConfiguration datasourceConfiguration = new DatasourceConfiguration();
|
||||||
|
datasourceConfiguration.setUrl("http://example.org/get");
|
||||||
|
datasourceConfiguration.setHeaders(List.of(new Property("X-Answer", "42")));
|
||||||
|
|
||||||
|
HashMap<String, DatasourceStorageDTO> storages1 = new HashMap<>();
|
||||||
|
storages1.put(
|
||||||
|
defaultEnvironmentId, new DatasourceStorageDTO(null, defaultEnvironmentId, datasourceConfiguration));
|
||||||
|
ds1.setDatasourceStorages(storages1);
|
||||||
|
|
||||||
|
Datasource ds2 = new Datasource();
|
||||||
|
ds2.setName("DS2");
|
||||||
|
ds2.setPluginId(installedPlugin.getId());
|
||||||
|
ds2.setWorkspaceId(workspaceId);
|
||||||
|
DatasourceConfiguration datasourceConfiguration2 = new DatasourceConfiguration();
|
||||||
|
DBAuth auth = new DBAuth();
|
||||||
|
auth.setPassword("awesome-password");
|
||||||
|
datasourceConfiguration2.setAuthentication(auth);
|
||||||
|
HashMap<String, DatasourceStorageDTO> storages2 = new HashMap<>();
|
||||||
|
storages2.put(
|
||||||
|
defaultEnvironmentId, new DatasourceStorageDTO(null, defaultEnvironmentId, datasourceConfiguration2));
|
||||||
|
ds2.setDatasourceStorages(storages2);
|
||||||
|
|
||||||
|
jsDatasource = new Datasource();
|
||||||
|
jsDatasource.setName("Default JS datasource");
|
||||||
|
jsDatasource.setWorkspaceId(workspaceId);
|
||||||
|
installedJsPlugin =
|
||||||
|
pluginRepository.findByPackageName("installed-js-plugin").block();
|
||||||
|
assert installedJsPlugin != null;
|
||||||
|
jsDatasource.setPluginId(installedJsPlugin.getId());
|
||||||
|
|
||||||
|
ds1 = datasourceService.create(ds1).block();
|
||||||
|
ds2 = datasourceService.create(ds2).block();
|
||||||
|
datasourceMap.put("DS1", ds1);
|
||||||
|
datasourceMap.put("DS2", ds2);
|
||||||
|
isSetupDone = true;
|
||||||
|
|
||||||
|
Mockito.when(pluginService.findAllByIdsWithoutPermission(Mockito.any(), Mockito.anyList()))
|
||||||
|
.thenReturn(Flux.fromIterable(List.of(installedPlugin, installedJsPlugin)));
|
||||||
|
}
|
||||||
|
|
||||||
|
private Application createGitConnectedApp(String applicationName) {
|
||||||
|
// Create application connected to git
|
||||||
|
Application testApplication = new Application();
|
||||||
|
testApplication.setName(applicationName);
|
||||||
|
testApplication.setWorkspaceId(workspaceId);
|
||||||
|
testApplication.setUpdatedAt(Instant.now());
|
||||||
|
testApplication.setLastDeployedAt(Instant.now());
|
||||||
|
testApplication.setModifiedBy("some-user");
|
||||||
|
testApplication.setGitApplicationMetadata(new GitApplicationMetadata());
|
||||||
|
GitApplicationMetadata gitData = new GitApplicationMetadata();
|
||||||
|
gitData.setBranchName("master");
|
||||||
|
gitData.setDefaultBranchName("master");
|
||||||
|
testApplication.setGitApplicationMetadata(gitData);
|
||||||
|
|
||||||
|
return applicationPageService
|
||||||
|
.createApplication(testApplication, workspaceId)
|
||||||
|
.flatMap(application1 -> {
|
||||||
|
application1.getGitApplicationMetadata().setDefaultApplicationId(application1.getId());
|
||||||
|
return applicationService.save(application1);
|
||||||
|
})
|
||||||
|
.block();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@WithUserDetails(value = "api_user")
|
||||||
|
void testGetPartialExport_nonGitConnectedApp_success() {
|
||||||
|
// Create an application with all resources
|
||||||
|
Application testApplication = new Application();
|
||||||
|
testApplication.setName("testGetPartialExport_nonGitConnectedApp_success");
|
||||||
|
testApplication.setWorkspaceId(workspaceId);
|
||||||
|
|
||||||
|
testApplication = applicationPageService
|
||||||
|
.createApplication(testApplication, workspaceId)
|
||||||
|
.block();
|
||||||
|
|
||||||
|
// Create pages
|
||||||
|
PageDTO page = new PageDTO();
|
||||||
|
page.setName("Page 2");
|
||||||
|
page.setApplicationId(testApplication.getId());
|
||||||
|
PageDTO savedPage = applicationPageService.createPage(page).block();
|
||||||
|
|
||||||
|
PartialExportFileDTO partialExportFileDTO = new PartialExportFileDTO();
|
||||||
|
partialExportFileDTO.setDatasourceList(List.of(
|
||||||
|
datasourceMap.get("DS1").getId(), datasourceMap.get("DS2").getId()));
|
||||||
|
|
||||||
|
// Get the partial export resources
|
||||||
|
Mono<ApplicationJson> partialExportFileDTOMono = partialExportService.getPartialExportResources(
|
||||||
|
testApplication.getId(), savedPage.getId(), null, partialExportFileDTO);
|
||||||
|
|
||||||
|
StepVerifier.create(partialExportFileDTOMono)
|
||||||
|
.assertNext(applicationJson -> {
|
||||||
|
assertThat(applicationJson.getDatasourceList().size()).isEqualTo(2);
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(0).getName())
|
||||||
|
.isEqualTo("DS1");
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(1).getName())
|
||||||
|
.isEqualTo("DS2");
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(0).getPluginId())
|
||||||
|
.isEqualTo("installed-plugin");
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(1).getPluginId())
|
||||||
|
.isEqualTo("installed-plugin");
|
||||||
|
})
|
||||||
|
.verifyComplete();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@WithUserDetails(value = "api_user")
|
||||||
|
public void testGetPartialExport_gitConnectedApp_branchResourceExported() {
|
||||||
|
Application application = createGitConnectedApp("testGetPartialExport_gitConnectedApp_branchResourceExported");
|
||||||
|
|
||||||
|
// update git branch name for page
|
||||||
|
PageDTO savedPage = new PageDTO();
|
||||||
|
savedPage.setName("Page 2");
|
||||||
|
savedPage.setApplicationId(application.getId());
|
||||||
|
DefaultResources defaultResources = new DefaultResources();
|
||||||
|
defaultResources.setApplicationId(application.getId());
|
||||||
|
defaultResources.setBranchName("master");
|
||||||
|
savedPage.setDefaultResources(defaultResources);
|
||||||
|
savedPage = applicationPageService
|
||||||
|
.createPageWithBranchName(savedPage, "master")
|
||||||
|
.block();
|
||||||
|
|
||||||
|
// Create Action
|
||||||
|
ActionDTO action = new ActionDTO();
|
||||||
|
action.setName("validAction");
|
||||||
|
action.setPageId(savedPage.getId());
|
||||||
|
action.setExecuteOnLoad(true);
|
||||||
|
ActionConfiguration actionConfiguration = new ActionConfiguration();
|
||||||
|
actionConfiguration.setHttpMethod(HttpMethod.GET);
|
||||||
|
actionConfiguration.setTimeoutInMillisecond("6000");
|
||||||
|
action.setActionConfiguration(actionConfiguration);
|
||||||
|
action.setDatasource(datasourceMap.get("DS1"));
|
||||||
|
|
||||||
|
ActionDTO savedAction =
|
||||||
|
layoutActionService.createSingleAction(action, Boolean.FALSE).block();
|
||||||
|
|
||||||
|
PartialExportFileDTO partialExportFileDTO = new PartialExportFileDTO();
|
||||||
|
partialExportFileDTO.setDatasourceList(List.of(
|
||||||
|
datasourceMap.get("DS1").getId(), datasourceMap.get("DS2").getId()));
|
||||||
|
partialExportFileDTO.setActionList(List.of(savedAction.getId()));
|
||||||
|
|
||||||
|
// Get the partial export resources
|
||||||
|
Mono<ApplicationJson> partialExportFileDTOMono = partialExportService.getPartialExportResources(
|
||||||
|
application.getId(), savedPage.getId(), "master", partialExportFileDTO);
|
||||||
|
|
||||||
|
StepVerifier.create(partialExportFileDTOMono)
|
||||||
|
.assertNext(applicationJson -> {
|
||||||
|
assertThat(applicationJson.getDatasourceList().size()).isEqualTo(2);
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(0).getName())
|
||||||
|
.isEqualTo("DS1");
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(1).getName())
|
||||||
|
.isEqualTo("DS2");
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(0).getPluginId())
|
||||||
|
.isEqualTo("installed-plugin");
|
||||||
|
assertThat(applicationJson.getDatasourceList().get(1).getPluginId())
|
||||||
|
.isEqualTo("installed-plugin");
|
||||||
|
assertThat(applicationJson.getActionList().size()).isEqualTo(1);
|
||||||
|
assertThat(applicationJson
|
||||||
|
.getActionList()
|
||||||
|
.get(0)
|
||||||
|
.getUnpublishedAction()
|
||||||
|
.getName())
|
||||||
|
.isEqualTo("validAction");
|
||||||
|
assertThat(applicationJson
|
||||||
|
.getActionList()
|
||||||
|
.get(0)
|
||||||
|
.getUnpublishedAction()
|
||||||
|
.getPageId())
|
||||||
|
.isEqualTo("Page 2");
|
||||||
|
assertThat(applicationJson.getActionList().get(0).getId()).isEqualTo("Page 2_validAction");
|
||||||
|
})
|
||||||
|
.verifyComplete();
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user