feature: add 'exec' returning a process and deprecates 'exec_command'

dcos_test_utils/dcos_cli.py

@@ -16,6 +16,8 @@
 
 import requests
 
+import deprecation
+
 log = logging.getLogger(__name__)
 
 DCOS_CLI_URL = os.getenv('DCOS_CLI_URL', 'https://downloads.dcos.io/binaries/cli/linux/x86-64/dcos-1.12/dcos')
@@ -92,6 +94,45 @@ def clear_cli_dir():
         if os.path.exists(path):
             shutil.rmtree(path)
 
+    def exec(self, cmd: str, stdin=None, check=True) -> subprocess.CompletedProcess:
+        """Execute CLI command and processes result.
+
+        This method expects that process won't block.
+
+        :param cmd: Program and arguments
+        :type cmd: str
+        :param stdin: File to use for stdin
+        :param check: Does it check for raised errors
+        :type stdin: File
+        :returns: A tuple with stdout and stderr
+        :rtype: subprocess.CompletedProcess
+        """
+
+        log.info('CMD: {!r}'.format(cmd))
+
+        try:
+            process = subprocess.run(
+                cmd,
+                stdin=stdin,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                env=self.env,
+                check=check)
+        except subprocess.CalledProcessError as e:
+            stderr = e.stderr.decode('utf-8')
+            stdout = e.stdout.decode('utf-8')
+            log.error('STDOUT: {}'.format(stdout))
+            log.error('STDERR: {}'.format(stderr))
+            raise
+
+        stdout, stderr = process.stdout.decode('utf-8'), process.stderr.decode('utf-8')
+
+        log.info('STDOUT: {}'.format(stdout))
+        log.info('STDERR: {}'.format(stderr))
+
+        return process
+
+    @deprecation.deprecated(details="Deprecated in favor of the `exec` function. DCOS-44823")
     def exec_command(self, cmd: str, stdin=None) -> tuple:
         """Execute CLI command and processes result.
 

requirements.txt

@@ -3,5 +3,6 @@ pytest
 requests
 retrying
 tox
+deprecation
 
 sphinx

tests/test_dcos_cli.py

@@ -1,9 +1,11 @@
-import pytest
 import subprocess
 
+import deprecation
+import pytest
 from dcos_test_utils import dcos_cli
 
 
+@deprecation.deprecated(details="Deprecated in favor of the `exec` function. DCOS-44823")
 def test_exec_command(caplog):
     cli = dcos_cli.DcosCli('')
     stdout, stderr = cli.exec_command(
@@ -15,10 +17,43 @@ def test_exec_command(caplog):
     assert any(rec.message.startswith('STDOUT:') for rec in caplog.records)
     assert any(rec.message.startswith('STDERR:') for rec in caplog.records)
 
-
+@deprecation.deprecated(details="Deprecated in favor of the `exec` function. DCOS-44823")
 def test_exec_command_fail(caplog):
     cli = dcos_cli.DcosCli('')
     with pytest.raises(subprocess.CalledProcessError):
         cli.exec_command(['/bin/sh', '-c', 'does-not-exist'])
     assert any(rec.message.startswith('CMD:') for rec in caplog.records)
     assert any(rec.message.startswith('STDERR:') for rec in caplog.records)
+
+def test_exec(caplog):
+    cli = dcos_cli.DcosCli('')
+    process = cli.exec(
+        ['/bin/sh', '-c', 'echo "hello, world!"']
+    )
+    assert process.stdout == b'hello, world!\n'
+    assert process.stderr == b''
+    assert process.returncode == 0
+    assert any(rec.message.startswith('CMD:') for rec in caplog.records)
+    assert any(rec.message.startswith('STDOUT:') for rec in caplog.records)
+    assert any(rec.message.startswith('STDERR:') for rec in caplog.records)
+
+def test_exec_fail(caplog):
+    cli = dcos_cli.DcosCli('')
+    with pytest.raises(subprocess.CalledProcessError):
+        cli.exec(['/bin/sh', '-c', 'does-not-exist'])
+    assert any(rec.message.startswith('CMD:') for rec in caplog.records)
+    assert any(rec.message.startswith('STDERR:') for rec in caplog.records)
+    assert any(rec.message.startswith('STDOUT:') for rec in caplog.records)
+
+def test_exec_fail_without_check(caplog):
+    cli = dcos_cli.DcosCli('')
+    process = cli.exec(
+        ['/bin/sh', '-c', 'does-not-exist'],
+        check=False
+    )
+    assert process.stdout == b''
+    assert b'does-not-exist' in process.stderr
+    assert process.returncode == 127
+    assert any(rec.message.startswith('CMD:') for rec in caplog.records)
+    assert any(rec.message.startswith('STDOUT:') for rec in caplog.records)
+    assert any(rec.message.startswith('STDERR:') for rec in caplog.records)