summaryrefslogtreecommitdiff
path: root/tests/dbshell/tests.py
blob: 8dd3d22c98436948c20fcc62a199bdcea13312eb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from unittest import mock

from django.core.management import call_command
from django.core.management.base import CommandError
from django.db import connection
from django.test import SimpleTestCase


class DbshellCommandTestCase(SimpleTestCase):
    def test_command_missing(self):
        msg = (
            "You appear not to have the %r program installed or on your path."
            % connection.client.executable_name
        )
        with self.assertRaisesMessage(CommandError, msg):
            with mock.patch("subprocess.run", side_effect=FileNotFoundError):
                call_command("dbshell")