fixup! Clean up and add behave logging (#956)

changelog.rst

@@ -5,6 +5,12 @@ Bug fixes:
 ----------
 * Avoid error message on the server side if hstore extension is not installed in the current database (#991). (Thanks: `Marcin Cieślak`_)
 
+Internal:
+---------
+
+* (Fixup) Clean up and add behave logging. (Thanks: `Marcin Cieślak`_, `Dick Marinus`_)
+* Override VISUAL environment variable for behave tests. (Thanks: `Marcin Cieślak`_)
+
 2.0.2:
 ======
 

tests/features/steps/wrappers.py

@@ -56,6 +56,8 @@ def run_cli(context, run_args=None):
     cmd_parts = [cli_cmd] + run_args
     cmd = ' '.join(cmd_parts)
     context.cli = pexpect.spawnu(cmd, cwd=context.package_root)
+    context.logfile = StringIO()
+    context.cli.logfile = context.logfile
     context.exit_sent = False
     context.currentdb = context.conf['dbname']
     context.cli.sendline('\pset pager always')