
    tni                     D    d Z ddlZddlmZ  G d dej                  ZdS )zN Unit tests for the function generate_pod_template in the file api_wrapper.py     N)generate_pod_templatec                       e Zd ZdZd Zd ZdS )TestGeneratePodTemplatezLUnit tests for the function generate_pod_template in the file api_wrapper.pyc                    t          dd          }|                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d	|           |                     d
|           dS )zDTest the basic required fields are present in the generated template	test_nametest_image_namezname: "test_name"zimageName: "test_image_name"zdockerArgs: ""zcontainerDiskInGb: 10zvolumeInGb: 0z	ports: ""zenv: []zisServerless: falseNr   assertInselfresults     d/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/tests/test_api/test_mutations_templates.pytest_basic_required_fieldsz2TestGeneratePodTemplate.test_basic_required_fields   s    &{4EFF)62224f===&///-v666ov...k6***i(((+V44444    c                 p   t          ddddddddd	d
d	  	        }|                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           dS )z>Test the optional fields are present in the generated templater   r   test_cmd   z/path/to/volumez
8000, 8001val1val2)VAR1VAR2T	test_auth)docker_start_cmdvolume_in_gbvolume_mount_pathportsenvis_serverlessregistry_auth_idzdockerArgs: "test_cmd"zvolumeInGb: 5z"volumeMountPath: "/path/to/volume"zports: "8000,8001"zEenv: [{ key: "VAR1", value: "val1" }, { key: "VAR2", value: "val2" }]zisServerless: truez%containerRegistryAuthId : "test_auth"Nr	   r   s     r   test_optional_fieldsz,TestGeneratePodTemplate.test_optional_fields   s    &'/00(

 

 

 	.777ov...:FCCC*F333S	
 	
 	
 	*F333=vFFFFFr   N)__name__
__module____qualname____doc__r   r     r   r   r   r      s=        VV
5 
5 
5G G G G Gr   r   )r$   unittestrunpod.api.mutations.templatesr   TestCaser   r%   r   r   <module>r)      sg    T T  @ @ @ @ @ @%G %G %G %G %Gh/ %G %G %G %G %Gr   