add tck test cases for session

tests/common/utils.py

@@ -5,6 +5,7 @@
 # This source code is licensed under Apache 2.0 License.
 
 import os
+import re
 import random
 import string
 import time
@@ -441,3 +442,11 @@ def get_conn_pool(host: str, port: int):
     if not pool.init([(host, port)], config):
         raise Exception("Fail to init connection pool.")
     return pool
+
+def parse_service_index(name: str):
+    name = name.lower()
+    pattern = r"(graphd|storaged|metad)\[(\d+)\]"
+    m = re.match(pattern, name)
+    if m and len(m.groups()) == 2:
+        return int(m.groups()[1])
+    return None

tests/tck/conftest.py

@@ -30,6 +30,7 @@
     response,
     resp_ok,
     params,
+    parse_service_index,
 )
 from tests.common.nebula_service import NebulaService
 from tests.tck.utils.table import dataset, table
@@ -338,6 +339,21 @@ def given_nebulacluster_with_param(
     class_fixture_variables["cluster"] = nebula_svc
     class_fixture_variables["pool"] = pool
 
+@when(parse('login "{graph}" with "{user}" and "{password}"'))
+def when_login_graphd(graph, user, password, class_fixture_variables, pytestconfig):
+    index = parse_service_index(graph)
+    assert index is not None, "Invalid graph name, name is {}".format(graph)
+    nebula_svc = class_fixture_variables.get("cluster")
+    assert nebula_svc is not None, "Cannot get the cluster"
+    assert index < len(nebula_svc.graphd_processes)
+    graphd_process = nebula_svc.graphd_processes[index]
+    graph_ip, graph_port = graphd_process.host, graphd_process.tcp_port
+    pool = get_conn_pool(graph_ip, graph_port)
+    sess = pool.get_session(user, password)
+    # do not release original session, as we may have cases to test multiple sessions.
+    # connection could be released after cluster stopped.
+    class_fixture_variables["session"] = sess
+    class_fixture_variables["pool"] = pool
 
 @when(parse("executing query:\n{query}"))
 def executing_query(query, graph_spaces, session, request):
@@ -608,6 +624,19 @@ def result_should_contain(request, result, graph_spaces):
     )
 
 
+@then(parse("the result should contain, replace the holders with cluster info:\n{result}"))
+def then_result_should_contain_replace(request, result, graph_spaces, class_fixture_variables):
+    result = replace_result_with_cluster_info(result, class_fixture_variables)
+    cmp_dataset(
+        request,
+        graph_spaces,
+        result,
+        order=False,
+        strict=True,
+        contains=CmpType.CONTAINS,
+    )
+
+
 @then(parse("the result should not contain:\n{result}"))
 def result_should_not_contain(request, result, graph_spaces):
     cmp_dataset(
@@ -804,3 +833,18 @@ def check_client_compatible(graph_spaces):
     assert (
         resp.error_code == ErrorCode.E_CLIENT_SERVER_INCOMPATIBLE
     ), f'The client was not rejected by server: {resp}'
+
+
+def replace_result_with_cluster_info(result, class_fixture_variables):
+    pattern = r"\$\{.*?\}"
+    holders = set(re.findall(pattern, result))
+    cluster = class_fixture_variables.get("cluster")
+    assert cluster is not None, "Cannot get the cluster"
+    for holder in holders:
+        try:
+            eval_string = holder[2:-1]
+            value = eval(eval_string)
+            result = result.replace(holder, str(value))
+        except:
+            raise
+    return result

tests/tck/features/admin/Sessions.feature

@@ -0,0 +1,33 @@
+# Copyright (c) 2021 vesoft inc. All rights reserved.
+#
+# This source code is licensed under Apache 2.0 License.
+Feature: Test sessions
+
+  Background:
+    Given a nebulacluster with 3 graphd and 1 metad and 1 storaged
+
+  Scenario: Show sessions
+    When executing query:
+      """
+      SHOW SESSIONS;
+      """
+    Then the result should contain:
+      | SessionId | UserName | SpaceName | CreateTime | UpdateTime | GraphAddr | Timezone | ClientIp    |
+      | /\d+/     | "root"   | ""        | /.*/       | /.*/       | /.*/      | 0        | "127.0.0.1" |
+    When executing query:
+      """
+      CREATE USER user1 WITH PASSWORD 'nebula1';
+      CREATE SPACE s1(vid_type=int);
+      USE s1;
+      """
+    Then the execution should be successful
+    And wait 3 seconds
+    When login "graphd[1]" with "user1" and "nebula1"
+    And executing query:
+      """
+      SHOW SESSIONS;
+      """
+    Then the result should contain, replace the holders with cluster info:
+      | SessionId | UserName | SpaceName | CreateTime | UpdateTime | GraphAddr                                           | Timezone | ClientIp    |
+      | /\d+/     | "root"   | "s1"      | /.*/       | /.*/       | "127.0.0.1:${cluster.graphd_processes[0].tcp_port}" | 0        | "127.0.0.1" |
+      | /\d+/     | "user1"  | ""        | /.*/       | /.*/       | "127.0.0.1:${cluster.graphd_processes[1].tcp_port}" | 0        | "127.0.0.1" |