1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
#!/usr/bin/python
import mock
import os
class VncTest(mock.TestCase):
def setUp(self):
self.setupModules(["_isys", "block", "logging", "ConfigParser"])
self.fs = mock.DiskIO()
import pyanaconda
pyanaconda.anaconda_log = mock.Mock()
self.OK = 22
import pyanaconda.vnc
pyanaconda.vnc.log = mock.Mock()
pyanaconda.vnc.os = mock.Mock()
pyanaconda.vnc.subprocess = mock.Mock()
pyanaconda.vnc.subprocess.Popen().communicate.return_value = (1, 2)
pyanaconda.vnc.subprocess.Popen().returncode = self.OK
pyanaconda.vnc.open = self.fs.open
self.ROOT = '/'
self.DISPLAY = '2'
self.DESKTOP = 'Desktop'
self.PASS = ''
self.LOG_FILE = '/tmp/vnc.log'
self.PW_INIT_FILE = '/tmp/vncpassword.dat'
self.PW_FILE = '/tmp/vncpassword'
self.VNCCONNECTHOST = 'host'
def tearDown(self):
self.tearDownModules()
def recover_vnc_password_1_test(self):
import pyanaconda.vnc
pyanaconda.vnc.open = mock.Mock(side_effect=IOError())
server = pyanaconda.vnc.VncServer()
server.recoverVNCPassword()
self.assertEqual(server.password, '')
def recover_vnc_password_2_test(self):
import pyanaconda.vnc
self.fs.open(self.PW_INIT_FILE, 'w').write('abcdef')
server = pyanaconda.vnc.VncServer(pw_init_file=self.PW_INIT_FILE)
server.recoverVNCPassword()
self.assertEqual(server.password, 'abcdef')
def set_vnc_password_1_test(self):
import pyanaconda.vnc
server = pyanaconda.vnc.VncServer()
pyanaconda.vnc.iutil = mock.Mock()
pyanaconda.vnc.os.pipe.return_value = (1, 2)
server.setVNCPassword()
self.assertEqual(
pyanaconda.vnc.iutil.execWithRedirect.call_args_list,
[(('vncpasswd', ['-f']), {'stdin': 1, 'stdout': '/tmp/vncpassword'})])
def initialize_test(self):
import pyanaconda.vnc
IP = '192.168.0.21'
HOSTNAME = 'desktop'
dev = mock.Mock()
dev.get.return_value = 'eth0'
pyanaconda.vnc.network = mock.Mock()
pyanaconda.vnc.network.Network().netdevices = [dev]
pyanaconda.vnc.network.getActiveNetDevs.return_value = [0]
pyanaconda.vnc.network.getDefaultHostname.return_value = HOSTNAME
pyanaconda.vnc.isys = mock.Mock()
pyanaconda.vnc.isys.getIPAddresses = mock.Mock(return_value=[IP])
server = pyanaconda.vnc.VncServer(display=self.DISPLAY)
server.initialize()
expected = "%s:%s (%s)" % (HOSTNAME, self.DISPLAY, IP)
self.assertEqual(server.connxinfo, expected)
def openlogfile_test(self):
import pyanaconda.vnc
FILE = 'file'
pyanaconda.vnc.os.O_RDWR = os.O_RDWR
pyanaconda.vnc.os.O_CREAT = os.O_CREAT
pyanaconda.vnc.os.open.return_value = FILE
server = pyanaconda.vnc.VncServer(log_file=self.LOG_FILE)
ret = server.openlogfile()
self.assertEqual(ret, FILE)
self.assertEqual(pyanaconda.vnc.os.open.call_args,
((self.LOG_FILE, os.O_RDWR | os.O_CREAT), {})
)
def connect_to_view_test(self):
import pyanaconda.vnc
pyanaconda.vnc.subprocess.Popen().communicate.return_value = (self.OK, '')
server = pyanaconda.vnc.VncServer(vncconnecthost=self.VNCCONNECTHOST)
ret = server.connectToView()
self.assertTrue(ret)
params = pyanaconda.vnc.subprocess.Popen.call_args[0][0]
self.assertTrue(self.VNCCONNECTHOST in params)
self.assertTrue(params[params.index(self.VNCCONNECTHOST)-1] == "-connect")
def start_server_test(self):
import pyanaconda.vnc
pyanaconda.vnc.VncServer.initialize = mock.Mock()
pyanaconda.vnc.VncServer.setVNCPassword = mock.Mock()
pyanaconda.vnc.VncServer.VNCListen = mock.Mock()
pyanaconda.vnc.subprocess.Popen().poll.return_value = None
pyanaconda.vnc.os.environ = {}
pyanaconda.vnc.time.sleep = mock.Mock()
server = pyanaconda.vnc.VncServer(root=self.ROOT, display=self.DISPLAY,
desktop=self.DESKTOP, password=self.PASS, vncconnecthost="")
server.openlogfile = mock.Mock()
server.startServer()
params = pyanaconda.vnc.subprocess.Popen.call_args[0][0]
self.assertTrue('desktop=%s'%self.DESKTOP in params)
self.assertTrue(':%s'%self.DISPLAY in params)
self.assertTrue(pyanaconda.vnc.VncServer.VNCListen.called)
self.assertTrue("DISPLAY" in pyanaconda.vnc.os.environ)
self.assertEqual(pyanaconda.vnc.os.environ['DISPLAY'], ":%s" % self.DISPLAY)
|
