U
ܦ—\™ ã @ sl d dl Z d dlZd dlZd dlZd dlZeZG dd„ dƒZe jdd„ ƒZ dd„ Z
ej ej Z
e
fdd „ZdS )
é Nc @ s: e Zd ZdZdd„ Zd
dd„Zdd„ Zd d
„ Zd d „ ZdS )Ú ImportKillerz@Context manager to make an import of a given name or names fail.c G s
|| _ d S ©N©Únames)Úselfr © r ú4/usr/lib/python3/dist-packages/keyring/tests/util.pyÚ__init__
s zImportKiller.__init__Nc C s || j kr| S d S r r )r ÚfullnameÚpathr r r Ú find_module s
zImportKiller.find_modulec C s || j kst‚t|ƒ‚d S r )r ÚAssertionErrorÚ ImportError)r r
r r r Ú load_module s zImportKiller.load_modulec C s8 i | _ | jD ]}tj |d ¡| j |< q tj d| ¡ d S )Nr )Úoriginalr ÚsysÚmodulesÚpopÚ meta_pathÚinsert)r Únamer r r Ú __enter__ s
zImportKiller.__enter__c G s6 t j | ¡ | j ¡ D ]\}}|d k r|t j|< qd S r )r r Úremover Úitemsr )r ÚargsÚkeyÚvaluer r r Ú__exit__ s zImportKiller.__exit__)N) Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r r r r r r r
s
r c k s€ i }| ¡ D ]6\}}| |¡||< |dkr:|| krB| |= q || |< q dV | ¡ D ](\}}|dkrr|| krz| |= qR|| |< qRdS )z¯Helper context manager to make and unmake changes to a dict.
A None is not a valid value for the destination, and so means that the
associated name should be removed.N)r Úget)Z destinationÚchangesr r r r r r ÚNoNoneDictMutator% s
r$ c K s t tjf| ŽS )z6A context manager to temporarily change the os.environ)r$ ÚosÚenviron)r# r r r ÚEnviron< s r' c C s& d}t d| ƒD ]}|t |¡7 }q|S )z2Generate a random string with length k
Ú r )ÚrangeÚrandomZchoice)ÚkÚsourceÚresultÚir r r Ú
random_stringD s r/ )Ú
contextlibr% r r* ÚstringÚtypeZ
__metaclass__r Úcontextmanagerr$ r' Z
ascii_lettersZdigitsZALPHABETr/ r r r r Ú s