class TornadoSessionTest(AsyncTestCase):
def setUp(self):
super(TornadoSessionTest, self).setUp()
self.graph = GraphDatabase("ws://localhost:8182/",
username="******",
password="******")
@gen_test
def test_manual_session(self):
session = yield self.graph.connect(session=str(uuid.uuid4()))
stream = session.send("v = 1+1", processor="session")
resp = yield stream.read()
stream = session.send("v", processor="session")
resp2 = yield stream.read()
self.assertEqual(resp.data[0], resp2.data[0])
session.close()
@gen_test
def test_no_session(self):
session = yield self.graph.connect()
with self.assertRaises(RuntimeError):
stream = session.send("v = 1+1", processor="session")
@gen_test
def test_session_obj_session(self):
session = yield self.graph.session()
stream = session.send("v = 1+1")
resp = yield stream.read()
stream = session.send("v")
resp2 = yield stream.read()
self.assertEqual(resp.data[0], resp2.data[0])
class TornadoSessionTest(AsyncTestCase):
def setUp(self):
super(TornadoSessionTest, self).setUp()
self.graph = GraphDatabase("ws://localhost:8182/",
username="******",
password="******")
@gen_test
def test_manual_session(self):
session = yield self.graph.connect(session=str(uuid.uuid4()))
stream = session.send("v = 1+1", processor="session")
resp = yield stream.read()
stream = session.send("v", processor="session")
resp2 = yield stream.read()
self.assertEqual(resp.data[0], resp2.data[0])
session.close()
@gen_test
def test_no_session(self):
session = yield self.graph.connect()
with self.assertRaises(RuntimeError):
stream = session.send("v = 1+1", processor="session")
@gen_test
def test_session_obj_session(self):
session = yield self.graph.session()
stream = session.send("v = 1+1")
resp = yield stream.read()
stream = session.send("v")
resp2 = yield stream.read()
self.assertEqual(resp.data[0], resp2.data[0])
class TrolliusSessionTest(unittest.TestCase):
def setUp(self):
self.loop = trollius.get_event_loop()
self.graph = GraphDatabase("ws://localhost:8182/",
username="******",
password="******",
future_class=Future)
def test_manual_session(self):
@trollius.coroutine
def go():
session = yield From(self.graph.connect(session=str(uuid.uuid4())))
stream = session.send("v = 1+1", processor="session")
resp = yield From(stream.read())
stream = session.send("v", processor="session")
resp2 = yield From(stream.read())
self.assertEqual(resp.data[0], resp2.data[0])
session.close()
self.loop.run_until_complete(go())
def test_no_session(self):
@trollius.coroutine
def go():
session = yield From(self.graph.connect())
with self.assertRaises(RuntimeError):
stream = session.send("v = 1+1", processor="session")
self.loop.run_until_complete(go())
def test_session_obj_session(self):
@trollius.coroutine
def go():
session = yield From(self.graph.session())
stream = session.send("v = 1+1")
resp = yield From(stream.read())
stream = session.send("v")
resp2 = yield From(stream.read())
self.assertEqual(resp.data[0], resp2.data[0])
self.loop.run_until_complete(go())
class AsyncioSessionTest(unittest.TestCase):
def setUp(self):
self.loop = asyncio.get_event_loop()
self.graph = GraphDatabase("ws://localhost:8182/",
username="******",
password="******",
future_class=Future)
def test_manual_session(self):
@asyncio.coroutine
def go():
session = yield from self.graph.connect(session=str(uuid.uuid4()))
stream = session.send("v = 1+1", processor="session")
resp = yield from stream.read()
stream = session.send("v", processor="session")
resp2 = yield from stream.read()
self.assertEqual(resp.data[0], resp2.data[0])
session.close()
self.loop.run_until_complete(go())
def test_no_session(self):
@asyncio.coroutine
def go():
session = yield from self.graph.connect()
with self.assertRaises(RuntimeError):
stream = session.send("v = 1+1", processor="session")
self.loop.run_until_complete(go())
def test_session_obj_session(self):
@asyncio.coroutine
def go():
session = yield from self.graph.session()
stream = session.send("v = 1+1")
resp = yield from stream.read()
stream = session.send("v")
resp2 = yield from stream.read()
self.assertEqual(resp.data[0], resp2.data[0])
self.loop.run_until_complete(go())
