STAR-431: Add option to prevent any file-I/O from cqlsh

Co-authored-by: Robert Stupp <[email protected]>
(cherry picked from commit 33e486a)
(cherry picked from commit 9fbd1b3)
(cherry picked from commit c479ab3)
(cherry picked from commit 6c74210)

Author: 2 people

cqlsh_tests/test_cqlsh.py

@@ -24,6 +24,7 @@
 from cassandra.concurrent import execute_concurrent_with_args
 from cassandra.query import BatchStatement, BatchType
 from ccmlib import common
+from ccmlib.node import ToolError
 
 from .cqlsh_tools import monkeypatch_driver, unmonkeypatch_driver
 from dtest import Tester, create_ks, create_cf
@@ -2589,6 +2590,33 @@ def test_cjk_output(self):
 """
         assert stdout_lines_sorted.find(expected) >= 0
 
+    @since('4.0')
+    def test_no_file_io(self):
+        def run_cqlsh_catch_toolerror(cmd, env):
+            """
+            run_cqlsh will throw ToolError if cqlsh exits with a non-zero exit code.
+            """
+            try:
+                out, err, _ = self.node1.run_cqlsh(cmd, env)
+            except ToolError as e:
+                return e.stdout, e.stderr
+
+        cqlsh_stdout, cqlsh_stderr, = run_cqlsh_catch_toolerror('COPY foo.bar FROM \'blah\';', ['--no-file-io'])
+        assert cqlsh_stdout == ''
+        assert 'No file I/O permitted' in cqlsh_stderr
+
+        cqlsh_stdout, cqlsh_stderr = run_cqlsh_catch_toolerror('DEBUG', ['--no-file-io'])
+        assert cqlsh_stdout == ''
+        assert 'No file I/O permitted' in cqlsh_stderr
+
+        cqlsh_stdout, cqlsh_stderr = run_cqlsh_catch_toolerror('CAPTURE \'nah\'', ['--no-file-io'])
+        assert cqlsh_stdout == ''
+        assert 'No file I/O permitted' in cqlsh_stderr
+
+        cqlsh_stdout, cqlsh_stderr = run_cqlsh_catch_toolerror('SOURCE \'nah\'', ['--no-file-io'])
+        assert cqlsh_stdout == ''
+        assert 'No file I/O permitted' in cqlsh_stderr
+
 
 class TestCqlLogin(Tester, CqlshMixin):
     """