Make the poll query test a little more robust.

2018-03-07 16:48:32 +00:00 · 2018-03-07 16:48:32 +00:00 · 1136e37cf0
parent abf0b1a7ae
commit 1136e37cf0
1 changed files with 4 additions and 2 deletions
--- a/web/pgadmin/tools/sqleditor/tests/test_poll_query_tool.py
+++ b/web/pgadmin/tools/sqleditor/tests/test_poll_query_tool.py
@ -99,9 +99,11 @@ NOTICE:  Hello, world!
            self.assertEquals(response.status_code, 200)
            response_data = json.loads(response.data.decode('utf-8'))

+            if self.expected_message[cnt] is not None:
                # Check the returned messages
-            self.assertEquals(self.expected_message[cnt],
+                self.assertIn(self.expected_message[cnt],
                              response_data['data']['additional_messages'])
+
            # Check the output
            self.assertEquals(self.expected_result[cnt],
                              response_data['data']['result'][0][0])