1 files changed, 71 insertions, 0 deletions
diff --git a/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/test-tools/component-test/framework/test_utils.py b/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/test-tools/component-test/framework/test_utils.py
new file mode 100644
index 000000000..e3eb645ae
--- /dev/null
+++ b/fluent-bit/lib/wasm-micro-runtime-WAMR-1.2.2/test-tools/component-test/framework/test_utils.py
@@ -0,0 +1,71 @@
+from __future__ import print_function
+#
+# Copyright (C) 2019 Intel Corporation.  All rights reserved.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+#
+
+import datetime
+import os
+import random
+import re
+import shlex
+import subprocess
+import sys
+import time
+import shutil
+from subprocess import check_output, CalledProcessError
+
+def t_getPIDs(process):
+    try:
+        pidlist = map(int, check_output(["pidof", process]).split())
+    except  CalledProcessError:
+        pidlist = []
+    #print process + ':list of PIDs = ' + ', '.join(str(e) for e in pidlist)
+    return pidlist
+
+
+def t_kill_process_by_name(p_keywords):
+    pid_list = []
+    ps_info = subprocess.check_output(shlex.split("ps aux")).split("\n")
+    for p in ps_info:
+        if p_keywords in p:
+            tmp = p.split(" ")
+            tmp = [x for x in tmp if len(x) > 0]
+            pid_list.append(tmp[1])
+
+    for pid in pid_list:
+        cmd = "kill -9 {}".format(pid)
+        subprocess.call(shlex.split(cmd))
+
+    return pid_list
+
+
+
+#proc    -> name of the process
+#kill = 1  -> search for pid for kill
+#kill = 0  -> search for name (default)
+
+def t_process_exists(proc, kill = 0):
+    ret = False
+    processes  = t_getPIDs(proc)
+
+    for pid in processes:
+            if kill == 0:
+                return True
+            else:
+                print("kill [" + proc + "], pid=" + str(pid))
+                os.kill((pid), 9)
+                ret = True
+    return ret
+
+def t_copy_files(source_dir, pattern, dest_dir):
+    files = os.listdir(source_dir)
+    for file in files:
+        if file in ('/', '.', '..'):
+            continue
+
+        if pattern in ('*', '') or files.endswith(pattern):
+            shutil.copy(source_dir+"/"+ file, dest_dir)
+
+
+