a ijv @sddlZddlZddlZddlZddlZddlmZddlmZm Z ddl Z ddl Z e ddZ GdddZe j dd d eeddfd d d Ze j ddddZdS)N)deque) GeneratorSequencec cs|td}|j|j|jdz*|VW|j|j|jdn"|j|j|jd0dS)z0 Fixture providing a memory filesystem. ZmemoryN)fsspec filesystemstoreclearZ pseudo_dirsappend)mr M/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/fsspec/conftest.pyr s        r c@s<eZdZdZddddZddeeeefdd d Z dS) InstanceCacheInspectorzQ Helper class to inspect instance caches of filesystem classes in tests. NreturncCs6ttjjg}|r2|}|||qdS)zU Clear instance caches of all currently imported filesystem classes. N)rrspecAbstractFileSystempopleftZclear_instance_cacheextend__subclasses__)selfclassesclsr r r r $s zInstanceCacheInspector.clearT) omit_zero)rrcCsxi}ttjjg}|rt|}t|j}t|jt t frdt|jt rJ|jn|jd}|s\|sd|||<| | q|S)a Gather counts of filesystem instances in the instance caches of all currently imported filesystem classes. Parameters ---------- omit_zero: Whether to omit instance types with no cached instances. r) rrrrrlen_cache isinstanceprotocolrstrrr)rroutrrcountkeyr r r gather_counts.s  z$InstanceCacheInspector.gather_counts) __name__ __module__ __qualname____doc__r booldictrintr"r r r r rs rfunctionT)scopeZautouserccs0t}|z|VW|n |0dS)as Fixture to ensure empty filesystem instance caches before and after a test. Used by default for all tests. Clears caches of all imported filesystem classes. Can be used to write test assertions about instance caches. Usage: def test_something(instance_caches): # Test code here fsspec.open("file://abc") fsspec.open("memory://foo/bar") # Test assertion assert instance_caches.gather_counts() == {"file": 1, "memory": 1} Returns ------- instance_caches: An instance cache inspector for clearing and inspecting caches. N)rr )Zicr r r instance_cachesGs r,)r+ccstdt|}ttj|dd}|ddWdn1sJ0Yt t j ddd|d d d d d g }zFt ddVW||zt|WqtyYq0n4||zt|WntyYn00dS)z6 Fixture providing a writable FTP filesystem. Z pyftpdlibrwbshelloi'Nz-mz-dz-uuserz-Ppassz-w) localhostiIr.r/)pytestZ importorskipropenospathjoinwrite subprocessPopensys executabletimesleep terminatewaitshutilrmtree Exception)ZtmpdirdfPr r r ftp_writablegs, ,   rF)r4r@r8r:r< collectionsrcollections.abcrrr2rZfixturer rr,rFr r r r s  (