diff --git a/tests/unit/plugins/module_utils/test_mariadb_replication.py b/tests/unit/plugins/module_utils/test_mariadb_replication.py index 7e8940f..7442e30 100644 --- a/tests/unit/plugins/module_utils/test_mariadb_replication.py +++ b/tests/unit/plugins/module_utils/test_mariadb_replication.py @@ -50,20 +50,20 @@ def test_uses_replica_terminology(f_output, c_output, c_ret_type): def test_start_group_replication(user, password, expected_query): """Test startgroupreplication function with different parameters.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import startgroupreplication - + cursor = MockCursor() module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + chm = [] if user: chm.append(" USER='%s'" % user) if password: chm.append(" PASSWORD='%s'" % password) - + result = startgroupreplication(module, cursor, chm, False) - + assert result is True assert cursor.executed_queries[0] == expected_query assert cursor.executed_queries[1] == "SHOW STATUS LIKE 'group_replication_status';" @@ -72,14 +72,14 @@ def test_start_group_replication(user, password, expected_query): def test_stop_group_replication(): """Test stopgroupreplication function.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import stopgroupreplication - + cursor = MockCursor() module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + result = stopgroupreplication(module, cursor, False) - + assert result is True assert cursor.executed_queries[0] == "STOP GROUP_REPLICATION" assert cursor.executed_queries[1] == "SHOW STATUS LIKE 'group_replication_status';" @@ -89,17 +89,17 @@ def test_start_group_replication_fail(): """Test startgroupreplication function with failure.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import startgroupreplication import pymysql - + cursor = MockCursor(status="ERROR") module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + # Mock the Warning exception pymysql.Warning = Exception - + result = startgroupreplication(module, cursor, [], True) - + assert result is False @@ -107,15 +107,15 @@ def test_stop_group_replication_fail(): """Test stopgroupreplication function with failure.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import stopgroupreplication import pymysql - + cursor = MockCursor(status="ERROR") module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + # Mock the Warning exception pymysql.Warning = Exception - + result = stopgroupreplication(module, cursor, True) - + assert result is False diff --git a/tests/unit/plugins/module_utils/test_mysql_replication.py b/tests/unit/plugins/module_utils/test_mysql_replication.py index cd4a9d7..19ba12b 100644 --- a/tests/unit/plugins/module_utils/test_mysql_replication.py +++ b/tests/unit/plugins/module_utils/test_mysql_replication.py @@ -54,20 +54,20 @@ def test_uses_replica_terminology(f_output, c_output, c_ret_type): def test_start_group_replication(user, password, expected_query): """Test startgroupreplication function with different parameters.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import startgroupreplication - + cursor = MockCursor() module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + chm = [] if user: chm.append(" USER='%s'" % user) if password: chm.append(" PASSWORD='%s'" % password) - + result = startgroupreplication(module, cursor, chm, False) - + assert result is True assert cursor.executed_queries[0] == expected_query assert cursor.executed_queries[1] == "SHOW STATUS LIKE 'group_replication_status';" @@ -76,14 +76,14 @@ def test_start_group_replication(user, password, expected_query): def test_stop_group_replication(): """Test stopgroupreplication function.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import stopgroupreplication - + cursor = MockCursor() module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + result = stopgroupreplication(module, cursor, False) - + assert result is True assert cursor.executed_queries[0] == "STOP GROUP_REPLICATION" assert cursor.executed_queries[1] == "SHOW STATUS LIKE 'group_replication_status';" @@ -93,17 +93,17 @@ def test_start_group_replication_fail(): """Test startgroupreplication function with failure.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import startgroupreplication import pymysql - + cursor = MockCursor(status="ERROR") module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + # Mock the Warning exception pymysql.Warning = Exception - + result = startgroupreplication(module, cursor, [], True) - + assert result is False @@ -111,15 +111,15 @@ def test_stop_group_replication_fail(): """Test stopgroupreplication function with failure.""" from ansible_collections.community.mysql.plugins.modules.mysql_replication import stopgroupreplication import pymysql - + cursor = MockCursor(status="ERROR") module = type('obj', (object,), { 'fail_json': lambda msg: None, }) - + # Mock the Warning exception pymysql.Warning = Exception - + result = stopgroupreplication(module, cursor, True) - + assert result is False