From e268e685107c46e733887c412b149a16881d27b4 Mon Sep 17 00:00:00 2001
From: Ravi Raja Merugu <rrmerugu@gmail.com>
Date: Fri, 6 Jan 2023 02:07:45 +0530
Subject: [PATCH] tests added

---
 .vscode/settings.json       |  7 ++++++-
 conftest.py                 |  7 +++++++
 tests/test_execute_query.py | 16 ++++++++--------
 3 files changed, 21 insertions(+), 9 deletions(-)
 create mode 100644 conftest.py

diff --git a/.vscode/settings.json b/.vscode/settings.json
index 57f8110..dcb88a4 100644
--- a/.vscode/settings.json
+++ b/.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
 }
\ No newline at end of file
diff --git a/conftest.py b/conftest.py
new file mode 100644
index 0000000..159ccb1
--- /dev/null
+++ b/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")
diff --git a/tests/test_execute_query.py b/tests/test_execute_query.py
index a1e57c1..40e99ca 100644
--- a/tests/test_execute_query.py
+++ b/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()