mirror of
https://github.com/ansible-collections/community.mysql.git
synced 2025-08-03 04:34:27 -07:00
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
|
|
class dummy_cursor_class():
|
|
"""Dummy class for returning an answer for SELECT VERSION()."""
|
|
def __init__(self, output, ret_val_type='dict'):
|
|
self.output = output
|
|
self.ret_val_type = ret_val_type
|
|
|
|
def execute(self, query):
|
|
pass
|
|
|
|
def fetchone(self):
|
|
if self.ret_val_type == 'dict':
|
|
return {'version': self.output}
|
|
|
|
elif self.ret_val_type == 'list':
|
|
return [self.output]
|
|
|
|
|
|
class MockCursor:
|
|
Warning = None
|
|
|
|
def __init__(self, status="ONLINE", warning = None):
|
|
self.status = status
|
|
self.executed_queries = []
|
|
self.Warning = warning
|
|
|
|
def execute(self, query):
|
|
self.executed_queries.append(query)
|
|
if self.status == "ERROR":
|
|
# Create a custom exception that mimics mysql_driver.Warning
|
|
class MockWarning(Exception):
|
|
pass
|
|
# Make it available as a class attribute for tests to use
|
|
MockCursor.Warning = MockWarning
|
|
raise MockWarning("Mocked execution error")
|
|
|
|
def fetchone(self):
|
|
if len(self.executed_queries) > 0 and "group_replication_status" in self.executed_queries[-1]:
|
|
return ["group_replication_status", self.status]
|
|
return None
|