o
a~ @ sh d dl Z d dlZd dlZd dlZd dlZG dd dZe jdd Zdd Zej ej
Z e fdd Z dS )
Nc @ s: e Zd ZdZdd Zd
ddZdd Zd d
Zd d ZdS ) ImportKillerz@Context manager to make an import of a given name or names fail.c G s
|| _ d S Nnames)selfr r 6/usr/lib/python3/dist-packages/keyring/testing/util.py__init__ s
zImportKiller.__init__Nc C s || j v r| S d S r r )r fullnamepathr r r find_module s
zImportKiller.find_modulec C s || j v sJ t|r )r ImportError)r r
r r r load_module s zImportKiller.load_modulec C s8 i | _ | jD ] }tj|d | j |< qtjd| d S )Nr )originalr sysmodulespop meta_pathinsert)r namer r r __enter__ s
zImportKiller.__enter__c G s6 t j| | j D ]
\}}|d ur|t j|< q d S r )r r remover itemsr )r argskeyvaluer r r __exit__ s
zImportKiller.__exit__r ) __name__
__module__ __qualname____doc__r r r r r r r r r r s
r c k s i }| D ]\}}| |||< |du r || v r | |= q|| |< qdV | D ]\}}|du r:|| v r9| |= q*|| |< q*dS )zHelper 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) destinationchangesr r r r r r NoNoneDictMutator# s"
r$ c K s t tjfi | S )z6A context manager to temporarily change the os.environ)r$ osenviron)r# r r r Environ: s r' c C s& d}t d| D ] }|t|7 }q|S )z-Generate a random string with length k r )rangerandomchoice)ksourceresultir r r
random_stringB s r0 )
contextlibr% r r* stringr contextmanagerr$ r'
ascii_lettersdigitsALPHABETr0 r r r r s