giv &ddlZddlZddlZddlZddlZddlmZddlmZm Z ddl Z ddl Z e j dZ GddZe j dd d eeddffd Ze j d d ZdS)N)deque) GeneratorSequencec#Ktjd}|j|j|jd |V|j|j|jddS#|j|j|jdwxYw)z0 Fixture providing a memory filesystem. memoryN)fsspec filesystemstoreclear pseudo_dirsappend)ms g/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/fsspec/conftest.pyrrs (##AGMMOOOMM!    R         R    s $B66ADc@eZdZdZd dZdddedeeeffdZ dS) InstanceCacheInspectorzQ Helper class to inspect instance caches of filesystem classes in tests. returnNcttjjg}|rS|}||||QdSdS)zU Clear instance caches of all currently imported filesystem classes. N)rr specAbstractFileSystempopleftclear_instance_cacheextend__subclasses__)selfclassesclss rr zInstanceCacheInspector.clear$s~7899 1//##C  $ $ & & & NN3--// 0 0 0 1 1 1 1 1T) omit_zerorci}ttjjg}|r|}t |j}t|jttfr7t|jtr|jn |jd}|s|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) rr rrrlen_cache isinstanceprotocolrstrrr)rroutrrcountkeys r gather_countsz$InstanceCacheInspector.gather_counts.s!7899 1//##C OOE#,388 %&0s&C&CXcllVW% %$CH NN3--// 0 0 0 1 r)rN) __name__ __module__ __qualname____doc__r booldictr%intr)rrrrsk111126$$sCx.rrfunctionT)scopeautouserc#Kt}| |V|dS#|wxYw)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 )ics rinstance_cachesr7GsQ. ! !BHHJJJ   s AA)r3c #Ktjdt|}ttj|dd5}|ddzdddn #1swxYwYtj tj ddd|d d d d d g } tj ddV|| t!j|dS#t$$rYdSwxYw#|| t!j|w#t$$rYwwxYwxYw)z6 Fixture providing a writable FTP filesystem. pyftpdlibr&wbshelloi'Nz-mz-dz-uuserz-Ppassz-w) localhostiIr;r<)pytest importorskipr%openospathjoinwrite subprocessPopensys executabletimesleep terminatewaitshutilrmtree Exception)tmpdirdfPs r ftp_writablerUgs   $$$ F A bgll1e$$d + +"q 5 !!!""""""""""""""" {D!T64QUV  A  1 ////    M& ! ! ! ! !    DD     M& ! ! ! !    D sTA99A=A=-D.D DD)E$?EE$ E!E$ E!!E$)rBrNrFrHrJ collectionsrcollections.abcrrr?r fixturerrr7rUr1rrrYs;  ////////  ! ! ! %%%%%%%%Pj$///#94#EF0/>j!!!"!r