RavenDB-21956 Cleanup queues after tests

djordjedjukic · Apr 12, 2024 · 9036847 · 9036847
1 parent a683a77
commit 9036847
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/test/SlowTests/Server/Documents/ETL/Queue/AzureQueueStorageEtlTestBase.cs b/test/SlowTests/Server/Documents/ETL/Queue/AzureQueueStorageEtlTestBase.cs
@@ -107,4 +107,21 @@ protected string GenerateLargeString()
 
         return builder.ToString();
     }
+
+    private void CleanupQueues()
+    {
+        QueueServiceClient client = new(ConnectionString);
+        var queues = client.GetQueues();
+
+        foreach (var queue in queues)
+        {
+            client.DeleteQueue(queue.Name);
+        }
+    }
+
+    public override void Dispose()
+    {
+        base.Dispose();
+        CleanupQueues();
+    }
 }
diff --git a/test/SlowTests/Server/Documents/ETL/Queue/AzureQueueStorageEtlTests.cs b/test/SlowTests/Server/Documents/ETL/Queue/AzureQueueStorageEtlTests.cs
@@ -101,7 +101,7 @@ public void Error_if_script_does_not_contain_any_loadTo_method()
         {
             Name = "test",
             ConnectionStringName = "test",
-            BrokerType = QueueBrokerType.RabbitMq,
+            BrokerType = QueueBrokerType.AzureQueueStorage,
             Transforms =
             {
                 new Transformation
@@ -114,9 +114,9 @@ public void Error_if_script_does_not_contain_any_loadTo_method()
         config.Initialize(new QueueConnectionString
         {
             Name = "Foo",
-            BrokerType = QueueBrokerType.RabbitMq,
-            RabbitMqConnectionSettings =
-                new RabbitMqConnectionSettings() { ConnectionString = "amqp://guest:guest@localhost:5672/" }
+            BrokerType = QueueBrokerType.AzureQueueStorage,
+            AzureQueueStorageConnectionSettings =
+                new AzureQueueStorageConnectionSettings() { ConnectionString = ConnectionString }
         });
 
         List<string> errors;