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"]