def run(self):
conn = self.connection()
raw_data = conn.recv(1024)
try:
if not raw_data:
raise Reconnect
data = loader(raw_data)
logger.info("Running command: %s" % data)
call(data)
except (InvalidFormat, Forbidden) as error:
logger.error(str(error))
raise Reconnect
finally:
conn.close()
logger.debug('connection closed')
def test_unable_to_parse(self):
with raises(InvalidFormat):
loader.loader("""foo": []}""", allowed=["foo"])
def test_loader_not_allowed(self):
with raises(Forbidden):
loader.loader("""{"foo": []}""")
def test_loader_simplest_obj(self):
result = loader.loader("""{"foo": []}""", allowed=["foo"])
assert result == ["foo"]
