Skip to content

Commit

Permalink
Add tests
Browse files Browse the repository at this point in the history
  • Loading branch information
kukushking committed Apr 20, 2021
1 parent 49c1839 commit 2a07cf1
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 0 deletions.
16 changes: 16 additions & 0 deletions pymysql/tests/test_connection.py
Expand Up @@ -577,6 +577,22 @@ def test_ssl_connect(self):
dummy_ssl_context.load_cert_chain.assert_not_called
dummy_ssl_context.set_ciphers.assert_not_called

dummy_ssl_context = mock.Mock(options=0)
with mock.patch(
"pymysql.connections.Connection.connect"
) as connect, mock.patch(
"pymysql.connections.ssl.create_default_context",
new=mock.Mock(return_value=dummy_ssl_context),
) as create_default_context:
pymysql.connect(
ssl_cadata="cadata",
ssl_verify_cert=True,
ssl_verify_identity=True,
)
create_default_context.assert_called_with(cafile=None, capath=None, cadata="cadata")
assert dummy_ssl_context.verify_mode == ssl.CERT_REQUIRED
assert dummy_ssl_context.check_hostname

dummy_ssl_context = mock.Mock(options=0)
with mock.patch(
"pymysql.connections.Connection.connect"
Expand Down
7 changes: 7 additions & 0 deletions tests/test_auth.py
Expand Up @@ -10,7 +10,9 @@
port = 3306

ca = os.path.expanduser("~/ca.pem")
cadata = open(ca, "r").read()
ssl = {"ca": ca, "check_hostname": False}
ssl_cadata = {"cadata": cadata, "check_hostname": False}

pass_sha256 = "pass_sha256_01234567890123456789"
pass_caching_sha2 = "pass_caching_sha2_01234567890123456789"
Expand All @@ -26,6 +28,11 @@ def test_sha256_no_passowrd_ssl():
con.close()


def test_sha256_no_passowrd_ssl_cadata():
con = pymysql.connect(user="nopass_sha256", host=host, port=port, ssl=ssl_cadata)
con.close()


def test_sha256_password():
con = pymysql.connect(
user="user_sha256", password=pass_sha256, host=host, port=port, ssl=None
Expand Down

0 comments on commit 2a07cf1

Please sign in to comment.