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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
|
"""
Info:
This tests DeviceManagerADB with a real device
Requirements:
- You must have a device connected
- It should be listed under 'adb devices'
Notes:
- Not all functions have been covered.
In particular, functions from the parent class
- No testing of properties is done
- The test case are very simple and it could be
done with deeper inspection of the return values
Author(s):
- Armen Zambrano <armenzg@mozilla.com>
Functions that are not being tested:
- launchProcess - DEPRECATED
- getIP
- recordLogcat
- saveScreenshot
- validateDir
- mkDirs
- getDeviceRoot
- shellCheckOutput
- processExist
I assume these functions are only useful for Android
- getAppRoot()
- updateApp()
- uninstallApp()
- uninstallAppAndReboot()
"""
import os
import re
import socket
import sys
import tempfile
import unittest
from StringIO import StringIO
from mozdevice import DeviceManagerADB, DMError
def find_mount_permissions(dm, mount_path):
for mount_point in dm._runCmd(["shell", "mount"]).output:
if mount_point.find(mount_path) > 0:
return re.search('(ro|rw)(?=,)', mount_point).group(0)
class DeviceManagerADBTestCase(unittest.TestCase):
tempLocalDir = "tempDir"
tempLocalFile = os.path.join(tempLocalDir, "tempfile.txt")
tempRemoteDir = None
tempRemoteFile = None
tempRemoteSystemFile = None
def setUp(self):
self.assertTrue(find_mount_permissions(self.dm, "/system"), "ro")
self.assertTrue(os.path.exists(self.tempLocalDir))
self.assertTrue(os.path.exists(self.tempLocalFile))
if self.dm.fileExists(self.tempRemoteFile):
self.dm.removeFile(self.tempRemoteFile)
self.assertFalse(self.dm.fileExists(self.tempRemoteFile))
if self.dm.fileExists(self.tempRemoteSystemFile):
self.dm.removeFile(self.tempRemoteSystemFile)
self.assertTrue(self.dm.dirExists(self.tempRemoteDir))
@classmethod
def setUpClass(self):
self.dm = DeviceManagerADB()
if not os.path.exists(self.tempLocalDir):
os.mkdir(self.tempLocalDir)
if not os.path.exists(self.tempLocalFile):
# Create empty file
open(self.tempLocalFile, 'w').close()
self.tempRemoteDir = self.dm.getTempDir()
self.tempRemoteFile = os.path.join(self.tempRemoteDir,
os.path.basename(self.tempLocalFile))
self.tempRemoteSystemFile = \
os.path.join("/system", os.path.basename(self.tempLocalFile))
@classmethod
def tearDownClass(self):
os.remove(self.tempLocalFile)
os.rmdir(self.tempLocalDir)
if self.dm.dirExists(self.tempRemoteDir):
# self.tempRemoteFile will get deleted with it
self.dm.removeDir(self.tempRemoteDir)
if self.dm.fileExists(self.tempRemoteSystemFile):
self.dm.removeFile(self.tempRemoteSystemFile)
class TestFileOperations(DeviceManagerADBTestCase):
def test_make_and_remove_directory(self):
dir1 = os.path.join(self.tempRemoteDir, "dir1")
self.assertFalse(self.dm.dirExists(dir1))
self.dm.mkDir(dir1)
self.assertTrue(self.dm.dirExists(dir1))
self.dm.removeDir(dir1)
self.assertFalse(self.dm.dirExists(dir1))
def test_push_and_remove_file(self):
self.dm.pushFile(self.tempLocalFile, self.tempRemoteFile)
self.assertTrue(self.dm.fileExists(self.tempRemoteFile))
self.dm.removeFile(self.tempRemoteFile)
self.assertFalse(self.dm.fileExists(self.tempRemoteFile))
def test_push_and_pull_file(self):
self.dm.pushFile(self.tempLocalFile, self.tempRemoteFile)
self.assertTrue(self.dm.fileExists(self.tempRemoteFile))
self.assertFalse(os.path.exists("pulled.txt"))
self.dm.getFile(self.tempRemoteFile, "pulled.txt")
self.assertTrue(os.path.exists("pulled.txt"))
os.remove("pulled.txt")
def test_push_and_pull_directory_and_list_files(self):
self.dm.removeDir(self.tempRemoteDir)
self.assertFalse(self.dm.dirExists(self.tempRemoteDir))
self.dm.pushDir(self.tempLocalDir, self.tempRemoteDir)
self.assertTrue(self.dm.dirExists(self.tempRemoteDir))
response = self.dm.listFiles(self.tempRemoteDir)
# The local dir that was pushed contains the tempLocalFile
self.assertIn(os.path.basename(self.tempLocalFile), response)
# Create a temp dir to pull to
temp_dir = tempfile.mkdtemp()
self.assertTrue(os.path.exists(temp_dir))
self.dm.getDirectory(self.tempRemoteDir, temp_dir)
self.assertTrue(os.path.exists(self.tempLocalFile))
def test_move_and_remove_directories(self):
dir1 = os.path.join(self.tempRemoteDir, "dir1")
dir2 = os.path.join(self.tempRemoteDir, "dir2")
self.assertFalse(self.dm.dirExists(dir1))
self.dm.mkDir(dir1)
self.assertTrue(self.dm.dirExists(dir1))
self.assertFalse(self.dm.dirExists(dir2))
self.dm.moveTree(dir1, dir2)
self.assertTrue(self.dm.dirExists(dir2))
self.dm.removeDir(dir1)
self.dm.removeDir(dir2)
self.assertFalse(self.dm.dirExists(dir1))
self.assertFalse(self.dm.dirExists(dir2))
def test_push_and_remove_system_file(self):
out = StringIO()
self.assertTrue(find_mount_permissions(self.dm, "/system") == "ro")
self.assertFalse(self.dm.fileExists(self.tempRemoteSystemFile))
self.assertRaises(DMError, self.dm.pushFile, self.tempLocalFile, self.tempRemoteSystemFile)
self.dm.shell(['mount', '-w', '-o', 'remount', '/system'], out)
self.assertTrue(find_mount_permissions(self.dm, "/system") == "rw")
self.assertFalse(self.dm.fileExists(self.tempRemoteSystemFile))
self.dm.pushFile(self.tempLocalFile, self.tempRemoteSystemFile)
self.assertTrue(self.dm.fileExists(self.tempRemoteSystemFile))
self.dm.removeFile(self.tempRemoteSystemFile)
self.assertFalse(self.dm.fileExists(self.tempRemoteSystemFile))
self.dm.shell(['mount', '-r', '-o', 'remount', '/system'], out)
out.close()
self.assertTrue(find_mount_permissions(self.dm, "/system") == "ro")
class TestOther(DeviceManagerADBTestCase):
def test_get_list_of_processes(self):
self.assertEquals(type(self.dm.getProcessList()), list)
def test_get_current_time(self):
self.assertEquals(type(self.dm.getCurrentTime()), int)
def test_get_info(self):
self.assertEquals(type(self.dm.getInfo()), dict)
def test_list_devices(self):
self.assertEquals(len(list(self.dm.devices())), 1)
def test_shell(self):
out = StringIO()
self.dm.shell(["echo", "$COMPANY", ";", "pwd"], out,
env={"COMPANY": "Mozilla"}, cwd="/", timeout=4, root=True)
output = str(out.getvalue()).rstrip().splitlines()
out.close()
self.assertEquals(output, ['Mozilla', '/'])
def test_port_forwarding(self):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(("", 0))
port = s.getsockname()[1]
s.close()
# If successful then no exception is raised
self.dm.forward("tcp:%s" % port, "tcp:2828")
def test_port_forwarding_error(self):
self.assertRaises(DMError, self.dm.forward, "", "")
if __name__ == '__main__':
dm = DeviceManagerADB()
if not dm.devices():
print "There are no connected adb devices"
sys.exit(1)
if find_mount_permissions(dm, "/system") == "rw":
print "We've found out that /system is mounted as 'rw'. This is because the command " \
"'adb remount' has been run before running this test case. Please reboot the device " \
"and try again."
sys.exit(1)
unittest.main()
|
