Steve Pham
38261d0800
* Updated all copyright headers to split the longer original copyright line into 2 shorter lines Signed-off-by: Steve Pham <spham@amazon.com>
46 lines
1.4 KiB
Python
Executable File
46 lines
1.4 KiB
Python
Executable File
"""
|
|
Copyright (c) Contributors to the Open 3D Engine Project.
|
|
For complete copyright and license terms please see the LICENSE at the root of this distribution.
|
|
|
|
SPDX-License-Identifier: Apache-2.0 OR MIT
|
|
"""
|
|
|
|
import unittest.mock as mock
|
|
import unittest
|
|
|
|
import time
|
|
import pytest
|
|
|
|
import ly_test_tools.environment.waiter
|
|
|
|
pytestmark = pytest.mark.SUITE_smoke
|
|
|
|
|
|
@mock.patch('time.sleep', mock.MagicMock)
|
|
class TestWaitFor(unittest.TestCase):
|
|
|
|
def test_WaitForFunctionCall_GivenExceptionTimeoutExceeded_RaiseException(self):
|
|
input_func = mock.MagicMock()
|
|
input_func.return_value = False
|
|
|
|
with self.assertRaises(Exception):
|
|
ly_test_tools.environment.waiter.wait_for(input_func, .001, Exception, 0)
|
|
|
|
def test_WaitForFunctionCall_TimeoutExceeded_RaiseAssertionError(self):
|
|
input_func = mock.MagicMock()
|
|
input_func.return_value = False
|
|
|
|
with self.assertRaises(Exception):
|
|
ly_test_tools.environment.waiter.wait_for(input_func, .001, interval=0)
|
|
|
|
def test_WaitForFunctionCall_TimeoutExceeded_EnoughTime(self):
|
|
input_func = mock.MagicMock()
|
|
input_func.return_value = False
|
|
|
|
timeout_end = time.time() + 0.1
|
|
try:
|
|
ly_test_tools.environment.waiter.wait_for(input_func, 0.1, Exception, interval=0.01)
|
|
except Exception:
|
|
pass
|
|
# It should have taken at least 1/10 second
|
|
assert time.time() > timeout_end |