B
øoh\™ ã @ 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 ú1lib/python3.7/site-packages/keyring/tests/util.pyÚ__init__
s zImportKiller.__init__Nc C s || j kr| S d S )N)r )r ÚfullnameÚpathr r r Ú find_module s
zImportKiller.find_modulec C s || j kst‚t|ƒ‚d S )N)r ÚAssertionErrorÚ ImportError)r r r r r Ú load_module s zImportKiller.load_modulec C s< i | _ x"| jD ]}tj |d ¡| j |< qW tj d| ¡ d S )Nr )Úoriginalr ÚsysÚmodulesÚpopÚ meta_pathÚinsert)r Únamer r r Ú __enter__ s zImportKiller.__enter__c G s: t j | ¡ x(| j ¡ D ]\}}|d k r|t j|< qW d S )N)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 }xB| ¡ D ]6\}}| |¡||< |dkr<|| krD| |= q|| |< qW dV x4| ¡ D ](\}}|dkrx|| kr€| |= qX|| |< qXW dS )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}x t d| ƒD ]}|t |¡7 }qW |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