g <GddZGddeZeZy)c4eZdZdZdZdZdZdZdZdZ y) MockRegistryzRegistry for mocks Registers mock()s, ensures that we only have one mock() per mocked_obj, and iterates over them to unstub each stubbed method. c"t|_yN) IdentityMapmocksselfs J/var/www/openai/venv/lib/python3.12/site-packages/mockito/mock_registry.py__init__zMockRegistry.__init__s  ] c"||j|<yr)rr objmocks r registerzMockRegistry.register!s 3r c:|jj|dSr)rget)r rs r mock_forzMockRegistry.mock_for$szz~~c4((r cz |jj|}|jy#t$rYywxYwr)rpopunstubKeyErrorrs r rzMockRegistry.unstub's8 ::>>#&D KKM   s . ::c|jD]}|j|jjyr)get_registered_mocksrrclear)r rs r unstub_allzMockRegistry.unstub_all/s.--/D KKM0 r c6|jjSr)rvaluesrs r rz!MockRegistry.get_registered_mocks4szz  ""r N) __name__ __module__ __qualname____doc__r rrrrrr r rrs% #) #r rc8eZdZdZdZdZdZd dZdZdZ y) rcg|_yr_storers r r zIdentityMap.__init__:s  r c`|j||jj||fyr)remover'append)r keyvalues r __setitem__zIdentityMap.__setitem__=s$ C C<(r cf|jDcgc]\}}||us ||fc}}|_ycc}}wrr&)r r+kvs r r)zIdentityMap.removeAs-*.++F+$!Q#1v+F Fs --cb|j|}||j||Str)rr)r)r r+rvs r rzIdentityMap.popDs- XXc] > KK I* r Nc>|jD] \}}||us |cS|Srr&)r r+defaultr/r,s r rzIdentityMap.getLs' HAuCx $r cL|jDcgc]\}}| c}}Scc}}wrr&)r r/r0s r rzIdentityMap.valuesRs""kk*kdak***s c"g|jddyrr&rs r rzIdentityMap.clearUs Ar r) rr r!r r-r)rrrrr#r r rr9s&)G +r rN)robjectr mock_registryr#r r r9s&.##D&@ r