From ec74b1ad57341a4b495e9381e38166c0fe80dae2 Mon Sep 17 00:00:00 2001
From: eeff <33566201+eeff@users.noreply.github.com>
Date: Tue, 24 Oct 2023 10:28:54 +0800
Subject: [PATCH] test: ft add template api

---
 tests/ft/neuron/api.py | 63 +++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 62 insertions(+), 1 deletion(-)

diff --git a/tests/ft/neuron/api.py b/tests/ft/neuron/api.py
index 327d4ffe..ce79df53 100644
--- a/tests/ft/neuron/api.py
+++ b/tests/ft/neuron/api.py
@@ -174,7 +174,7 @@ def get_plugins():
 
 
 @gen_check
-def add_template(name, plugin, groups):
+def add_template(name, plugin, groups=[]):
     return requests.post(url=config.BASE_URL + "/api/v2/template", json={"name": name, "plugin": plugin, "groups": groups}, headers={"Authorization": config.default_jwt})
 
 
@@ -183,6 +183,67 @@ def del_template(name):
     return requests.delete(url=config.BASE_URL + "/api/v2/template" + f"?name={name}", headers={"Authorization": config.default_jwt})
 
 
+@gen_check
+def get_template(name):
+    return requests.get(url=config.BASE_URL + "/api/v2/template", headers={"Authorization": config.default_jwt}, params={"name": name})
+
+
+@gen_check
+def get_templates():
+    return requests.get(url=config.BASE_URL + "/api/v2/template", headers={"Authorization": config.default_jwt})
+
+
+@gen_check
+def inst_template(name, node):
+    return requests.post(url=config.BASE_URL + "/api/v2/template/inst", json={"name": name, "node": node}, headers={"Authorization": config.default_jwt})
+
+
+@gen_check
+def add_template_group(tmpl, group, interval=100):
+    return requests.post(url=config.BASE_URL + '/api/v2/template/group', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "interval": interval})
+
+
+@gen_check
+def update_template_group(tmpl, group, new_name="", interval=0):
+    if len(new_name) > 0 and interval > 0:
+        return requests.put(url=config.BASE_URL + '/api/v2/template/group', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "new_name": new_name, "interval": interval})
+    elif len(new_name) > 0:
+        return requests.put(url=config.BASE_URL + '/api/v2/template/group', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "new_name": new_name})
+    elif interval > 0:
+        return requests.put(url=config.BASE_URL + '/api/v2/template/group', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "interval": interval})
+    else:
+        assert False
+
+
+@gen_check
+def del_template_group(tmpl, group):
+    return requests.delete(url=config.BASE_URL + '/api/v2/template/group', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group})
+
+
+@gen_check
+def get_template_groups(tmpl):
+    return requests.get(url=config.BASE_URL + "/api/v2/template/group", headers={"Authorization": config.default_jwt}, params={"name": tmpl})
+
+
+@gen_check
+def add_template_tags(tmpl, group, tags):
+    return requests.post(url=config.BASE_URL + '/api/v2/template/tag', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "tags": tags})
+
+
+@gen_check
+def update_template_tags(tmpl, group, tags):
+    return requests.put(url=config.BASE_URL + '/api/v2/template/tag', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "tags": tags})
+
+
+@gen_check
+def del_template_tags(tmpl, group, tags):
+    return requests.delete(url=config.BASE_URL + '/api/v2/template/tag', headers={"Authorization": config.default_jwt}, json={"template": tmpl, "group": group, "tags": tags})
+
+
+def get_template_tags(tmpl, group, name):
+    return requests.get(url=config.BASE_URL + "/api/v2/template/tag", headers={"Authorization": config.default_jwt}, params={"template": tmpl, "group": group, "name": name})
+
+
 # plugin setting