flakyness test

src/test/regress/expected/citus_drain_node.out

@@ -115,6 +115,13 @@ SELECT nodename, nodeport, COUNT(*)
  localhost |    57638 |     2
 (2 rows)
 
+-- flakyness test
+SELECT floor(random() * 2) + 1;
+ ?column?
+---------------------------------------------------------------------
+        2
+(1 row)
+
 RESET search_path;
 SET client_min_messages TO WARNING;
 DROP SCHEMA citus_drain_node CASCADE;

src/test/regress/sql/citus_drain_node.sql

@@ -56,6 +56,9 @@ SELECT nodename, nodeport, COUNT(*)
  WHERE placement.groupid = node.groupid
    AND node.noderole = 'primary' GROUP BY nodename, nodeport ORDER BY 1,2;
 
+-- flakyness test
+SELECT floor(random() * 2) + 1;
+
 RESET search_path;
 
 SET client_min_messages TO WARNING;