
    tni                     L    d Z ddlZddlmZmZmZ  G d dej                  ZdS )<Unit tests for the error classes in the runpod.error module.    N)AuthenticationError
QueryErrorRunPodErrorc                   0    e Zd ZdZd Zd Zd Zd Zd ZdS )TestErrorClassesr   c                 n    d}t          |          }|                     t          |          |           dS )z*Test the RunPodError class with a message.zAn error occurredNr   assertEqualstrself	error_msgerrs      M/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/tests/test_error.pytest_run_pod_error_with_messagez0TestErrorClasses.test_run_pod_error_with_message   s6    '	)$$S9-----    c                 h    t                      }|                     t          |          d           dS )z-Test the RunPodError class without a message.NoneNr
   r   r   s     r   "test_run_pod_error_without_messagez3TestErrorClasses.test_run_pod_error_without_message   s-    mmS6*****r   c                 n    d}t          |          }|                     t          |          |           dS )z#Test the AuthenticationError class.zAuthentication failedN)r   r   r   r   s      r   test_authentication_errorz*TestErrorClasses.test_authentication_error   s6    +	!),,S9-----r   c                     d}d}t          ||          }|                     t          |          |           |                     |j        |           dS )z3Test the QueryError class with a message and query.zQuery failedz(SELECT * FROM some_table WHERE conditionN)r   r   r   query)r   r   	query_strr   s       r   'test_query_error_with_message_and_queryz8TestErrorClasses.test_query_error_with_message_and_query   sU    "	>	I..S9---I.....r   c                     t                      }|                     t          |          d           |                     |j                   dS )z5Test the QueryError class without a message or query.r   N)r   r   r   assertIsNoner   r   s     r   *test_query_error_without_message_and_queryz;TestErrorClasses.test_query_error_without_message_and_query%   sC    llS6***#)$$$$$r   N)	__name__
__module____qualname____doc__r   r   r   r   r     r   r   r   r   	   se        FF. . .+ + +
. . ./ / /% % % % %r   r   )r$   unittestrunpod.errorr   r   r   TestCaser   r%   r   r   <module>r)      sl    B B  F E E E E E E E E E %  %  %  %  %x(  %  %  %  %  %r   