tests added

.vscode/settings.json

@@ -1,4 +1,9 @@
 {
     "python.terminal.activateEnvironment": true,
-    "python.defaultInterpreterPath": "${env:PYTHON_INSTALL_LOC}"
+    "python.defaultInterpreterPath": "${env:PYTHON_INSTALL_LOC}",
+    "python.testing.pytestArgs": [
+        "."
+    ],
+    "python.testing.unittestEnabled": false,
+    "python.testing.pytestEnabled": true
 }

conftest.py

@@ -0,0 +1,7 @@
+import pytest
+import os
+
+
+@pytest.fixture(scope="function")
+def gremlin_url() -> str:
+    return os.environ.get("GREMLIN_SERVER_URL", "ws://megamind.local:8182/gremlin")

tests/test_execute_query.py

@@ -1,11 +1,11 @@
 from invana import InvanaGraph
 
 
-def test_execute_query():
-    graph = InvanaGraph("ws://localhost:8182/gremlin")
-    # response = graph.connector.execute_query("g.addV('person').property('name', 'Rav').next()")
-    response = graph.connector.execute_query("g.V().limit(1).elementMap().toList()")
-    print("===================",response.data)
-    assert response.status_code == 200
-    assert response.data is not None
-    graph.close_connection()
+class TestConnection:
+
+    def test_execute_query(self, gremlin_url: str):
+        graph = InvanaGraph(gremlin_url)
+        response = graph.connector.execute_query("g.V().limit(1).elementMap().toList()")
+        assert response.status_code == 200
+        assert response.data is not None
+        graph.close_connection()