From 40d4deecf3354f951fbb5869fa8d2c612b47e02e Mon Sep 17 00:00:00 2001
From: Michal Jurc <mjurc@redhat.com>
Date: Wed, 7 Jun 2017 09:05:10 +0200
Subject: [PATCH] [WFLY-8904] Fix AuthenticationTestCase and
 GetCallerPrincipalTestCase for Elytron and unignore it

* AuthenticationTestCase.testAuthentication_TwoBeans_ReAuth__BadPwd_ViaServlet: adding an expected exception for Elytron.
* Modifying GetCallerPrincipalTestCase to Elytron lifecycle behaviour.
* Unignoring the test cases with Elytron profile.
---
 .../ejb/security/AuthenticationTestCase.java      | 15 ++++++---------
 .../GetCallerPrincipalTestCase.java               | 13 +++++--------
 2 files changed, 11 insertions(+), 17 deletions(-)

diff --git a/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/AuthenticationTestCase.java b/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/AuthenticationTestCase.java
index 9f80bad6330b..905c944f462b 100644
--- a/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/AuthenticationTestCase.java
+++ b/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/AuthenticationTestCase.java
@@ -57,16 +57,15 @@
 import org.jboss.as.test.integration.security.common.AbstractSecurityDomainSetup;
 import org.jboss.as.test.shared.TestSuiteEnvironment;
 import org.jboss.as.test.shared.integration.ejb.security.Util;
-import org.jboss.as.test.shared.util.AssumeTestGroupUtil;
 import org.jboss.logging.Logger;
 import org.jboss.shrinkwrap.api.Archive;
 import org.jboss.shrinkwrap.api.ShrinkWrap;
 import org.jboss.shrinkwrap.api.asset.StringAsset;
 import org.jboss.shrinkwrap.api.spec.WebArchive;
-import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.experimental.categories.Category;
 import org.junit.runner.RunWith;
+import org.wildfly.security.auth.server.SecurityDomain;
 
 /**
  * Test case to hold the authentication scenarios, these range from calling a servlet which calls a bean to calling a bean which
@@ -85,12 +84,6 @@ public class AuthenticationTestCase {
 
     private static final Logger log = Logger.getLogger(AuthenticationTestCase.class.getName());
 
-    @BeforeClass
-    public static void beforeClass() {
-        //Conditionally ignore all the tests, although only testICIR_TwoBeans_ViaServlet() needs this treatment.
-        //There seems to be an issue with the system property not getting propagated to the server
-        AssumeTestGroupUtil.assumeElytronProfileTestsEnabled();
-    }
     /*
      * Authentication Scenarios
      *
@@ -254,7 +247,11 @@ public void testAuthentication_TwoBeans_ReAuth__BadPwd_ViaServlet() throws Excep
             getWhoAmI("?method=doubleWhoAmI&username=user2&password=bad_password");
             fail("Expected IOException");
         } catch (IOException e) {
-            assertTrue(e.getMessage().contains("javax.ejb.EJBAccessException"));
+            if (SecurityDomain.getCurrent() == null) {
+                assertTrue(e.getMessage().contains("javax.ejb.EJBAccessException"));
+            } else {
+                assertTrue(e.getMessage().contains("javax.ejb.EJBException: java.lang.SecurityException: ELY01151"));
+            }
         }
     }
 
diff --git a/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/callerprincipal/GetCallerPrincipalTestCase.java b/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/callerprincipal/GetCallerPrincipalTestCase.java
index 45e31aecc671..dc248779763e 100644
--- a/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/callerprincipal/GetCallerPrincipalTestCase.java
+++ b/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/ejb/security/callerprincipal/GetCallerPrincipalTestCase.java
@@ -40,7 +40,6 @@
 import javax.naming.NamingException;
 
 import org.jboss.as.test.shared.integration.ejb.security.Util;
-import org.jboss.as.test.shared.util.AssumeTestGroupUtil;
 import org.junit.Assert;
 
 import org.jboss.arquillian.container.test.api.Deployer;
@@ -63,7 +62,6 @@
 import org.jboss.shrinkwrap.api.spec.JavaArchive;
 import org.jboss.staxmapper.XMLElementReader;
 import org.jboss.staxmapper.XMLElementWriter;
-import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.experimental.categories.Category;
 import org.junit.runner.RunWith;
@@ -97,11 +95,6 @@ public class GetCallerPrincipalTestCase {
 
     public static final String LOCAL_USER = "$local";
 
-    @BeforeClass
-    public static void beforeClass() {
-        AssumeTestGroupUtil.assumeElytronProfileTestsEnabled();
-    }
-
     @ArquillianResource
     Deployer deployer;
 
@@ -224,7 +217,11 @@ public void testStatefulLifecycle() throws Exception {
 
             bean.remove();
 
-            Assert.assertEquals(LOCAL_USER +  "stop", results.getSfsb("predestroy"));
+            if (System.getProperty("elytron") == null) {
+                Assert.assertEquals(LOCAL_USER + "stop", results.getSfsb("predestroy"));
+            } else {
+                Assert.assertEquals(ANONYMOUS + "stop", results.getSfsb("predestroy"));
+            }
             return null;
         };
         try {