def test_transmit(self):
s = Storage()
s.storage = MagicMock()
s.storage.search = MagicMock()
search = "foo"
s.search(search)
s.storage.search.assert_called_once_with(search)
def test_transmit(self):
s = Storage()
s.storage = MagicMock()
s.storage.search = MagicMock()
search = "foo"
s.search(search)
s.storage.search.assert_called_once_with(search)
class RequestHandler:
__instance = None
# poor man's singleton
def instance():
if RequestHandler.__instance is None:
RequestHandler.__instance = RequestHandler()
return RequestHandler.__instance
def __init__(self):
self.storage = Storage()
# respond to a request
def handle(self, request):
try:
# search_channel requests
if request['request'] == "twitchcancer.search":
# we need a channel to search for
if request['data'] is not None:
return self.storage.search(request['data'])
except KeyError:
logger.warning("got a malformed request %s", request)
except TypeError:
logger.warning("got a malformed request %s", request)
return {}
class RequestHandler:
__instance = None
# poor man's singleton
def instance():
if RequestHandler.__instance is None:
RequestHandler.__instance = RequestHandler()
return RequestHandler.__instance
def __init__(self):
self.storage = Storage()
# respond to a request
def handle(self, request):
try:
# search_channel requests
if request['request'] == "twitchcancer.search":
# we need a channel to search for
if request['data'] is not None:
return self.storage.search(request['data'])
except KeyError:
logger.warning("got a malformed request %s", request)
except TypeError:
logger.warning("got a malformed request %s", request)
return {}
