3 # http://www.drdobbs.com/testing/unit-testing-with-python/240165163
13 from zipfile import ZipFile
15 localmodule = os.path.realpath(
16 os.path.join(os.path.dirname(inspect.getfile(inspect.currentframe())), '..'))
17 print('localmodule: ' + localmodule)
18 if localmodule not in sys.path:
19 sys.path.insert(0, localmodule)
21 import fdroidserver.common
22 import fdroidserver.metadata
25 class CommonTest(unittest.TestCase):
26 '''fdroidserver/common.py'''
28 def _set_build_tools(self):
29 build_tools = os.path.join(fdroidserver.common.config['sdk_path'], 'build-tools')
30 if os.path.exists(build_tools):
31 fdroidserver.common.config['build_tools'] = ''
32 for f in sorted(os.listdir(build_tools), reverse=True):
33 versioned = os.path.join(build_tools, f)
34 if os.path.isdir(versioned) \
35 and os.path.isfile(os.path.join(versioned, 'aapt')):
36 fdroidserver.common.config['build_tools'] = versioned
40 print('no build-tools found: ' + build_tools)
44 for cmd in ('aapt', 'adb', 'android', 'zipalign'):
45 path = fdroidserver.common.find_sdk_tools_cmd(cmd)
47 self.assertTrue(os.path.exists(path))
48 self.assertTrue(os.path.isfile(path))
50 def test_find_sdk_tools_cmd(self):
51 fdroidserver.common.config = dict()
52 # TODO add this once everything works without sdk_path set in config
54 sdk_path = os.getenv('ANDROID_HOME')
55 if os.path.exists(sdk_path):
56 fdroidserver.common.config['sdk_path'] = sdk_path
57 if os.path.exists('/usr/bin/aapt'):
58 # this test only works when /usr/bin/aapt is installed
60 build_tools = os.path.join(sdk_path, 'build-tools')
61 if self._set_build_tools():
64 print('no build-tools found: ' + build_tools)
66 def testIsApkDebuggable(self):
68 fdroidserver.common.fill_config_defaults(config)
69 fdroidserver.common.config = config
70 self._set_build_tools()
71 config['aapt'] = fdroidserver.common.find_sdk_tools_cmd('aapt')
72 # these are set debuggable
74 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip.apk'))
75 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-badsig.apk'))
76 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-badcert.apk'))
77 for apkfile in testfiles:
78 debuggable = fdroidserver.common.isApkAndDebuggable(apkfile, config)
79 self.assertTrue(debuggable,
80 "debuggable APK state was not properly parsed!")
81 # these are set NOT debuggable
83 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-release.apk'))
84 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-release-unsigned.apk'))
85 for apkfile in testfiles:
86 debuggable = fdroidserver.common.isApkAndDebuggable(apkfile, config)
87 self.assertFalse(debuggable,
88 "debuggable APK state was not properly parsed!")
90 def testPackageNameValidity(self):
91 for name in ["org.fdroid.fdroid",
92 "org.f_droid.fdr0ID"]:
93 self.assertTrue(fdroidserver.common.is_valid_package_name(name),
94 "{0} should be a valid package name".format(name))
95 for name in ["0rg.fdroid.fdroid",
98 "/org.fdroid.fdroid"]:
99 self.assertFalse(fdroidserver.common.is_valid_package_name(name),
100 "{0} should not be a valid package name".format(name))
102 def test_prepare_sources(self):
104 teststr = 'FAKE_STR_FOR_TESTING'
106 tmpdir = os.path.join(os.path.dirname(__file__), '..', '.testfiles')
107 if not os.path.exists(tmpdir):
109 tmptestsdir = tempfile.mkdtemp(prefix='test_prepare_sources', dir=tmpdir)
110 shutil.copytree(os.path.join(os.path.dirname(__file__), 'source-files'),
111 os.path.join(tmptestsdir, 'source-files'))
113 testdir = os.path.join(tmptestsdir, 'source-files', 'fdroid', 'fdroidclient')
116 config['sdk_path'] = os.getenv('ANDROID_HOME')
117 config['ndk_paths'] = {'r10d': os.getenv('ANDROID_NDK_HOME')}
118 config['build_tools'] = 'FAKE_BUILD_TOOLS_VERSION'
119 fdroidserver.common.config = config
120 app = fdroidserver.metadata.App()
121 app.id = 'org.fdroid.froid'
122 build = fdroidserver.metadata.Build()
123 build.commit = 'master'
124 build.forceversion = True
125 build.forcevercode = True
126 build.gradle = ['yes']
127 build.target = 'android-' + str(testint)
128 build.versionName = teststr
129 build.versionCode = testint
132 # no need to change to the correct commit here
133 def gotorevision(self, rev, refresh=True):
136 # no srclib info needed, but it could be added...
140 fdroidserver.common.prepare_source(FakeVcs(), app, build, testdir, testdir, testdir)
142 with open(os.path.join(testdir, 'build.gradle'), 'r') as f:
144 self.assertIsNotNone(re.search("\s+compileSdkVersion %s\s+" % testint, filedata))
146 with open(os.path.join(testdir, 'AndroidManifest.xml')) as f:
148 self.assertIsNone(re.search('android:debuggable', filedata))
149 self.assertIsNotNone(re.search('android:versionName="%s"' % build.versionName, filedata))
150 self.assertIsNotNone(re.search('android:versionCode="%s"' % build.versionCode, filedata))
152 def test_fdroid_popen_stderr_redirect(self):
153 commands = ['sh', '-c', 'echo stdout message && echo stderr message 1>&2']
155 p = fdroidserver.common.FDroidPopen(commands)
156 self.assertEqual(p.output, 'stdout message\nstderr message\n')
158 p = fdroidserver.common.FDroidPopen(commands, stderr_to_stdout=False)
159 self.assertEqual(p.output, 'stdout message\n')
161 def test_signjar(self):
162 fdroidserver.common.config = None
163 config = fdroidserver.common.read_config(fdroidserver.common.options)
164 config['jarsigner'] = fdroidserver.common.find_sdk_tools_cmd('jarsigner')
165 fdroidserver.common.config = config
167 basedir = os.path.dirname(__file__)
168 tmpdir = os.path.join(basedir, '..', '.testfiles')
169 if not os.path.exists(tmpdir):
171 sourcedir = os.path.join(basedir, 'signindex')
172 testsdir = tempfile.mkdtemp(prefix='test_signjar', dir=tmpdir)
173 for f in ('testy.jar', 'guardianproject.jar',):
174 sourcefile = os.path.join(sourcedir, f)
175 testfile = os.path.join(testsdir, f)
176 shutil.copy(sourcefile, testsdir)
177 fdroidserver.common.signjar(testfile)
178 # these should be resigned, and therefore different
179 self.assertNotEqual(open(sourcefile, 'rb').read(), open(testfile, 'rb').read())
181 def test_verify_apk_signature(self):
182 fdroidserver.common.config = None
183 config = fdroidserver.common.read_config(fdroidserver.common.options)
184 config['jarsigner'] = fdroidserver.common.find_sdk_tools_cmd('jarsigner')
185 fdroidserver.common.config = config
187 self.assertTrue(fdroidserver.common.verify_apk_signature('urzip.apk'))
188 self.assertFalse(fdroidserver.common.verify_apk_signature('urzip-badcert.apk'))
189 self.assertFalse(fdroidserver.common.verify_apk_signature('urzip-badsig.apk'))
190 self.assertTrue(fdroidserver.common.verify_apk_signature('urzip-release.apk'))
191 self.assertFalse(fdroidserver.common.verify_apk_signature('urzip-release-unsigned.apk'))
193 def test_verify_apks(self):
194 fdroidserver.common.config = None
195 config = fdroidserver.common.read_config(fdroidserver.common.options)
196 config['jarsigner'] = fdroidserver.common.find_sdk_tools_cmd('jarsigner')
197 fdroidserver.common.config = config
199 basedir = os.path.dirname(__file__)
200 sourceapk = os.path.join(basedir, 'urzip.apk')
202 tmpdir = os.path.join(basedir, '..', '.testfiles')
203 if not os.path.exists(tmpdir):
205 testdir = tempfile.mkdtemp(prefix='test_verify_apks', dir=tmpdir)
206 print('testdir', testdir)
208 copyapk = os.path.join(testdir, 'urzip-copy.apk')
209 shutil.copy(sourceapk, copyapk)
210 self.assertTrue(fdroidserver.common.verify_apk_signature(copyapk))
211 self.assertIsNone(fdroidserver.common.verify_apks(sourceapk, copyapk, tmpdir))
213 unsignedapk = os.path.join(testdir, 'urzip-unsigned.apk')
214 with ZipFile(sourceapk, 'r') as apk:
215 with ZipFile(unsignedapk, 'w') as testapk:
216 for info in apk.infolist():
217 if not info.filename.startswith('META-INF/'):
218 testapk.writestr(info, apk.read(info.filename))
219 self.assertIsNone(fdroidserver.common.verify_apks(sourceapk, unsignedapk, tmpdir))
221 twosigapk = os.path.join(testdir, 'urzip-twosig.apk')
222 otherapk = ZipFile(os.path.join(basedir, 'urzip-release.apk'), 'r')
223 with ZipFile(sourceapk, 'r') as apk:
224 with ZipFile(twosigapk, 'w') as testapk:
225 for info in apk.infolist():
226 testapk.writestr(info, apk.read(info.filename))
227 if info.filename.startswith('META-INF/'):
228 testapk.writestr(info, otherapk.read(info.filename))
230 self.assertFalse(fdroidserver.common.verify_apk_signature(twosigapk))
231 self.assertIsNone(fdroidserver.common.verify_apks(sourceapk, twosigapk, tmpdir))
234 if __name__ == "__main__":
235 parser = optparse.OptionParser()
236 parser.add_option("-v", "--verbose", action="store_true", default=False,
237 help="Spew out even more information than normal")
238 (fdroidserver.common.options, args) = parser.parse_args(['--verbose'])
240 newSuite = unittest.TestSuite()
241 newSuite.addTest(unittest.makeSuite(CommonTest))