
    EԤg                         d dl mZmZmZ d dlmZmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZ d dlmZ  e            Z G d de          ZdS )    )TestLdapConnector	ReconTaskReconResultTask)DumpIDMIDWithAPITaskGenerateUserGroups)MakeUsersTask)DSLdapModifyTask)GatlingTask)ReportSimulation)ValidationNumUsers)AM)StepTask)get_pyrock_run)PlatformUtils)FailExceptionc                   0    e Zd Zd Zed             Zd ZdS )UpdateRcsClientSecretTaskc                     | j         dk    rt          d          t          | j        t                    st          d| j         d          d S )N
controllerz#Task must be executed on controllerzTarget z must be an AM component)sourcer   
isinstancetargetr   
ValueErrorselfs    \/mnt/disk1/xslou/workshop/lodestar-fork/pyrock/tests/usecases/idc_idm_recon/tasks/mytasks.pyprezUpdateRcsClientSecretTask.pre   s[    ;,&& EFFF$+r** 	NLt{LLLMMM	N 	N    c                      t                               d           t          t                                                                                      d S )NzReset RCSClient password)
components)
pyrock_runlogr   get_componentsset_rcs_client_secret r   r   step1zUpdateRcsClientSecretTask.step1!   sD    1222!:!:!<!<===SSUUUUUr   c                 .    |                                   dS )zPost methodN)set_result_passr   s    r   postzUpdateRcsClientSecretTask.post&   s    r   N)__name__
__module____qualname__r   staticmethodr&   r)   r%   r   r   r   r      sS        N N N V V \V    r   r   N)pyrock.tasks.idm.reconr   r   r   pyrock.tasks.idm.generalr   r    pyrock.tasks.deployment.datainitr   pyrock.tasks.scenario.ds_sdkr	   pyrock.tasks.scenario.gatlingr
   -pyrock.lib.report.json.tasks.ReportSimulationr   "pyrock.tasks.deployment.validationr   shared.lib.components.amr   #pyrock.lib.scheduler.tasks.StepTaskr   pyrock.lib.PyRockRunr   shared.lib.platform_utilsr   shared.lib.utils.exceptionr   r!   r   r%   r   r   <module>r:      s3   Q P P P P P P P P P M M M M M M M M : : : : : : 9 9 9 9 9 9 5 5 5 5 5 5 J J J J J J A A A A A A ' ' ' ' ' ' 8 8 8 8 8 8 / / / / / / 3 3 3 3 3 3 4 4 4 4 4 4^
         r   