mirror of
https://github.com/ansible-collections/community.mysql.git
synced 2025-04-06 10:40:36 -07:00
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import pytest
|
|
|
|
try:
|
|
from unittest.mock import MagicMock
|
|
except ImportError:
|
|
from mock import MagicMock
|
|
|
|
from ansible_collections.community.mysql.plugins.modules.mysql_info import MySQL_Info
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'suffix,cursor_output,server_implementation',
|
|
[
|
|
('mysql', '5.5.1-mysql', 'mysql'),
|
|
('log', '5.7.31-log', 'mysql'),
|
|
('mariadb', '10.5.0-mariadb', 'mariadb'),
|
|
('', '8.0.22', 'mysql'),
|
|
]
|
|
)
|
|
def test_get_info_suffix(suffix, cursor_output, server_implementation):
|
|
def __cursor_return_value(input_parameter):
|
|
if input_parameter == "SHOW GLOBAL VARIABLES":
|
|
cursor.fetchall.return_value = [{"Variable_name": "version", "Value": cursor_output}]
|
|
else:
|
|
cursor.fetchall.return_value = MagicMock()
|
|
|
|
cursor = MagicMock()
|
|
cursor.execute.side_effect = __cursor_return_value
|
|
|
|
info = MySQL_Info(MagicMock(), cursor, server_implementation)
|
|
|
|
assert info.get_info([], [], False)['version']['suffix'] == suffix
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'server_implementation',
|
|
[
|
|
('mysql'),
|
|
('mariadb'),
|
|
]
|
|
)
|
|
def test_is_mariadb(server_implementation):
|
|
cursor = MagicMock()
|
|
|
|
info = MySQL_Info(MagicMock(), cursor, server_implementation)
|
|
|
|
assert info.is_mariadb() == (server_implementation == "mariadb")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'server_implementation',
|
|
[
|
|
('mysql'),
|
|
('mariadb'),
|
|
]
|
|
)
|
|
def test_is_mysql(server_implementation):
|
|
cursor = MagicMock()
|
|
|
|
info = MySQL_Info(MagicMock(), cursor, server_implementation)
|
|
|
|
assert info.is_mysql() == (server_implementation == "mysql")
|