58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
|
import os
|
||
|
|
import shlex
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
import fsspec
|
||
|
|
|
||
|
|
pytest.importorskip("notebook")
|
||
|
|
requests = pytest.importorskip("requests")
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture()
|
||
|
|
def jupyter(tmpdir):
|
||
|
|
tmpdir = str(tmpdir)
|
||
|
|
os.environ["JUPYTER_TOKEN"] = "blah"
|
||
|
|
try:
|
||
|
|
cmd = f'jupyter notebook --notebook-dir="{tmpdir}" --no-browser --port=5566'
|
||
|
|
P = subprocess.Popen(shlex.split(cmd))
|
||
|
|
except FileNotFoundError:
|
||
|
|
pytest.skip("notebook not installed correctly")
|
||
|
|
try:
|
||
|
|
timeout = 15
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
r = requests.get("http://localhost:5566/?token=blah")
|
||
|
|
r.raise_for_status()
|
||
|
|
break
|
||
|
|
except (requests.exceptions.BaseHTTPError, OSError):
|
||
|
|
time.sleep(0.1)
|
||
|
|
timeout -= 0.1
|
||
|
|
if timeout < 0:
|
||
|
|
pytest.xfail("Timed out for jupyter")
|
||
|
|
yield "http://localhost:5566/?token=blah", tmpdir
|
||
|
|
finally:
|
||
|
|
P.terminate()
|
||
|
|
|
||
|
|
|
||
|
|
def test_simple(jupyter):
|
||
|
|
url, d = jupyter
|
||
|
|
fs = fsspec.filesystem("jupyter", url=url)
|
||
|
|
assert fs.ls("") == []
|
||
|
|
|
||
|
|
fs.pipe("afile", b"data")
|
||
|
|
assert fs.cat("afile") == b"data"
|
||
|
|
assert "afile" in os.listdir(d)
|
||
|
|
|
||
|
|
with fs.open("bfile", "wb") as f:
|
||
|
|
f.write(b"more")
|
||
|
|
with fs.open("bfile", "rb") as f:
|
||
|
|
assert f.read() == b"more"
|
||
|
|
|
||
|
|
assert fs.info("bfile")["size"] == 4
|
||
|
|
fs.rm("afile")
|
||
|
|
|
||
|
|
assert "afile" not in os.listdir(d)
|