From a3a969b6e80e94c94db22ebe432a00b080bddd52 Mon Sep 17 00:00:00 2001
From: Emelia Lei <wlei29@bloomberg.net>
Date: Wed, 29 Jan 2025 17:15:59 -0500
Subject: [PATCH] Add integration tests

Signed-off-by: Emelia Lei <wlei29@bloomberg.net>
---
 src/integration-tests/test_migrate_to_csl.py | 55 ++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 src/integration-tests/test_migrate_to_csl.py

diff --git a/src/integration-tests/test_migrate_to_csl.py b/src/integration-tests/test_migrate_to_csl.py
new file mode 100644
index 000000000..59946f466
--- /dev/null
+++ b/src/integration-tests/test_migrate_to_csl.py
@@ -0,0 +1,55 @@
+# Copyright 2025 Bloomberg Finance L.P.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+Testing migrating to CSL.
+"""
+import blazingmq.dev.it.testconstants as tc
+from blazingmq.dev.it.fixtures import (  # pylint: disable=unused-import
+    Cluster,
+    multi_node,
+)
+
+
+def test_assign_queue(multi_node: Cluster):
+    """
+    This test is to make sure when a queue assignment happens,
+    it goes thought the CSL path, not the non-CSL.
+    """
+    proxies = multi_node.proxy_cycle()
+    proxy = next(proxies)
+
+    leader = multi_node.last_known_leader
+    members = multi_node.nodes(exclude=leader)
+
+    # find the two nodes which are not the known leader
+    uri = tc.URI_PRIORITY_SC
+
+    producer = proxy.create_client("producer")
+    producer.open(uri, flags=["write"], succeed=True)
+
+    timeout = 1
+
+    for member in members:
+        assert member.outputs_regex(
+            "(Applying cluster message with type = UPDATE).*(queueAssignmentAdvisory)",
+            timeout,
+        )
+        assert member.outputs_regex(
+            "(Committed advisory).*queueAssignmentAdvisory", timeout
+        )
+        assert not member.outputs_regex(
+            "'QueueUnAssignmentAdvisory' will be applied to", timeout
+        )