66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
# fmt: off
|
|
|
|
import pytest
|
|
import subprocess
|
|
import sys
|
|
from typing import List
|
|
from conftest import ShellTest
|
|
from tools.shell.tests.conftest import random_filepath
|
|
|
|
|
|
@pytest.mark.parametrize("command", [".sh ls", ".cd ..", ".log file", ".import file.csv tbl", ".open new_file", ".output out", ".once out", ".excel out", ".read myfile.sql"])
|
|
def test_safe_mode_command(shell, command):
|
|
test = (
|
|
ShellTest(shell, ['-safe'])
|
|
.statement(command)
|
|
)
|
|
result = test.run()
|
|
result.check_stderr('cannot be used in -safe mode')
|
|
|
|
|
|
@pytest.mark.parametrize("param", [(".sh ls", 'cannot be used in -safe mode'), ("INSTALL extension", "Permission Error")])
|
|
def test_safe_mode_dot_command(shell, param):
|
|
command = param[0]
|
|
expected_error = param[1]
|
|
test = (
|
|
ShellTest(shell)
|
|
.statement('.safe_mode')
|
|
.statement(command)
|
|
)
|
|
result = test.run()
|
|
result.check_stderr(expected_error)
|
|
|
|
def test_safe_mode_database_basic(shell, random_filepath):
|
|
test = (
|
|
ShellTest(shell, [random_filepath, '-safe'])
|
|
.statement('CREATE TABLE integers(i INT)')
|
|
.statement('INSERT INTO integers VALUES (1), (2), (3)')
|
|
.statement('SELECT SUM(i) FROM integers')
|
|
)
|
|
result = test.run()
|
|
result.check_stdout("6")
|
|
|
|
@pytest.mark.parametrize("command", [".sh ls", ".cd ..", ".log file", ".import file.csv tbl", ".open new_file", ".output out", ".once out", ".excel out", ".read myfile.sql"])
|
|
@pytest.mark.parametrize("persistent", [False, True])
|
|
def test_safe_mode_database_commands(shell, random_filepath, command, persistent):
|
|
arguments = ['-safe'] if not persistent else [random_filepath, '-safe']
|
|
test = (
|
|
ShellTest(shell, arguments)
|
|
.statement(command)
|
|
)
|
|
result = test.run()
|
|
result.check_stderr('cannot be used in -safe mode')
|
|
|
|
@pytest.mark.parametrize("sql", ["COPY (SELECT 42) TO 'test.csv'", "LOAD spatial", "INSTALL spatial", "ATTACH 'file.db' AS file"])
|
|
@pytest.mark.parametrize("persistent", [False, True])
|
|
def test_safe_mode_query(shell, random_filepath, sql, persistent):
|
|
arguments = ['-safe'] if not persistent else [random_filepath, '-safe']
|
|
test = (
|
|
ShellTest(shell, arguments)
|
|
.statement(sql)
|
|
)
|
|
result = test.run()
|
|
result.check_stderr('disabled')
|
|
|
|
# fmt: on
|