fix stuff

scripts/test/smoketest.py

@@ -0,0 +1,89 @@
+import os
+import sys
+import time
+import signal
+import shutil
+import tempfile
+import requests
+import threading
+import subprocess as sp
+
+
+class Cpp(object):
+    def __init__(self, args):
+        self.p = sp.Popen([sys.executable, "-m", "copyparty"] + args)
+        # , stdout=sp.PIPE, stderr=sp.PIPE)
+
+        self.t = threading.Thread(target=self._run)
+        self.t.daemon = True
+        self.t.start()
+
+    def _run(self):
+        self.so, self.se = self.p.communicate()
+
+    def stop(self, wait):
+        # self.p.kill()
+        os.kill(self.p.pid, signal.SIGINT)
+        if wait:
+            self.t.join()
+
+
+def main():
+    ub = "http://127.0.0.1:4321/"
+    td = os.path.join("srv", "smoketest")
+    try:
+        shutil.rmtree(td)
+    except:
+        pass
+
+    os.mkdir(td)
+
+    vidp = os.path.join(tempfile.gettempdir(), "smoketest.h264")
+    if not os.path.exists(vidp):
+        cmd = "ffmpeg -f lavfi -i testsrc=48x32:3 -t 1 -c:v libx264 -tune animation -preset veryslow -crf 69"
+        sp.check_call(cmd.split(" ") + [vidp])
+
+    with open(vidp, "rb") as f:
+        ovid = f.read()
+
+    args = ["-p", "4321"]
+    pdirs = []
+
+    for d1 in ["r", "w", "a"]:
+        pdirs.append("{}/{}".format(td, d1))
+        for d2 in ["r", "w", "a"]:
+            d = os.path.join(td, d1, "j", d2)
+            pdirs.append(d.replace("\\", "/"))
+            os.makedirs(d)
+
+    udirs = [x.split("/", 2)[2] for x in pdirs]
+    for pd, ud in zip(pdirs, udirs):
+        # args += ["-v", "{}:{}:{}".format(d.split("/", 1)[1], d, d[-1])]
+        args += ["-v", "{}:{}:{}".format(pd, ud, ud[-1])]
+
+    cpp = Cpp(args)
+
+    up = False
+    for n in range(30):
+        try:
+            time.sleep(0.1)
+            requests.get(ub + "?h", timeout=0.1)
+            up = True
+            break
+        except:
+            pass
+
+    assert up
+    # for d in dirs:
+    #    rd, fn = d.rsplit("/", 1)
+    #    requests.post(ub + rd, files={"act": "mkdir", "name": fn})
+
+    for d in udirs:
+        vid = ovid + "\n{}".format(d).encode("utf-8")
+        requests.post(ub + d, data={"act": "bput"}, files={"f": ("a.h264", vid)})
+    time.sleep(3)
+    cpp.stop(True)
+
+
+if __name__ == "__main__":
+    main()