change safe_psql() call to execute() in pubsub.py

zilder · zilder · commit 50e02ffe6100 · 2018-06-01T12:45:07.000+03:00
diff --git a/testgres/api.py b/testgres/api.py
@@ -28,7 +28,7 @@
 ...             replica.catchup()  # wait until changes are visible
 ...             print(replica.execute('postgres', 'select count(*) from test'))
 PostgresNode(name='...', port=..., base_dir='...')
-[(3L,)]
+[(3,)]
 """
 from .node import PostgresNode
 
diff --git a/testgres/pubsub.py b/testgres/pubsub.py
@@ -77,7 +77,7 @@ def drop(self, dbname=None, username=None):
         """
         Drop publication
         """
-        self.node.safe_psql(
+        self.node.execute(
             "drop publication {}".format(self.name),
             dbname=dbname,
             username=username)
