
    tni]                     D    d Z ddlZddlmZ  G d dej                  ZdS )+Tests for the endpoint mutation generation.    N)generate_endpoint_mutationc                       e Zd ZdZd Zd ZdS )TestGenerateEndpointMutationr   c                    t          dd          }|                     d|           |                     d|           |                     d|           |                     d|           |                     d|           dS )	zTest the required fields.	test_nametest_template_idzname: "test_name"templateId: "test_template_id"zgpuIds: "AMPERE_16"znetworkVolumeId: ""zlocations: ""Nr   assertInselfresults     c/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/tests/test_api/test_mutation_endpoints.pytest_required_fieldsz1TestGenerateEndpointMutation.test_required_fields   s    +K9KLL)62226???+V444+V444ov.....    c                    t          ddddddddd	d
d          }|                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           |                     d|           dS )zTest all the fields.r   r	   	AMPERE_20test_volume_idUS_WEST
   WORKER_COUNT         Tzname: "test_name-fb"r
   zgpuIds: "AMPERE_20"z!networkVolumeId: "test_volume_id"zlocations: "US_WEST"zidleTimeout: 10zscalerType: "WORKER_COUNT"zscalerValue: 5zworkersMin: 2zworkersMax: 4Nr   r   s     r   test_all_fieldsz,TestGenerateEndpointMutation.test_all_fields   s
   +
 
 	,f5556???+V44496BBB,f555'0002F;;;&///ov...ov.....r   N)__name__
__module____qualname____doc__r   r    r   r   r   r      s8        55/ / // / / / /r   r   )r    unittestrunpod.api.mutations.endpointsr   TestCaser   r!   r   r   <module>r%      s]    1 1  E E E E E E$/ $/ $/ $/ $/8#4 $/ $/ $/ $/ $/r   