a )&i@sddlZddlZddlmZddlZddlmZddlmZddl m Z m Z m Z m Z ejdddejd<d Zd ZeGd d d eZed krd Zd ZddlZedS) N)DynamicTestCase) KDCBaseTest)KerberosCredentials)AES256_CTS_HMAC_SHA1_96ARCFOUR_HMAC_MD5KDC_ERR_S_PRINCIPAL_UNKNOWN NT_PRINCIPALz bin/python1ZPYTHONUNBUFFEREDFcseZdZejjejjejjdZddddddZ e fdd Z e d d Z d d Z fddZdddZdddZddZddZZS)SpnTests)ZcomputerZserverZrodczldap/{{account}}z&ldap/{{account}}/{netbios_domain_name}z"ldap/{{account}}/{dns_domain_name}zldap/{{account}}/testz'ldap/{{account}}:test/{dns_domain_name})Z2_partZ3_part_our_domainZ3_part_our_realmZ3_part_not_our_realmZ3_part_instancecstd|_dSN)super setUpClass_mock_rodc_creds)cls __class__z'SpnTests._format_spns..)map)r2spnsr;rr:r _format_spnsszSpnTests._format_spnscCs>|}|j||d}|dur:|}|j|d}|S)N)r4r5Zaccount)r*formatr+r,Z get_username)r2rr;r3Z account_namerrrr(s  zSpnTests._format_spncCsV||j}||jjur@|j}|durR||}|t|_n|j|d|id}|S)Nr)rZopts) r@rvaluesr-r1r_get_mock_rodc_credstypeZget_cached_creds)r2rr?r;rrrr's   zSpnTests._get_credsc Cs0|}|D](}|j|jd}||jvr |j|q |}t||j}t |}t |jtj d|d<| |t }||||j||j||j||j||j||||j|j|tjdgd}t|djddd} | | |!||} |"|| |S)NrAZservicePrincipalNamezmsDS-KeyVersionNumber)baseZscopeattrsr)idx)#Zget_mock_rodc_ctxrBZmynameZSPNsappendr*ldbZDnZacct_dnZMessageZMessageElementZFLAG_MOD_REPLACEZmodifyrZguessZget_lpZ set_realmr$upperZ set_domainZ domain_nameZ set_passwordZ acct_passZ set_usernameZset_workstationZsamnameZset_dnZset_spnsearchZ SCOPE_BASEintgetZset_kvnoZget_keysZcreds_set_keys) r2r?Zrodc_ctxrr3Zrodc_dnmsgr;resZkvnokeysrrrrDs@            zSpnTests._get_mock_rodc_creds)N)N)__name__ __module__ __qualname__rr-ZCOMPUTERr.r1rr classmethodr rr6r7r@r(r'rD __classcell__rrrrr +s( ;   r __main__)ossysZ samba.testsrrJZsamba.tests.krb5.kdc_base_testrZsamba.tests.krb5.raw_testcaserZ"samba.tests.krb5.rfc4120_constantsrrrrpathinsertenvironr8r9r rRZunittestmainrrrrs$    %