a
    ij                     @  s^   d Z ddlmZ ddlZddlZddlmZ ddlZejdddd Z	G d	d
 d
ej
ZdS )zUtils for tests.    )annotationsN)mockT)Zautousec                  c  s:   t d} t tjd< | tjd< dV  tjd= tjd= dS )z,colabtools only works in Colab, so mock it..zcolabtools.frontendZ
colabtoolsN)_ColabtoolsFrontEndMockr   Z	MagicMocksysmodules)Zmodule_mock r   U/home/ghrups/robot-bench/.venv/lib/python3.9/site-packages/etils/ecolab/test_utils.pymock_colabtools   s    
r	   c                   @  s   e Zd ZdZddddZdS )r   z*Mock of `from colabtools import frontend`.z	list[str])returnc                 C  s   dgS )NZ	some_ldapr   )selfr   r   r   %GetUsersWhoHaveConnectedToThisSession(   s    z=_ColabtoolsFrontEndMock.GetUsersWhoHaveConnectedToThisSessionN)__name__
__module____qualname____doc__r   r   r   r   r   r   %   s   r   )r   
__future__r   r   typesZunittestr   ZpytestZfixturer	   
ModuleTyper   r   r   r   r   <module>   s   


