102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
|
"""Cmdline tests."""
|
||
|
|
from pyicloud import cmdline
|
||
|
|
from . import PyiCloudServiceMock, AUTHENTICATED_USER, REQUIRES_2SA_USER, DEVICES
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import pickle
|
||
|
|
import pytest
|
||
|
|
from unittest import TestCase
|
||
|
|
if sys.version_info >= (3, 3):
|
||
|
|
from unittest.mock import patch # pylint: disable=no-name-in-module,import-error
|
||
|
|
else:
|
||
|
|
from mock import patch
|
||
|
|
|
||
|
|
class TestCmdline(TestCase):
|
||
|
|
"""Cmdline test cases."""
|
||
|
|
main = None
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
cmdline.PyiCloudService = PyiCloudServiceMock
|
||
|
|
self.main = cmdline.main
|
||
|
|
|
||
|
|
def test_no_arg(self):
|
||
|
|
"""Test no args."""
|
||
|
|
with pytest.raises(SystemExit, match="2"):
|
||
|
|
self.main()
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit, match="2"):
|
||
|
|
self.main(None)
|
||
|
|
|
||
|
|
with pytest.raises(SystemExit, match="2"):
|
||
|
|
self.main([])
|
||
|
|
|
||
|
|
def test_help(self):
|
||
|
|
"""Test the help command."""
|
||
|
|
with pytest.raises(SystemExit, match="0"):
|
||
|
|
self.main(['--help'])
|
||
|
|
|
||
|
|
def test_username(self):
|
||
|
|
"""Test the username command."""
|
||
|
|
# No username supplied
|
||
|
|
with pytest.raises(SystemExit, match="2"):
|
||
|
|
self.main(['--username'])
|
||
|
|
|
||
|
|
@patch("getpass.getpass")
|
||
|
|
def test_username_password_invalid(self, mock_getpass):
|
||
|
|
"""Test username and password commands."""
|
||
|
|
# No password supplied
|
||
|
|
mock_getpass.return_value = None
|
||
|
|
with pytest.raises(SystemExit, match="2"):
|
||
|
|
self.main(['--username', 'invalid_user'])
|
||
|
|
|
||
|
|
# Bad username or password
|
||
|
|
mock_getpass.return_value = "invalid_pass"
|
||
|
|
with pytest.raises(RuntimeError, match="Bad username or password for invalid_user"):
|
||
|
|
self.main(['--username', 'invalid_user'])
|
||
|
|
|
||
|
|
# We should not use getpass for this one, but we reset the password at login fail
|
||
|
|
with pytest.raises(RuntimeError, match="Bad username or password for invalid_user"):
|
||
|
|
self.main(['--username', 'invalid_user', '--password', 'invalid_pass'])
|
||
|
|
|
||
|
|
|
||
|
|
@patch('pyicloud.cmdline.input')
|
||
|
|
def test_username_password_requires_2sa(self, mock_input):
|
||
|
|
"""Test username and password commands."""
|
||
|
|
# Valid connection for the first time
|
||
|
|
mock_input.return_value = "0"
|
||
|
|
with pytest.raises(SystemExit, match="0"):
|
||
|
|
self.main([
|
||
|
|
'--username', REQUIRES_2SA_USER,
|
||
|
|
'--password', 'valid_pass',
|
||
|
|
'--non-interactive',
|
||
|
|
])
|
||
|
|
|
||
|
|
def test_device_outputfile(self):
|
||
|
|
"""Test the outputfile command."""
|
||
|
|
with pytest.raises(SystemExit, match="0"):
|
||
|
|
self.main([
|
||
|
|
'--username', AUTHENTICATED_USER,
|
||
|
|
'--password', 'valid_pass',
|
||
|
|
'--non-interactive',
|
||
|
|
'--outputfile'
|
||
|
|
])
|
||
|
|
|
||
|
|
for key in DEVICES:
|
||
|
|
file_name = DEVICES[key].content['name'].strip().lower() + ".fmip_snapshot"
|
||
|
|
|
||
|
|
pickle_file = open(file_name, "rb")
|
||
|
|
assert pickle_file
|
||
|
|
|
||
|
|
contents = []
|
||
|
|
with pickle_file as opened_file:
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
contents.append(pickle.load(opened_file))
|
||
|
|
except EOFError:
|
||
|
|
break
|
||
|
|
assert contents == [DEVICES[key].content]
|
||
|
|
|
||
|
|
pickle_file.close()
|
||
|
|
os.remove(file_name)
|