mirror of
https://github.com/vikstrous/pirate-get
synced 2025-01-10 10:04:21 +01:00
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
import importlib
|
||
|
|
from unittest import mock
|
||
|
|
from unittest.mock import patch, call, MagicMock
|
||
|
|
|
||
|
|
import pirate.pirate
|
||
|
|
|
||
|
|
class TestPirate(unittest.TestCase):
|
||
|
|
|
||
|
|
@patch('sys.exit')
|
||
|
|
def test_unsupported(self, mock_exit):
|
||
|
|
sys.version = '3.2.1 (default, dec 7 2015, 12:58:09) \n[gcc 5.2.0]'
|
||
|
|
importlib.reload(pirate.pirate)
|
||
|
|
mock_exit.assert_called_once_with(1)
|
||
|
|
|
||
|
|
@patch('sys.exit')
|
||
|
|
def test_unsupported2(self, mock_exit):
|
||
|
|
sys.version = '2.5.1 (default, dec 7 2015, 12:58:09) \n[gcc 5.2.0]'
|
||
|
|
importlib.reload(pirate.pirate)
|
||
|
|
mock_exit.assert_called_once_with(1)
|
||
|
|
|
||
|
|
|
||
|
|
@patch('sys.exit')
|
||
|
|
def test_supported(self, mock_exit):
|
||
|
|
sys.version = '3.5.1 (default, dec 7 2015, 12:58:09) \n[gcc 5.2.0]'
|
||
|
|
importlib.reload(pirate.pirate)
|
||
|
|
mock_exit.assert_not_called()
|
||
|
|
|
||
|
|
@patch('sys.exit')
|
||
|
|
def test_supported_exact(self, mock_exit):
|
||
|
|
sys.version = '3.4.0 (default, dec 7 2015, 12:58:09) \n[gcc 5.2.0]'
|
||
|
|
importlib.reload(pirate.pirate)
|
||
|
|
mock_exit.assert_not_called()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|