a
    )&ig/                     @   s   d dl Z d dlZd dlZd dlmZ d dlm  m  mZ e j	
d d dejd< dZdZG dd deZed	krdZdZd dlZe  dS )
    N)AsReqBaseTestz
bin/python1ZPYTHONUNBUFFEREDFc                       sz  e Zd Z fddZddddZdd Zd	d
 Zdd Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Z d?d@ Z!dAdB Z"dCdD Z#dEdF Z$dGdH Z%dIdJ Z&dKdL Z'dMdN Z(dOdP Z)dQdR Z*dSdT Z+dUdV Z,dWdX Z-dYdZ Z.  Z/S )[	SaltTestsc                    s   t    t| _t| _d S )N)supersetUpglobal_asn1_printZdo_asn1_printglobal_hexdumpZ
do_hexdump)self	__class__ =/usr/lib/python3/dist-packages/samba/tests/krb5/salt_tests.pyr   $   s    
zSaltTests.setUpN)optsc                C   s2   z| j ||dW S  tjy,   |   Y n0 d S )Naccount_typer   )Zget_cached_credsldbZLdbErrorZfail)r	   r   r   r   r   r   
_get_creds)   s    zSaltTests._get_credsc                 C   sN   |  |}| | | |}| |d d tjj | |d d | d S )Nr   etypeZsalt)Zget_saltZassertIsNotNone_run_as_req_enc_timestampZassertEqualkcryptoZEnctypeZAES256)r	   client_credsZexpected_saltZetype_info2r   r   r   _run_salt_test3   s
    


zSaltTests._run_salt_testc                 C   s$   | j | jjddid}| | d S Nname_suffixzfoo@barr   r   AccountTypeUSERr   r	   r   r   r   r   test_salt_at_user<   s
    zSaltTests.test_salt_at_userc                 C   s$   | j | jjddid}| | d S r   r   r   COMPUTERr   r   r   r   r   test_salt_at_macB   s
    zSaltTests.test_salt_at_macc                 C   s$   | j | jjddid}| | d S Nr   zFoo@barr   r   r   r   r   r   test_salt_at_case_userH   s
    z SaltTests.test_salt_at_case_userc                 C   s$   | j | jjddid}| | d S r"   r   r   r   r   r   test_salt_at_case_macN   s
    zSaltTests.test_salt_at_case_macc                 C   s$   | j | jjddid}| | d S Nr   zfoo@@barr   r   r   r   r   r   test_salt_double_at_userT   s
    z"SaltTests.test_salt_double_at_userc                 C   s$   | j | jjddid}| | d S r%   r   r   r   r   r   test_salt_double_at_macZ   s
    z!SaltTests.test_salt_double_at_macc                 C   s$   | j | jjddid}| | d S NZname_prefixz@foor   r   r   r   r   r   test_salt_at_start_user`   s
    z!SaltTests.test_salt_at_start_userc                 C   s$   | j | jjddid}| | d S r(   r   r   r   r   r   test_salt_at_start_macf   s
    z SaltTests.test_salt_at_start_macc                 C   s$   | j | jjddid}| | d S Nr   foo@r   r   r   r   r   r   test_salt_at_end_userl   s
    zSaltTests.test_salt_at_end_userc                 C   s$   | j | jjddid}| | d S r+   r   r   r   r   r   test_salt_at_end_macr   s
    zSaltTests.test_salt_at_end_macc                 C   s&   | j | jjdddd}| | d S )Nr,   Fr   
add_dollarr   r   r   r   r   r   test_salt_at_end_no_dollar_macx   s    z(SaltTests.test_salt_at_end_no_dollar_macc                 C   s$   | j | jjddid}| | d S )Nr0   Fr   r   r   r   r   r   test_salt_no_dollar_mac   s
    z!SaltTests.test_salt_no_dollar_macc                 C   s&   | j | jjdddd}| | d S )Nfoo$barFr/   r   r   r   r   r   r   test_salt_dollar_mid_mac   s    z"SaltTests.test_salt_dollar_mid_macc                 C   s$   | j | jjddid}| | d S Nr   r3   r   r   r   r   r   r   test_salt_dollar_user   s
    zSaltTests.test_salt_dollar_userc                 C   s$   | j | jjddid}| | d S r5   r   r   r   r   r   test_salt_dollar_mac   s
    zSaltTests.test_salt_dollar_macc                 C   s$   | j | jjddid}| | d S Nr   zfoo$r   r   r   r   r   r   test_salt_dollar_end_user   s
    z#SaltTests.test_salt_dollar_end_userc                 C   s$   | j | jjddid}| | d S r8   r   r   r   r   r   test_salt_dollar_end_mac   s
    z"SaltTests.test_salt_dollar_end_macc                 C   s$   | j | jjddid}| | d S )NupnZfoo0r   r   r   r   r   r   test_salt_upn_user   s
    zSaltTests.test_salt_upn_userc                 C   s$   | j | jjddid}| | d S )Nr;   Zfoo1r   r   r   r   r   r   test_salt_upn_mac   s
    zSaltTests.test_salt_upn_macc                 C   s$   | j | jjddid}| | d S )Nr;   z	host/foo2r   r   r   r   r   r   test_salt_upn_host_user   s
    z!SaltTests.test_salt_upn_host_userc                 C   s$   | j | jjddid}| | d S )Nr;   z	host/foo3r   r   r   r   r   r   test_salt_upn_host_mac   s
    z SaltTests.test_salt_upn_host_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zfoo4@r   	get_samdbdomain_dns_namer   r   r   r   r	   Zrealmr   r   r   r   test_salt_upn_realm_user   s    
z"SaltTests.test_salt_upn_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zfoo5@r   rA   rB   r   r   r    r   rC   r   r   r   test_salt_upn_realm_mac   s    
z!SaltTests.test_salt_upn_realm_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   z
host/foo6@r   r@   rC   r   r   r   test_salt_upn_host_realm_user   s    
z'SaltTests.test_salt_upn_host_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   z
host/foo7@r   rE   rC   r   r   r   test_salt_upn_host_realm_mac   s    
z&SaltTests.test_salt_upn_host_realm_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zfoo8$@r   r@   rC   r   r   r   test_salt_upn_dollar_realm_user   s    
z)SaltTests.test_salt_upn_dollar_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zfoo9$@r   rE   rC   r   r   r   test_salt_upn_dollar_realm_mac   s    
z(SaltTests.test_salt_upn_dollar_realm_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zhost/foo10$@r   r@   rC   r   r   r   $test_salt_upn_host_dollar_realm_user   s    
z.SaltTests.test_salt_upn_host_dollar_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zhost/foo11$@r   rE   rC   r   r   r   #test_salt_upn_host_dollar_realm_mac   s    
z-SaltTests.test_salt_upn_host_dollar_realm_macc                 C   s$   | j | jjddid}| | d S )Nr;   zfoo12@other.realmr   r   r   r   r   r   test_salt_upn_other_realm_user   s
    z(SaltTests.test_salt_upn_other_realm_userc                 C   s$   | j | jjddid}| | d S )Nr;   zfoo13@other.realmr   r   r   r   r   r   test_salt_upn_other_realm_mac   s
    z'SaltTests.test_salt_upn_other_realm_macc                 C   s$   | j | jjddid}| | d S )Nr;   zhost/foo14@other.realmr   r   r   r   r   r   #test_salt_upn_host_other_realm_user   s
    z-SaltTests.test_salt_upn_host_other_realm_userc                 C   s$   | j | jjddid}| | d S )Nr;   zhost/foo15@other.realmr   r   r   r   r   r   "test_salt_upn_host_other_realm_mac  s
    z,SaltTests.test_salt_upn_host_other_realm_macc                 C   s$   | j | jjddid}| | d S )Nr;   ZFoo16r   r   r   r   r   r   test_salt_upn_case_user  s
    z!SaltTests.test_salt_upn_case_userc                 C   s$   | j | jjddid}| | d S )Nr;   ZFoo17r   r   r   r   r   r   test_salt_upn_case_mac  s
    z SaltTests.test_salt_upn_case_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zfoo$18@r   r@   rC   r   r   r   #test_salt_upn_dollar_mid_realm_user  s    
z-SaltTests.test_salt_upn_dollar_mid_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zfoo$19@r   rE   rC   r   r   r   "test_salt_upn_dollar_mid_realm_mac  s    
z,SaltTests.test_salt_upn_dollar_mid_realm_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zhost/foo$20@r   r@   rC   r   r   r   (test_salt_upn_host_dollar_mid_realm_user&  s    
z2SaltTests.test_salt_upn_host_dollar_mid_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   zhost/foo$21@r   rE   rC   r   r   r   'test_salt_upn_host_dollar_mid_realm_mac-  s    
z1SaltTests.test_salt_upn_host_dollar_mid_realm_macc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   z
foo22@bar@r   r@   rC   r   r   r   test_salt_upn_at_realm_user4  s    
z%SaltTests.test_salt_upn_at_realm_userc                 C   s4   |    }| j| jjdd| id}| | d S )Nr;   z
foo23@bar@r   rE   rC   r   r   r   test_salt_upn_at_realm_mac;  s    
z$SaltTests.test_salt_upn_at_realm_mac)0__name__
__module____qualname__r   r   r   r   r!   r#   r$   r&   r'   r)   r*   r-   r.   r1   r2   r4   r6   r7   r9   r:   r<   r=   r>   r?   rD   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   __classcell__r   r   r
   r   r   "   sZ   
	r   __main__)sysosr   Zsamba.tests.krb5.as_req_testsr   Zsamba.tests.krb5.kcryptoZtestsZkrb5r   pathinsertenvironr   r   r   rY   Zunittestmainr   r   r   r   <module>   s    
  #