35 lines
832 B
Python
35 lines
832 B
Python
import re
|
|
|
|
class MockPlPy:
|
|
def __init__(self):
|
|
self._reset()
|
|
|
|
def _reset(self):
|
|
self.infos = []
|
|
self.notices = []
|
|
self.debugs = []
|
|
self.logs = []
|
|
self.warnings = []
|
|
self.errors = []
|
|
self.fatals = []
|
|
self.executes = []
|
|
self.results = []
|
|
self.prepares = []
|
|
self.results = []
|
|
|
|
def _define_result(self, query, result):
|
|
pattern = re.compile(query, re.IGNORECASE | re.MULTILINE)
|
|
self.results.append([pattern, result])
|
|
|
|
def notice(self, msg):
|
|
self.notices.append(msg)
|
|
|
|
def info(self, msg):
|
|
self.infos.append(msg)
|
|
|
|
def execute(self, query): # TODO: additional arguments
|
|
for result in self.results:
|
|
if result[0].match(query):
|
|
return result[1]
|
|
return []
|