From 56d53b191a50deeecb782a1e4b723e906ad17b4f Mon Sep 17 00:00:00 2001
From: Pedro Igor <pigor.craveiro@gmail.com>
Date: Fri, 27 Dec 2019 16:21:34 -0300
Subject: [PATCH] [KEYCLOAK-8779] - Fixing PartialImportTest

---
 .../testsuite/admin/partialimport/PartialImportTest.java | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/testsuite/integration-arquillian/tests/base/src/test/java/org/keycloak/testsuite/admin/partialimport/PartialImportTest.java b/testsuite/integration-arquillian/tests/base/src/test/java/org/keycloak/testsuite/admin/partialimport/PartialImportTest.java
index 4a8907d638..65575f50aa 100644
--- a/testsuite/integration-arquillian/tests/base/src/test/java/org/keycloak/testsuite/admin/partialimport/PartialImportTest.java
+++ b/testsuite/integration-arquillian/tests/base/src/test/java/org/keycloak/testsuite/admin/partialimport/PartialImportTest.java
@@ -41,6 +41,7 @@ import org.keycloak.representations.idm.UserRepresentation;
 import org.keycloak.testsuite.AbstractAuthTest;
 import org.keycloak.testsuite.Assert;
 import org.keycloak.testsuite.admin.ApiUtil;
+import org.keycloak.testsuite.arquillian.annotation.EnableFeature;
 import org.keycloak.testsuite.util.AssertAdminEvents;
 import org.keycloak.testsuite.util.RealmBuilder;
 
@@ -61,6 +62,8 @@ import org.keycloak.admin.client.resource.AuthorizationResource;
 import org.keycloak.common.constants.ServiceAccountConstants;
 import org.keycloak.partialimport.ResourceType;
 import org.keycloak.representations.idm.authorization.ResourceServerRepresentation;
+
+import static org.keycloak.common.Profile.Feature.UPLOAD_SCRIPTS;
 import static org.keycloak.testsuite.auth.page.AuthRealm.MASTER;
 import org.keycloak.util.JsonSerialization;
 
@@ -442,6 +445,7 @@ public class PartialImportTest extends AbstractAuthTest {
         }
     }
 
+    @EnableFeature(value = UPLOAD_SCRIPTS, skipRestart = true)
     @Test
     public void testAddClientsWithServiceAccountsAndAuthorization() throws IOException {
         setFail();
@@ -591,6 +595,7 @@ public class PartialImportTest extends AbstractAuthTest {
         testSkip();
     }
 
+    @EnableFeature(value = UPLOAD_SCRIPTS, skipRestart = true)
     @Test
     public void testAddClientsSkipWithServiceAccountsAndAuthorization() throws IOException {
         addClients(true);
@@ -647,6 +652,7 @@ public class PartialImportTest extends AbstractAuthTest {
         testOverwrite();
     }
 
+    @EnableFeature(value = UPLOAD_SCRIPTS, skipRestart = true)
     @Test
     public void testAddClientsOverwriteWithServiceAccountsAndAuthorization() throws IOException {
         addClients(true);
@@ -658,6 +664,7 @@ public class PartialImportTest extends AbstractAuthTest {
         assertEquals(NUM_ENTITIES * 2, results.getOverwritten());
     }
 
+    @EnableFeature(value = UPLOAD_SCRIPTS, skipRestart = true)
     @Test
     public void testAddClientsOverwriteServiceAccountsWithNoServiceAccounts() throws IOException {
         addClients(true);
@@ -733,6 +740,7 @@ public class PartialImportTest extends AbstractAuthTest {
         assertEquals(NUM_ENTITIES * NUM_RESOURCE_TYPES, results.getSkipped());
     }
 
+    @EnableFeature(value = UPLOAD_SCRIPTS, skipRestart = true)
     @Test
     public void testEverythingSkipWithServiceAccounts() throws IOException {
         setSkip();
@@ -749,6 +757,7 @@ public class PartialImportTest extends AbstractAuthTest {
         assertEquals(NUM_ENTITIES * NUM_RESOURCE_TYPES, results.getOverwritten());
     }
 
+    @EnableFeature(value = UPLOAD_SCRIPTS, skipRestart = true)
     @Test
     public void testEverythingOverwriteWithServiceAccounts() throws IOException {
         setOverwrite();