test_custom_evaluate_timeout

Complexity

Total Complexity

2

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	TestCustomEvaluateTimeout.test_custom_evaluate_timeout_with_script()	0	28	1
A	TestCustomEvaluateTimeout._get_evaluate_timeout()	0	2	1

import integ_test_base


class TestCustomEvaluateTimeout(integ_test_base.IntegTestBase):
    def _get_evaluate_timeout(self) -> str:
        return '5'

    def test_custom_evaluate_timeout_with_script(self):
        payload = (
            '''
            {
                "data": { "_arg1": 1 },
                "script":
                "import time\\nwhile True:\\n    time.sleep(1)\\nreturn 1"
            }
            ''')
        headers = {
            'Content-Type':
                "application/json",
            'TabPy-Client':
                "Integration test for testing custom evaluate timeouts with "
                "scripts."
        }

        conn = self._get_connection()
        conn.request('POST', '/evaluate', payload, headers)
        res = conn.getresponse()
        actual_error_message = res.read().decode('utf-8')

        self.assertEqual(
            '{"message": '
            '"User defined script timed out. Timeout is set to 5.0 s.", '
            '"info": {}}',
            actual_error_message)
        self.assertEqual(408, res.status)