summaryrefslogtreecommitdiff
path: root/tests/dbshell/test_mysql.py
diff options
context:
space:
mode:
authordjango-bot <ops@djangoproject.com>2022-02-03 20:24:19 +0100
committerMariusz Felisiak <felisiak.mariusz@gmail.com>2022-02-07 20:37:05 +0100
commit9c19aff7c7561e3a82978a272ecdaad40dda5c00 (patch)
treef0506b668a013d0063e5fba3dbf4863b466713ba /tests/dbshell/test_mysql.py
parentf68fa8b45dfac545cfc4111d4e52804c86db68d3 (diff)
Refs #33476 -- Reformatted code with Black.
Diffstat (limited to 'tests/dbshell/test_mysql.py')
-rw-r--r--tests/dbshell/test_mysql.py254
1 files changed, 133 insertions, 121 deletions
diff --git a/tests/dbshell/test_mysql.py b/tests/dbshell/test_mysql.py
index b1e8e8a5c9..28410b05e1 100644
--- a/tests/dbshell/test_mysql.py
+++ b/tests/dbshell/test_mysql.py
@@ -19,153 +19,165 @@ class MySqlDbshellCommandTestCase(SimpleTestCase):
def test_basic_params_specified_in_settings(self):
expected_args = [
- 'mysql',
- '--user=someuser',
- '--host=somehost',
- '--port=444',
- 'somedbname',
+ "mysql",
+ "--user=someuser",
+ "--host=somehost",
+ "--port=444",
+ "somedbname",
]
- expected_env = {'MYSQL_PWD': 'somepassword'}
+ expected_env = {"MYSQL_PWD": "somepassword"}
self.assertEqual(
- self.settings_to_cmd_args_env({
- 'NAME': 'somedbname',
- 'USER': 'someuser',
- 'PASSWORD': 'somepassword',
- 'HOST': 'somehost',
- 'PORT': 444,
- 'OPTIONS': {},
- }),
+ self.settings_to_cmd_args_env(
+ {
+ "NAME": "somedbname",
+ "USER": "someuser",
+ "PASSWORD": "somepassword",
+ "HOST": "somehost",
+ "PORT": 444,
+ "OPTIONS": {},
+ }
+ ),
(expected_args, expected_env),
)
def test_options_override_settings_proper_values(self):
settings_port = 444
options_port = 555
- self.assertNotEqual(settings_port, options_port, 'test pre-req')
+ self.assertNotEqual(settings_port, options_port, "test pre-req")
expected_args = [
- 'mysql',
- '--user=optionuser',
- '--host=optionhost',
- '--port=%s' % options_port,
- 'optiondbname',
+ "mysql",
+ "--user=optionuser",
+ "--host=optionhost",
+ "--port=%s" % options_port,
+ "optiondbname",
]
- expected_env = {'MYSQL_PWD': 'optionpassword'}
- for keys in [('database', 'password'), ('db', 'passwd')]:
+ expected_env = {"MYSQL_PWD": "optionpassword"}
+ for keys in [("database", "password"), ("db", "passwd")]:
with self.subTest(keys=keys):
database, password = keys
self.assertEqual(
- self.settings_to_cmd_args_env({
- 'NAME': 'settingdbname',
- 'USER': 'settinguser',
- 'PASSWORD': 'settingpassword',
- 'HOST': 'settinghost',
- 'PORT': settings_port,
- 'OPTIONS': {
- database: 'optiondbname',
- 'user': 'optionuser',
- password: 'optionpassword',
- 'host': 'optionhost',
- 'port': options_port,
- },
- }),
+ self.settings_to_cmd_args_env(
+ {
+ "NAME": "settingdbname",
+ "USER": "settinguser",
+ "PASSWORD": "settingpassword",
+ "HOST": "settinghost",
+ "PORT": settings_port,
+ "OPTIONS": {
+ database: "optiondbname",
+ "user": "optionuser",
+ password: "optionpassword",
+ "host": "optionhost",
+ "port": options_port,
+ },
+ }
+ ),
(expected_args, expected_env),
)
def test_options_non_deprecated_keys_preferred(self):
expected_args = [
- 'mysql',
- '--user=someuser',
- '--host=somehost',
- '--port=444',
- 'optiondbname',
+ "mysql",
+ "--user=someuser",
+ "--host=somehost",
+ "--port=444",
+ "optiondbname",
]
- expected_env = {'MYSQL_PWD': 'optionpassword'}
+ expected_env = {"MYSQL_PWD": "optionpassword"}
self.assertEqual(
- self.settings_to_cmd_args_env({
- 'NAME': 'settingdbname',
- 'USER': 'someuser',
- 'PASSWORD': 'settingpassword',
- 'HOST': 'somehost',
- 'PORT': 444,
- 'OPTIONS': {
- 'database': 'optiondbname',
- 'db': 'deprecatedoptiondbname',
- 'password': 'optionpassword',
- 'passwd': 'deprecatedoptionpassword',
- },
- }),
+ self.settings_to_cmd_args_env(
+ {
+ "NAME": "settingdbname",
+ "USER": "someuser",
+ "PASSWORD": "settingpassword",
+ "HOST": "somehost",
+ "PORT": 444,
+ "OPTIONS": {
+ "database": "optiondbname",
+ "db": "deprecatedoptiondbname",
+ "password": "optionpassword",
+ "passwd": "deprecatedoptionpassword",
+ },
+ }
+ ),
(expected_args, expected_env),
)
def test_options_charset(self):
expected_args = [
- 'mysql',
- '--user=someuser',
- '--host=somehost',
- '--port=444',
- '--default-character-set=utf8',
- 'somedbname',
+ "mysql",
+ "--user=someuser",
+ "--host=somehost",
+ "--port=444",
+ "--default-character-set=utf8",
+ "somedbname",
]
- expected_env = {'MYSQL_PWD': 'somepassword'}
+ expected_env = {"MYSQL_PWD": "somepassword"}
self.assertEqual(
- self.settings_to_cmd_args_env({
- 'NAME': 'somedbname',
- 'USER': 'someuser',
- 'PASSWORD': 'somepassword',
- 'HOST': 'somehost',
- 'PORT': 444,
- 'OPTIONS': {'charset': 'utf8'},
- }),
+ self.settings_to_cmd_args_env(
+ {
+ "NAME": "somedbname",
+ "USER": "someuser",
+ "PASSWORD": "somepassword",
+ "HOST": "somehost",
+ "PORT": 444,
+ "OPTIONS": {"charset": "utf8"},
+ }
+ ),
(expected_args, expected_env),
)
def test_can_connect_using_sockets(self):
expected_args = [
- 'mysql',
- '--user=someuser',
- '--socket=/path/to/mysql.socket.file',
- 'somedbname',
+ "mysql",
+ "--user=someuser",
+ "--socket=/path/to/mysql.socket.file",
+ "somedbname",
]
- expected_env = {'MYSQL_PWD': 'somepassword'}
+ expected_env = {"MYSQL_PWD": "somepassword"}
self.assertEqual(
- self.settings_to_cmd_args_env({
- 'NAME': 'somedbname',
- 'USER': 'someuser',
- 'PASSWORD': 'somepassword',
- 'HOST': '/path/to/mysql.socket.file',
- 'PORT': None,
- 'OPTIONS': {},
- }),
+ self.settings_to_cmd_args_env(
+ {
+ "NAME": "somedbname",
+ "USER": "someuser",
+ "PASSWORD": "somepassword",
+ "HOST": "/path/to/mysql.socket.file",
+ "PORT": None,
+ "OPTIONS": {},
+ }
+ ),
(expected_args, expected_env),
)
def test_ssl_certificate_is_added(self):
expected_args = [
- 'mysql',
- '--user=someuser',
- '--host=somehost',
- '--port=444',
- '--ssl-ca=sslca',
- '--ssl-cert=sslcert',
- '--ssl-key=sslkey',
- 'somedbname',
+ "mysql",
+ "--user=someuser",
+ "--host=somehost",
+ "--port=444",
+ "--ssl-ca=sslca",
+ "--ssl-cert=sslcert",
+ "--ssl-key=sslkey",
+ "somedbname",
]
- expected_env = {'MYSQL_PWD': 'somepassword'}
+ expected_env = {"MYSQL_PWD": "somepassword"}
self.assertEqual(
- self.settings_to_cmd_args_env({
- 'NAME': 'somedbname',
- 'USER': 'someuser',
- 'PASSWORD': 'somepassword',
- 'HOST': 'somehost',
- 'PORT': 444,
- 'OPTIONS': {
- 'ssl': {
- 'ca': 'sslca',
- 'cert': 'sslcert',
- 'key': 'sslkey',
+ self.settings_to_cmd_args_env(
+ {
+ "NAME": "somedbname",
+ "USER": "someuser",
+ "PASSWORD": "somepassword",
+ "HOST": "somehost",
+ "PORT": 444,
+ "OPTIONS": {
+ "ssl": {
+ "ca": "sslca",
+ "cert": "sslcert",
+ "key": "sslkey",
+ },
},
- },
- }),
+ }
+ ),
(expected_args, expected_env),
)
@@ -173,16 +185,16 @@ class MySqlDbshellCommandTestCase(SimpleTestCase):
self.assertEqual(
self.settings_to_cmd_args_env(
{
- 'NAME': 'somedbname',
- 'USER': None,
- 'PASSWORD': None,
- 'HOST': None,
- 'PORT': None,
- 'OPTIONS': {},
+ "NAME": "somedbname",
+ "USER": None,
+ "PASSWORD": None,
+ "HOST": None,
+ "PORT": None,
+ "OPTIONS": {},
},
- ['--help'],
+ ["--help"],
),
- (['mysql', 'somedbname', '--help'], None),
+ (["mysql", "somedbname", "--help"], None),
)
def test_crash_password_does_not_leak(self):
@@ -190,19 +202,19 @@ class MySqlDbshellCommandTestCase(SimpleTestCase):
# crash.
args, env = DatabaseClient.settings_to_cmd_args_env(
{
- 'NAME': 'somedbname',
- 'USER': 'someuser',
- 'PASSWORD': 'somepassword',
- 'HOST': 'somehost',
- 'PORT': 444,
- 'OPTIONS': {},
+ "NAME": "somedbname",
+ "USER": "someuser",
+ "PASSWORD": "somepassword",
+ "HOST": "somehost",
+ "PORT": 444,
+ "OPTIONS": {},
},
[],
)
if env:
env = {**os.environ, **env}
- fake_client = Path(__file__).with_name('fake_client.py')
+ fake_client = Path(__file__).with_name("fake_client.py")
args[0:1] = [sys.executable, str(fake_client)]
with self.assertRaises(subprocess.CalledProcessError) as ctx:
subprocess.run(args, check=True, env=env)
- self.assertNotIn('somepassword', str(ctx.exception))
+ self.assertNotIn("somepassword", str(ctx.exception))