refactor: python normal test

Author: kehiy

.gitignore

@@ -31,3 +31,6 @@ cmd/config.yaml
 /client
 log.ttrace
 build/
+
+# python cache files
+test/__pycache__

test/normal_test.py

@@ -39,7 +39,7 @@ def test_ping_ok():
 def test_new_set_ok():
     #* Creating some random sets.
     for i in range(10):
-        set_names.append(utils.get_random_string(i+2))
+        set_names.append(utils.get_random_string_name(i+2))
 
         query = f"SET {set_names[i]}"
         response = utils.make_query(query, sock)
@@ -52,7 +52,7 @@ def test_new_sub_set_ok():
     #* Creating some random sub sets.
     for s in set_names:
         for i in range(7):
-            sub_set_names.append(utils.get_random_string(i+2))
+            sub_set_names.append(utils.get_random_string_name(i+2))
 
             query = f"SSET {s} {sub_set_names[i]}"
             response = utils.make_query(query, sock)
@@ -67,7 +67,7 @@ def test_push_element_ok():
     for s in set_names:
         for i in range(7):
             for _ in range(1_000):
-                element_value = utils.get_random_string(i+8)
+                element_value = utils.get_random_string_name(i+8)
                 elements_value.append(element_value)
 
                 query = f"PUSH {s} {sub_set_names[i]} {element_value} {int(time.mktime(time.gmtime()))}"

test/utils.py

@@ -1,9 +1,10 @@
 import random
 import string
+import socket
 
-def get_random_string(length):
+def get_random_string_name(length: int) -> str:
     return ''.join(random.choice(string.ascii_lowercase) for i in range(length))
 
-def make_query(query, sock):
+def make_query(query: str, sock: socket.socket) -> str:
     sock.sendall(query.encode())
     return sock.recv(1024).decode()