feat: add version api & test (#44)

* feat: add version api & test * delete print info

Author: ChenZiHong-Gavin

hugegraph-python-client/src/pyhugegraph/api/version.py

@@ -0,0 +1,40 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from pyhugegraph.api.common import HugeParamsBase
+from pyhugegraph.utils.exceptions import NotFoundError
+from pyhugegraph.utils.huge_requests import HugeSession
+from pyhugegraph.utils.util import check_if_success
+
+
+class VersionManager(HugeParamsBase):
+ def __init__(self, graph_instance):
+ super().__init__(graph_instance)
+ self.__session = HugeSession.new_session()
+
+ def close(self):
+ if self.__session:
+ self.__session.close()
+
+ def version(self):
+ url = f"{self._host}/versions"
+ response = self.__session.get(
+ url, auth=self._auth, headers=self._headers, timeout=self._timeout
+ )
+ if check_if_success(response, NotFoundError(response.content)):
+ return response.json()
+ return {}

hugegraph-python-client/src/pyhugegraph/client.py

@@ -24,6 +24,7 @@
 from pyhugegraph.api.task import TaskManager
 from pyhugegraph.api.traverser import TraverserManager
 from pyhugegraph.api.variable import VariableManager
+from pyhugegraph.api.version import VersionManager
 from pyhugegraph.structure.graph_instance import GraphInstance
 
 
@@ -40,6 +41,7 @@ def __init__(self, ip, port, graph, user, pwd, timeout=10):
  self._task = None
  self._metrics = None
  self._traverser = None
+ self._version = None
 
  def schema(self):
  self._schema = self._schema or SchemaManager(self._graph_instance)
@@ -76,3 +78,7 @@ def metrics(self):
  def traverser(self):
  self._traverser = self._traverser or TraverserManager(self._graph_instance)
  return self._traverser
+
+ def version(self):
+ self._version = self._version or VersionManager(self._graph_instance)
+ return self._version

hugegraph-python-client/src/tests/api/test_version.py

@@ -0,0 +1,48 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+import unittest
+
+from tests.client_utils import ClientUtils
+
+
+class TestVersion(unittest.TestCase):
+ client = None
+ version = None
+
+ @classmethod
+ def setUpClass(cls):
+ cls.client = ClientUtils()
+ cls.version = cls.client.version
+
+ @classmethod
+ def tearDownClass(cls):
+ cls.client.clear_graph_all_data()
+
+ def setUp(self):
+ self.client.clear_graph_all_data()
+
+ def tearDown(self):
+ pass
+
+ def test_version(self):
+ version = self.version.version()
+ self.assertIsInstance(version, dict)
+ self.assertIn("version", version['versions'])
+ self.assertIn("core", version['versions'])
+ self.assertIn("gremlin", version['versions'])
+ self.assertIn("api", version['versions'])

hugegraph-python-client/src/tests/client_utils.py

@@ -41,6 +41,7 @@ def __init__(self):
  self.task = self.client.task()
  self.metrics = self.client.metrics()
  self.traverser = self.client.traverser()
+ self.version = self.client.version()
 
  def init_property_key(self):
  schema = self.schema