Fix flakiness of AuditLoggerAuthTest and CQLUserAuditTest by awaiting of default superuser role creation

patch by Dmitry Konstantinov; reviewed by Michael Semb Wever for CASSANDRA-19165

Author: netudima

test/unit/org/apache/cassandra/audit/AuditLoggerAuthTest.java

@@ -36,6 +36,7 @@
 import org.junit.Test;
 
 import org.apache.cassandra.ServerTestUtils;
+import org.apache.cassandra.auth.AuthTestUtils;
 import org.apache.cassandra.auth.CassandraAuthorizer;
 import org.apache.cassandra.auth.CassandraRoleManager;
 import org.apache.cassandra.auth.PasswordAuthenticator;
@@ -82,6 +83,7 @@ public static void setup() throws Exception
 
         SUPERUSER_SETUP_DELAY_MS.setLong(0);
         embedded = ServerTestUtils.startEmbeddedCassandraService();
+        AuthTestUtils.waitForExistingRoles();
 
         executeWithCredentials(
         Arrays.asList(getCreateRoleCql(TEST_USER, true, false, false),
@@ -368,6 +370,8 @@ private static void executeWithCredentials(List<String> queries, String username
             }
             catch (AuthenticationException e)
             {
+                if (expectedType == null)
+                    throw e;
                 authFailed = true;
             }
             catch (UnauthorizedException ue)

test/unit/org/apache/cassandra/transport/CQLUserAuditTest.java

@@ -44,6 +44,7 @@
 import org.apache.cassandra.audit.AuditLogEntryType;
 import org.apache.cassandra.audit.AuditLogManager;
 import org.apache.cassandra.audit.DiagnosticEventAuditLogger;
+import org.apache.cassandra.auth.AuthTestUtils;
 import org.apache.cassandra.auth.CassandraRoleManager;
 import org.apache.cassandra.auth.PasswordAuthenticator;
 import org.apache.cassandra.config.DatabaseDescriptor;
@@ -78,6 +79,8 @@ public static void setup() throws Exception
         SUPERUSER_SETUP_DELAY_MS.setLong(0);
 
         embedded = ServerTestUtils.startEmbeddedCassandraService();
+        AuthTestUtils.waitForExistingRoles();
+
 
         executeAs(Arrays.asList("CREATE ROLE testuser WITH LOGIN = true AND SUPERUSER = false AND PASSWORD = 'foo'",
                                 "CREATE ROLE testuser_nologin WITH LOGIN = false AND SUPERUSER = false AND PASSWORD = 'foo'",
@@ -214,20 +217,23 @@ private static ArrayList<AuditEvent> executeAs(List<String> queries, String user
                                                    AuditLogEntryType expectedAuthType) throws Exception
     {
         boolean authFailed = false;
-        Cluster cluster = Cluster.builder().addContactPoints(InetAddress.getLoopbackAddress())
+        try(Cluster cluster = Cluster.builder().addContactPoints(InetAddress.getLoopbackAddress())
                                  .withoutJMXReporting()
                                  .withCredentials(username, password)
-                                 .withPort(DatabaseDescriptor.getNativeTransportPort()).build();
-        try (Session session = cluster.connect())
-        {
-            for (String query : queries)
-                session.execute(query);
-        }
-        catch (AuthenticationException e)
+                                 .withPort(DatabaseDescriptor.getNativeTransportPort()).build())
         {
-            authFailed = true;
+            try (Session session = cluster.connect())
+            {
+                for (String query : queries)
+                    session.execute(query);
+            }
+            catch (AuthenticationException e)
+            {
+                if (expectedAuthType == null)
+                    throw e;
+                authFailed = true;
+            }
         }
-        cluster.close();
 
         if (expectedAuthType == null) return null;