a ij?@sdZddlmZddlZddlZddlZddlZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZdd lmZed efZejZejd d Gd ddeZejdddddddddddddd dddddddddddddddddZdS)zTest utils for epath.) annotationsN)AnyCallableIteratorOptionalUnion)mock)backend)gpath) stat_utils)PathLike.F)eqcseZdZUdZded<fddZdddd d d Zdd d ddZdd d ddZddd ddZ ddd ddZ dddd ddddZ dddd ddddZ ddd dd Z ddd d!d"Zdddd#d$d%Zdddd#d&d'Zddd dd(d)d*Zdd+d d,d-ZZS). _MockBackendz#Backend with functions overwritten.zdict[str, _MockFn]mock_fnscs:tt|}|j|dur2|j|}t||}n|}|S)N)getattrsuperr functoolspartial)selffn_nameZ original_fnZ fn_to_call __class__Q/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/epath/testing.py_get_fn+s   z_MockBackend._get_fnr strztyping.IO[Union[str, bytes]])pathmodereturncCs|d||S)Nopenr)rrrrrrr4sz_MockBackend.openbool)rrcCs|d|S)Nexistsr rrrrrr";sz_MockBackend.existscCs|d|S)Nisdirr r#rrrr$>sz_MockBackend.isdirz list[str]cCs|d|S)Nlistdirr r#rrrr%Asz_MockBackend.listdircCs|d|S)Nglobr r#rrrr&Dsz_MockBackend.globFexist_okNone)rr(rcCs|d||dS)Nmakedirsr'r rrr(rrrr*Gsz_MockBackend.makedirscCs|d||dS)Nmkdirr'r r+rrrr,Jsz_MockBackend.mkdircCs|d|S)Nrmtreer r#rrrr-Msz_MockBackend.rmtreecCs|d|S)Nremover r#rrrr.Psz_MockBackend.remove)rdstrcCs|d||S)Nrenamer rrr/rrrr0Ssz_MockBackend.renamecCs|d||S)Nreplacer r1rrrr2Vsz_MockBackend.replace)rr/ overwritercCs|d|||dS)Ncopy)r3r )rrr/r3rrrr4Ysz_MockBackend.copyzstat_utils.StatResultcCs|d|S)Nstatr r#rrrr5\sz_MockBackend.stat)__name__ __module__ __qualname____doc____annotations__rrr"r$r%r&r*r,r-r.r0r2r4r5 __classcell__rrrrr%s  r) r4r"r&r$r%r*r,rr.r0r2r-r5zOptional[_MockFn]zIterator[None])r4r"r&r$r%r*r,rr.r0r2r-r5rc #szt||| |||||||| | | d } t| dfddtjD}tjtd|dVWdn1sl0YdS)aMock epath. Mock the file system by replacing the given function by their mock. Only the function passed are mocked. The mock function should have signature: `(original_fn, path)` + eventual args/kwargs for specific functions. Args: copy: New function (after mocking) exists: New function (after mocking) glob: New function (after mocking) isdir: New function (after mocking) listdir: New function (after mocking) makedirs: New function (after mocking) mkdir: New function (after mocking) open: New function (after mocking) remove: New function (after mocking) rename: New function (after mocking) replace: New function (after mocking) rmtree: New function (after mocking) stat: New function (after mocking) Yields: None ) rr4r0r"r&r$r%r*r,r.r2r-r5)rcsi|] }|qSrr).0kZ mock_backendrr zmock_epath.._PREFIX_TO_BACKENDN)dictrr rArpatchobject)r4r"r&r$r%r*r,rr.r0r2r-r5rZnew_prefix_to_backendrr>r mock_epath`s&* rE)r9 __future__r contextlib dataclassesrtypingrrrrrZunittestrZ etils.epathr r r Zetils.epath.typingr Z_MockFnZ_OsPathBackendZ _backend_cls dataclassrcontextmanagerrErrrrs<        :