42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
|
"""Test methods in __main__."""
|
||
|
|
from unittest.mock import patch, PropertyMock
|
||
|
|
|
||
|
|
from homeassistant import __main__ as main
|
||
|
|
|
||
|
|
|
||
|
|
@patch('sys.exit')
|
||
|
|
def test_validate_python(mock_exit):
|
||
|
|
"""Test validate Python version method."""
|
||
|
|
with patch('sys.version_info',
|
||
|
|
new_callable=PropertyMock(return_value=(2, 7, 8))):
|
||
|
|
main.validate_python()
|
||
|
|
assert mock_exit.called is True
|
||
|
|
|
||
|
|
mock_exit.reset_mock()
|
||
|
|
|
||
|
|
with patch('sys.version_info',
|
||
|
|
new_callable=PropertyMock(return_value=(3, 2, 0))):
|
||
|
|
main.validate_python()
|
||
|
|
assert mock_exit.called is True
|
||
|
|
|
||
|
|
mock_exit.reset_mock()
|
||
|
|
|
||
|
|
with patch('sys.version_info',
|
||
|
|
new_callable=PropertyMock(return_value=(3, 4, 1))):
|
||
|
|
main.validate_python()
|
||
|
|
assert mock_exit.called is True
|
||
|
|
|
||
|
|
mock_exit.reset_mock()
|
||
|
|
|
||
|
|
with patch('sys.version_info',
|
||
|
|
new_callable=PropertyMock(return_value=(3, 4, 2))):
|
||
|
|
main.validate_python()
|
||
|
|
assert mock_exit.called is False
|
||
|
|
|
||
|
|
mock_exit.reset_mock()
|
||
|
|
|
||
|
|
with patch('sys.version_info',
|
||
|
|
new_callable=PropertyMock(return_value=(3, 5, 1))):
|
||
|
|
main.validate_python()
|
||
|
|
assert mock_exit.called is False
|