Current File : //proc/self/root/usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyc
�
���^c @@ s� d Z d d l m Z m Z m Z d d l Z d d l Z d d l m Z e j
d d f k rn d d l Z n d d l Z d d l
Z
d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z d d l m Z d d
l m Z e � Z d d l m Z m Z d d l m Z m! Z! m" Z" d d
l# m$ Z$ d � Z% d � Z& d e j' f d � � YZ( d � Z) d � Z* d � Z+ e, d k o�e- e. k r�e+ � n d S( u:
Tests to try out salt key.RaetKey Potentially ephemeral
i ( t absolute_importt print_functiont unicode_literalsN( t mapi i ( t odict( t Timert
StoreTimer( t storing( t
getConsole( t raetingt nacling( t estatingt keepingt stacking( t RaetKeyc C@ s t j d t j j � d S( Nt verbosity( t consolet reinitt Wordaget concise( ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt setUpModule# s c C@ s d S( N( ( ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt tearDownModule' s t
BasicTestCasec B@ sD e Z d Z d � Z d � Z d � Z d � Z d � Z d � Z RS( u c C@ sH t j d d � | _ t d | j d d � | _ t j d d d d d
d � | _ t j j
| j d � } t j j | � s� t j | � n t j j
| d
� } t j j | � s� t j | � n t j j
| d � } t j j | � s� t j | � n t j j
| d � } t j j | � s0t j | � n t j j
| d � | _
t j j | j
� r�t j | j
� j } t j | j
| t j Bt j B� n t j j
| j d � | _ t j j
| j d � | _ t d d d d d | d | j d | j d t d t d d � | _ t d | j � | _ t j d d d d d
d � | _ d S( Nt stampg t storet durationg �?t prefixu saltt suffixu maint diru /tmpu pkiu acceptedu pendingu rejectedu local.keyu cacheu sockt __roleu mastert idt pki_dirt sock_dirt cachedirt open_modet auto_acceptt transportu raett optsu base( R t StoreR R t timert tempfilet mkdtempt saltDirpatht ost patht joint existst makedirst
localFilepatht statt st_modet chmodt S_IWUSRt cacheDirpatht sockDirpatht dictt Falset TrueR% R t
mainKeepert baseDirpath( t selft
pkiDirpatht acceptedDirpatht pendingDirpatht rejectedDirpatht mode( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt setUp. s@ !$ c C@ s, t j j | j � r( t j | j � n d S( N( R+ R, R. R* t shutilt rmtree( R< ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt tearDownZ s c C@ s� t � } | | d <t j j | d d | � | d <t j � } | j | d <| j | d <t j � } | j | d <| j | d <| S( u8
Creates odict and populates with data to setup road stack
{
name: stack name local estate name
dirpath: dirpath for keep files
sighex: signing key
verhex: verify key
prihex: private key
pubhex: public key
}
u nameu roadu keepu dirpathu sighexu verhexu prihexu pubhex(
R R+ R, R- R
t Signert keyhext verhext Privateert pubhex( R< t namet baset datat signert privateer( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt createRoadData^ s
c
C@ s� t j d j | j j � � t | j d <| j | j d � | j | j j
� i g d 6g d 6g d 6g d 6� | j j � } | j | i � | j d d d | j
� } | j j | d
| d � | j j � } | j | i t j j j | d
� d 6t j j j | d � d
6� | j j
� } | j | i g d 6| j g d 6g d 6g d 6� | j d d d | j
� } | j d d d | j
� } | j j | d | d | d � } | j | d � | j j | d | d | d � } | j | d � | j j
� } | j | i d d g d 6| j g d 6g d 6g d 6� | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j � } | j | i d d g d 6g d 6g d 6� | j j � } | j | i i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6� d S( u?
Basic function of RaetKey in auto accept mode
u {0}
u auto_acceptu acceptedu localu rejectedu pendingRK u mainRL u prihexu sighexu privu signu other1u other2u nameu pubhexu verhexu minion_idu pubu verifyu
acceptanceN( R t terset formatt testAutoAcceptt __doc__R9 R% t
assertTruet assertDictEqualR: t all_keyst
read_localRP R; t write_localt saltt utilst stringutilst to_strR0 t statust assertEqualt read_remotet list_keyst read_all_remote(
R< t localkeyst maint allkeyst other1t other2R^ t
remotekeyst listkeyst
allremotekeys( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyRS v sl
$$
c
C@ s| t j d j | j j � � t | j d <| j | j d � | j | j j
� i g d 6g d 6g d 6g d 6� | j j � } | j | i � | j d d d | j
� } | j j | d
| d � | j j � } | j | i t j j j | d
� d 6t j j j | d � d
6� | j j
� } | j | i g d 6| j g d 6g d 6g d 6� | j d d d | j
� } | j d d d | j
� } | j j | d | d | d � } | j | d � | j j | d | d | d � } | j | d � | j j
� } | j | i g d 6| j g d 6d d g d 6g d 6� | j j | d � } | j | i � | j j | d � } | j | i � | j j � } | j | i g d 6g d 6d d g d 6� | j j � } | j | i i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6� | j j � | j j
� } | j | i d d g d 6| j g d 6g d 6g d 6� | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j � } | j | i d d g d 6g d 6g d 6� | j j � } | j | i i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6� d S( uC
Basic function of RaetKey in non auto accept mode
u {0}
u auto_acceptu acceptedu localu rejectedu pendingRK u mainRL u prihexu sighexu privu signu other1u other2u nameu pubhexu verhexu verifyu minion_idu
acceptanceu pubN( R RQ RR RS RT R8 R% t assertFalseRV R: RW RX RP R; RY RZ R[ R\ R] R0 R^ R_ R` Ra Rb t
accept_all(
R< Rc Rd Re Rf Rg R^ Rh Ri Rj ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt testManualAccept� s�
$$
"
c
C@ sr t j d j | j j � � t | j d <| j | j d � | j | j j
� i g d 6g d 6g d 6g d 6� | j j � } | j | i � | j d d d | j
� } | j j | d
| d � | j j � } | j | i t j j j | d
� d 6t j j j | d � d
6� | j j
� } | j | i g d 6| j g d 6g d 6g d 6� | j d d d | j
� } | j d d d | j
� } | j j | d | d | d � } | j | d � | j j | d | d | d � } | j | d � | j j
� } | j | i d d g d 6| j g d 6g d 6g d 6� | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j � } | j | i d d g d 6g d 6g d 6� | j j � } | j | i i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6� | j j d | d � | j j
� } | j | i d g d 6| j g d 6g d 6g d 6� | j j | d � } | j | i � | j j | d � } | j | i d d 6t j j j | d � d 6t j j j | d � d 6� | j j � } | j | i d g d 6g d 6g d 6� | j j � } | j | i i t j j j | d � d 6d d 6d d 6t j j j | d � d 6d 6� d S( u9
Basic function of RaetKey to delete key
u {0}
u auto_acceptu acceptedu localu rejectedu pendingRK u mainRL u prihexu sighexu privu signu other1u other2u nameu pubhexu verhexu minion_idu pubu verifyu
acceptancet matchN( R RQ RR t
testDeleteRT R9 R% RU RV R: RW RX RP R; RY RZ R[ R\ R] R0 R^ R_ R` Ra Rb t
delete_key(
R< Rc Rd Re Rf Rg R^ Rh Ri Rj ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyRo ' s�
$$
"
( t __name__t
__module__RT RB RE RP RS Rm Ro ( ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyR + s , F kc C@ s; t | � } t j | g � } t j d d � j | � d S( u
Unittest Runner
R i N( R t unittestt TestSuitet TextTestRunnert run( t testt suite( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt runOne� s c C@ sc g } d d d g } | j t t t t | � � � � t j | � } t j d d � j | � d S( u
Unittest runner
u testAutoAcceptu testManualAcceptu
testDeleteR i N( t extendt listR R Rs Rt Ru Rv ( t testst namesRx ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt runSome� s "c C@ sE t j � } | j t j � j t � � t j d d � j | � d S( u
Unittest runner
R i N( Rs Rt t addTestt
TestLoadert loadTestsFromTestCaseR Ru Rv ( Rx ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt runAll� s u __main__(/ RT t
__future__R R R t syst salt.utils.stringutilsRZ t salt.ext.six.movesR t version_infot unittest2Rs R+ R1 t timeR( RC t ioflo.aid.odictingR t ioflo.aid.timingR R t
ioflo.baseR t ioflo.base.consolingR R t raetR R
t raet.roadR R R
t salt.keyR R R t TestCaseR Ry R~ R� Rq t __package__t None( ( ( sB /usr/lib/python2.7/site-packages/salt/daemons/test/test_raetkey.pyt <module> s: � g
Mini Shell