From 33567e966fbc4dc8c2a1f98e1eda829ccedce54f Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Fri, 26 May 2017 11:54:41 -0400
Subject: [PATCH] started some simple alias tests
---
tests/test_aliases.py | 38 ++++++++++++++++++++++++++++++++++++++
1 file changed, 38 insertions(+)
create mode 100644 tests/test_aliases.py
diff --git a/tests/test_aliases.py b/tests/test_aliases.py
new file mode 100644
index 0000000..c31dcdd
--- /dev/null
+++ b/tests/test_aliases.py
@@ -0,0 +1,38 @@
+import json
+import uuid
+
+import pytest
+
+from aiogremlin import driver
+from aiogremlin.gremlin_python.driver import request, serializer
+
+@pytest.mark.asyncio
+async def test_gremlin_query(event_loop, cluster):
+ alias = { 'g': 'g' }
+ cluster = await driver.Cluster.open(event_loop, aliases=alias)
+ client = await cluster.connect()
+ assert client.aliases == alias
+ resp = await client.submit("1 + 1")
+ async for msg in resp:
+ print(msg)
+ await cluster.close()
+
+
+@pytest.mark.asyncio
+async def test_alias_serialization(event_loop):
+ alias = { 'g': 'g' }
+ message = '1 + 1'
+ cluster = await driver.Cluster.open(event_loop, aliases=alias)
+ client = await cluster.connect()
+ # This is the code client/conn uses on submit
+ message = request.RequestMessage(
+ processor='', op='eval',
+ args={'gremlin': message,
+ 'aliases': client._aliases})
+ request_id = str(uuid.uuid4())
+ message = serializer.GraphSONMessageSerializer().serialize_message(
+ request_id, message)
+ message = message.decode('utf-8')[34:]
+ aliases = json.loads(message)['args']['aliases']
+ assert aliases == alias
+ await cluster.close()
--
GitLab