allow single arg in append_conf()

funbringer · funbringer · commit 04e74abc6c17 · 2018-02-28T15:10:38.000+03:00
diff --git a/testgres/node.py b/testgres/node.py
@@ -384,21 +384,22 @@ def get_auth_method(t):
 
         return self
 
-    def append_conf(self, filename, string):
+    @method_decorator(positional_args_hack(['filename', 'line']))
+    def append_conf(self, line, filename=PG_CONF_FILE):
         """
-        Append line to a config file (i.e. postgresql.conf).
+        Append line to a config file.
 
         Args:
-            filename: name of the config file.
-            string: string to be appended to config.
+            line: string to be appended to config.
+            filename: config file (postgresql.conf by default).
 
         Returns:
             This instance of PostgresNode.
         """
 
         config_name = os.path.join(self.data_dir, filename)
-        with io.open(config_name, "a") as conf:
-            conf.write(u"".join([string, '\n']))
+        with io.open(config_name, 'a') as conf:
+            conf.write(u''.join([line, '\n']))
 
         return self
 
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -171,7 +171,7 @@ def test_reload(self):
 
             # change client_min_messages and save old value
             cmm_old = node.execute('show client_min_messages')
-            node.append_conf('postgresql.conf', 'client_min_messages = DEBUG1')
+            node.append_conf('client_min_messages = DEBUG1')
 
             # reload config
             node.reload()
