1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
"""Tests for function in ....mysqldb module"""
"""module contains test for db_utils"""
from unittest import mock
import pytest
from gn_libs.mysqldb import parse_db_url, database_connection
@pytest.mark.unit_test
@mock.patch("gn_libs.mysqldb.mdb")
@mock.patch("gn_libs.mysqldb.parse_db_url")
def test_database_connection(mock_db_parser, mock_sql):
"""test for creating database connection"""
mock_db_parser.return_value = {
"host": "localhost",
"user": "guest",
"password": "4321",
"database": "users",
"port": 3306
}
mock_sql.Error = Exception
with database_connection("mysql://guest:4321@localhost/users") as _conn:
mock_sql.connect.assert_called_with(
db="users", user="guest", passwd="4321", host="localhost",
port=3306)
@pytest.mark.unit_test
@pytest.mark.parametrize(
"sql_uri,expected",
(("mysql://theuser:passwd@thehost:3306/thedb",
{
"host": "thehost",
"port": 3306,
"user": "theuser",
"password": "passwd",
"database": "thedb"
}),
(("mysql://auser:passwd@somehost:3307/thedb?"
"unix_socket=/run/mysqld/mysqld.sock&connect_timeout=30"),
{
"host": "somehost",
"port": 3307,
"user": "auser",
"password": "passwd",
"database": "thedb",
"unix_socket": "/run/mysqld/mysqld.sock",
"connect_timeout": 30
}),
("mysql://guest:4321@localhost/users",
{
"host": "localhost",
"port": 3306,
"user": "guest",
"password": "4321",
"database": "users"
}),
("mysql://localhost/users",
{
"host": "localhost",
"port": 3306,
"user": None,
"password": None,
"database": "users"
})))
def test_parse_db_url(sql_uri, expected):
"""Test that valid URIs are passed into valid connection dicts"""
assert parse_db_url(sql_uri) == expected
@pytest.mark.unit_test
@pytest.mark.parametrize(
"sql_uri,invalidopt",
(("mysql://localhost/users?socket=/run/mysqld/mysqld.sock", "socket"),
("mysql://localhost/users?connect_timeout=30¬avalidoption=value",
"notavalidoption")))
def test_parse_db_url_with_invalid_options(sql_uri, invalidopt):
"""Test that invalid options cause the function to raise an exception."""
with pytest.raises(AssertionError) as exc_info:
parse_db_url(sql_uri)
assert exc_info.value.args[0] == f"Invalid database connection option ({invalidopt}) provided."
|
