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>
35 lines
983 B
Python
Executable File
35 lines
983 B
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
|
|
|
|
Unit tests for image_capture.py
|
|
"""
|
|
import unittest.mock as mock
|
|
import unittest
|
|
|
|
import pytest
|
|
|
|
import ly_test_tools.image.image_capture
|
|
|
|
pytestmark = pytest.mark.SUITE_smoke
|
|
|
|
|
|
class TestScreenCap(unittest.TestCase):
|
|
|
|
@mock.patch('pyscreenshot.grab')
|
|
def test_ScreenCap_CoordsAndNameGiven_Used(self, mock_grab):
|
|
mock_grab.return_value = mock.MagicMock()
|
|
mock_grab.return_value.save = mock.MagicMock()
|
|
x1 = 10
|
|
x2 = 20
|
|
y1 = 30
|
|
y2 = 40
|
|
image_name = 'test_capture.png'
|
|
|
|
ly_test_tools.image.image_capture.screencap(x1, y1, x2, y2, filename=image_name)
|
|
|
|
mock_grab.assert_called_once_with(bbox=(x1, y1, x2, y2), childprocess=False)
|
|
mock_grab.return_value.save.assert_called_once_with(image_name)
|