o iv @sddlZddlZddlZddlZddlZddlmZddlmZm Z ddl Z ddl Z e ddZ GdddZe j dd d d eeddffd d Ze j ddddZdS)N)deque) GeneratorSequencec cs|td}|j|j|jdz|VW|j|j|jddS|j|j|jdw)z0 Fixture providing a memory filesystem. ZmemoryN)fsspec filesystemstoreclearZ pseudo_dirsappend)mr ^/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/fsspec/conftest.pyr s       r c@s:eZdZdZd ddZdddedeeeffd d Z dS) InstanceCacheInspectorzQ Helper class to inspect instance caches of filesystem classes in tests. returnNcCs<ttjjg}|r|}||||s dSdS)zU Clear instance caches of all currently imported filesystem classes. N)rrspecAbstractFileSystempopleftZclear_instance_cacheextend__subclasses__)selfclassesclsr r r r $s  zInstanceCacheInspector.clearT) omit_zerorcCszi}ttjjg}|r;|}t|j}t|jt t fr2t|jt r%|jn|jd}|s.|s2|||<| | |s |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)rN) __name__ __module__ __qualname____doc__r booldictrintr!r r r r rs $ rfunctionT)scopeZautouserccs0t}|z |VW|dS|w)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}|ddWdn1s&wYt t j ddd|d d d d d g }z$t ddVW||zt|WdSty_YdSw||zt|WwtyxYww)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_writablegs4    rE)r3r?r7r9r; collectionsrcollections.abcrrr1rZfixturer rr+rEr r r r s    (