Update BaseStateIntegrationTests.cs

Author: sierpinskid

Assets/Plugins/StreamChat/Tests/StatefulClient/BaseStateIntegrationTests.cs

@@ -49,11 +49,14 @@ protected static IEnumerable<AuthCredentials> OtherAdminUsersCredentials
         /// <summary>
         /// Create temp channel with random id that will be removed in [TearDown]
         /// </summary>
-        protected async Task<IStreamChannel> CreateUniqueTempChannelAsync(string name = null, bool watch = true)
+        protected async Task<IStreamChannel> CreateUniqueTempChannelAsync(string name = null, bool watch = true, StreamChatClient overrideClient = null)
         {
             var channelId = "random-channel-11111-" + Guid.NewGuid();
+            var client = overrideClient ?? Client;
 
-            var channelState = await Client.InternalGetOrCreateChannelWithIdAsync(ChannelType.Messaging, channelId, name, watch: watch);
+            var channelState = await TryAsync(() => client.InternalGetOrCreateChannelWithIdAsync(ChannelType.Messaging, channelId, name, watch: watch), 
+                result => result != null,
+                ex => ex is StreamApiException apiException && apiException.IsRateLimitExceededError());
             _tempChannels.Add(channelState);
             return channelState;
         }
@@ -151,19 +154,28 @@ protected static async Task WaitWithTimeoutAsync(Task task, int maxSeconds, stri
         /// <summary>
         /// Timeout will be doubled on each subsequent attempt. So max timeout = <see cref="initTimeoutMs"/> * 2^<see cref="maxAttempts"/>
         /// </summary>
-        protected static async Task<T> TryAsync<T>(Func<Task<T>> task, Predicate<T> successCondition, int maxAttempts = 20,
+        protected static async Task<T> TryAsync<T>(Func<Task<T>> task, Predicate<T> successCondition, Func<Exception, bool> continueOnException = null, int maxAttempts = 20,
             int initTimeoutMs = 150)
         {
             var response = default(T);
 
             var attemptsLeft = maxAttempts;
             while (attemptsLeft > 0)
             {
-                response = await task();
-
-                if (successCondition(response))
+                try
+                {
+                    response = await task();
+                    if (successCondition(response))
+                    {
+                        return response;
+                    }
+                }
+                catch (Exception ex)                 
                 {
-                    return response;
+                    if (continueOnException == null || !continueOnException(ex))
+                    {
+                        throw ex;
+                    }
                 }
 
                 var delay = initTimeoutMs * Math.Pow(2, (maxAttempts - attemptsLeft));
@@ -243,4 +255,4 @@ private async Task DeleteTempChannelsAsync()
         }
     }
 }
-#endif
+#endif