o g@s<ddlmZddlZddlmZmZmZmZddl Z ddl Z ddl m Z ddl mZejdejdeefddZejdejdefd d Zejdejdeefd d Zejdejdefd dZejdejdeeeeffddZejdejdeeeffddZgdZdeedeefddZejdejdefddZejdejdefddZejdejdeefddZejdejdeeeeffddZdeeeefdeeeffd d!Zejdejdeeeffd"d#Zdeeeffd$d%Z ejdejdeeeffd&d'Z!d(edefd)d*Z"dS)+N)AnyDictListOptional)TokenTransportHeader)collection_namedrawreturncCsdttdddd}|tjt||d}t}g}|D]}||vr%q||||q|S)N min_sizemax_size)lenvalid_action_spacestlistsrsetaddappend)rlengthnamesseenunseennamerT/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/test/auth/strategies.pymany_unique_names s   rcCs|tjtjtjdddS)N2alphabetr r)rtextstring ascii_lettersdigitsrrrr random_token$sr'cCs|ttjtjdgSN)r sampled_fromr AUTHORIZATIONX_CHROMA_TOKENr&rrrrandom_token_transport_header+sr,cC|tjtjdddSNrr rr"r#r$r&rrrrandom_user_name8r1c Cs|tjtttjtddddddd}g}t}t}|D]}|d|vr-|S|dD]}||vr9nq1q"||d|dD]}||qF|||S)Nr r )idtokensrr4r5)rrfixed_dictionariesr1r'rrr)rusers unseen_users seen_users seen_tokensusertokenrrrrandom_users_with_tokens=s8      r=cCs |t}td|i}||dS)Nr7)r7filename)r=_dump_to_tmpfile)rr7r>rrrtoken_test_conf]s   r@)z system:resetztenant:create_tenanttenant:get_tenantzdb:create_databasedb:get_databasezdb:list_collectionszdb:create_collectionzdb:get_or_create_collectioncollection:get_collectioncollection:delete_collectioncollection:update_collectioncollection:addcollection:deletecollection:getcollection:querycollection:peekcollection:countcollection:updatecollection:upsertauthorized_actionscsfddtDS)Ncsg|]}|vr|qSrr.0actionrNrr |sz(unauthorized_actions..rrRrrRrunauthorized_actions{srUcCr-r.r0r&rrrrandom_role_namer2rVcCs|ttSr()rr)rr&rrr random_actionsrWcsd|tjtdddtfdddDrddvr$d d d tS) Nrr r c3s|]}|vVqdSr(rrOactionsrr s  z2random_allowed_actions_for_role..) rFrGrHrIrJrLrMrKrErDrCrJrHrArB)rsetsrWanyrlistr&rrXrrandom_allowed_actions_for_roles     r^cCsb|tjtttdddd}g}t}|D]}|d|vr"q||d||q|S)Nr4rYrr r r4)rrr6rVr^rrr)rroles unseen_rolesrrolerrr random_roless"    rcr`cCs,i}|D]}||dd|diiq|S)Nr4rY)update)r` roles_mappingrbrrr_transform_roles_for_flushsrfcCsR|t}|t}|D]}|tjdt|dd}||d|d<q ||dS)Nrr) min_value max_valuer4rbr7r`)r=rcrintegersr)rr7r`r; role_indexrrrrandom_users_and_roless   rlcCsddgddgdtdgdS)N__root__)r4r5rbr_rirTrrrr_root_user_and_roles rncCsb|t}t}|d|d|d|dt|dt|dd}|d|d|dS)Nr7r`)r7re)r7r`r>)rlrnextendr?rf)rusers_and_rolesroot_user_and_roler>rrrrbac_test_confs  rrdatacCsNtjdd}t|jd}t||Wd|jS1swY|jS)NF)deletew)tempfileNamedTemporaryFileopenryamldump)rstmpfrrrr?s  r?)#hypothesis.strategies strategiesrrvtypingrrrrryr#chromadbr!chromadb.test.property.strategiesr compositeDrawFnstrrr'r,r1r=r@rrUrVrWr^rcrfrlrnrrr?rrrrsH    ""&