From d946fb1862dacf3569d7d47759427d1460b64d11 Mon Sep 17 00:00:00 2001
From: Gevorg Davoian <gdavoian@mirantis.com>
Date: Tue, 19 Jul 2016 12:53:27 +0300
Subject: [PATCH] Fix pika functional tests

Change-Id: I05f2cbd914857da7a75ca068e99614156797d1ed
Closes-Bug: #1599777
Depends-On: Ic6acc5d006344e08c219724e488fc9222786d849
---
 oslo_messaging/tests/functional/test_rabbitmq.py | 15 +++++++++++----
 oslo_messaging/tests/utils.py                    |  3 +++
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/oslo_messaging/tests/functional/test_rabbitmq.py b/oslo_messaging/tests/functional/test_rabbitmq.py
index 74fb73e08..9a0427eda 100644
--- a/oslo_messaging/tests/functional/test_rabbitmq.py
+++ b/oslo_messaging/tests/functional/test_rabbitmq.py
@@ -34,11 +34,15 @@ class ConnectedPortMatcher(object):
 
 
 class RabbitMQFailoverTests(test_utils.BaseTestCase):
+    DRIVERS = [
+        "rabbit",
+    ]
+
     def test_failover_scenario(self):
-        # NOTE(sileht): run this test only if functionnal suite run of a driver
+        # NOTE(sileht): run this test only if functional suite run of a driver
         # that use rabbitmq as backend
         self.driver = os.environ.get('TRANSPORT_DRIVER')
-        if self.driver not in ["pika", "rabbit"]:
+        if self.driver not in self.DRIVERS:
             self.skipTest("TRANSPORT_DRIVER is not set to a rabbit driver")
 
         # NOTE(sileht): Allow only one response at a time, to
@@ -51,7 +55,6 @@ class RabbitMQFailoverTests(test_utils.BaseTestCase):
                     rabbit_retry_backoff=0,
                     group='oslo_messaging_rabbit')
 
-        # 
         self.pifpaf = self.useFixture(rabbitmq.RabbitMQDriver(cluster=True,
                                                               port=5692))
 
@@ -60,6 +63,10 @@ class RabbitMQFailoverTests(test_utils.BaseTestCase):
         self.n2 = self.pifpaf.env["PIFPAF_RABBITMQ_NODENAME2"]
         self.n3 = self.pifpaf.env["PIFPAF_RABBITMQ_NODENAME3"]
 
+        # NOTE(gdavoian): additional tweak for pika driver
+        if self.driver == "pika":
+            self.url = self.url.replace("rabbit", "pika")
+
         # ensure connections will be establish to the first node
         self.pifpaf.stop_node(self.n2)
         self.pifpaf.stop_node(self.n3)
@@ -118,7 +125,7 @@ class RabbitMQFailoverTests(test_utils.BaseTestCase):
         rpc_server = self.servers.servers[0].server
         # FIXME(sileht): Check other connections
         connections = [
-            rpc_server.listener._poll_style_listener._connection
+            rpc_server.listener._connection
         ]
         for conn in connections:
             self.assertEqual(
diff --git a/oslo_messaging/tests/utils.py b/oslo_messaging/tests/utils.py
index aa2a56343..c5fca8711 100644
--- a/oslo_messaging/tests/utils.py
+++ b/oslo_messaging/tests/utils.py
@@ -39,6 +39,9 @@ class BaseTestCase(base.BaseTestCase):
         self.messaging_conf.transport_driver = 'fake'
         self.conf = self.messaging_conf.conf
 
+        self.conf.project = 'project'
+        self.conf.prog = 'prog'
+
         moxfixture = self.useFixture(moxstubout.MoxStubout())
         self.mox = moxfixture.mox
         self.stubs = moxfixture.stubs