1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import unittest
try:
from unittest.mock import patch
except ImportError:
from mock import patch
from adb_shell.auth.keygen import get_user_info
class TestKeygen(unittest.TestCase):
def test_get_user_info(self):
with patch('adb_shell.auth.keygen.os.getlogin', side_effect=OSError), patch('adb_shell.auth.keygen.socket.gethostname', return_value=''):
user_host = get_user_info()
self.assertEqual(user_host, ' unknown@unknown')
with patch('adb_shell.auth.keygen.os.getlogin', return_value=''), patch('adb_shell.auth.keygen.socket.gethostname', return_value=''):
user_host = get_user_info()
self.assertEqual(user_host, ' unknown@unknown')
|
