3 # http://www.drdobbs.com/testing/unit-testing-with-python/240165163
14 from zipfile import ZipFile
17 localmodule = os.path.realpath(
18 os.path.join(os.path.dirname(inspect.getfile(inspect.currentframe())), '..'))
19 print('localmodule: ' + localmodule)
20 if localmodule not in sys.path:
21 sys.path.insert(0, localmodule)
23 import fdroidserver.signindex
24 import fdroidserver.common
25 import fdroidserver.metadata
28 class CommonTest(unittest.TestCase):
29 '''fdroidserver/common.py'''
31 def _set_build_tools(self):
32 build_tools = os.path.join(fdroidserver.common.config['sdk_path'], 'build-tools')
33 if os.path.exists(build_tools):
34 fdroidserver.common.config['build_tools'] = ''
35 for f in sorted(os.listdir(build_tools), reverse=True):
36 versioned = os.path.join(build_tools, f)
37 if os.path.isdir(versioned) \
38 and os.path.isfile(os.path.join(versioned, 'aapt')):
39 fdroidserver.common.config['build_tools'] = versioned
43 print('no build-tools found: ' + build_tools)
47 for cmd in ('aapt', 'adb', 'android', 'zipalign'):
48 path = fdroidserver.common.find_sdk_tools_cmd(cmd)
50 self.assertTrue(os.path.exists(path))
51 self.assertTrue(os.path.isfile(path))
53 def test_find_sdk_tools_cmd(self):
54 fdroidserver.common.config = dict()
55 # TODO add this once everything works without sdk_path set in config
57 sdk_path = os.getenv('ANDROID_HOME')
58 if os.path.exists(sdk_path):
59 fdroidserver.common.config['sdk_path'] = sdk_path
60 if os.path.exists('/usr/bin/aapt'):
61 # this test only works when /usr/bin/aapt is installed
63 build_tools = os.path.join(sdk_path, 'build-tools')
64 if self._set_build_tools():
67 print('no build-tools found: ' + build_tools)
69 def testIsApkDebuggable(self):
71 fdroidserver.common.fill_config_defaults(config)
72 fdroidserver.common.config = config
73 self._set_build_tools()
74 config['aapt'] = fdroidserver.common.find_sdk_tools_cmd('aapt')
75 # these are set debuggable
77 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip.apk'))
78 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-badsig.apk'))
79 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-badcert.apk'))
80 for apkfile in testfiles:
81 debuggable = fdroidserver.common.isApkAndDebuggable(apkfile)
82 self.assertTrue(debuggable,
83 "debuggable APK state was not properly parsed!")
84 # these are set NOT debuggable
86 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-release.apk'))
87 testfiles.append(os.path.join(os.path.dirname(__file__), 'urzip-release-unsigned.apk'))
88 for apkfile in testfiles:
89 debuggable = fdroidserver.common.isApkAndDebuggable(apkfile)
90 self.assertFalse(debuggable,
91 "debuggable APK state was not properly parsed!")
93 def testPackageNameValidity(self):
94 for name in ["org.fdroid.fdroid",
95 "org.f_droid.fdr0ID"]:
96 self.assertTrue(fdroidserver.common.is_valid_package_name(name),
97 "{0} should be a valid package name".format(name))
98 for name in ["0rg.fdroid.fdroid",
101 "/org.fdroid.fdroid"]:
102 self.assertFalse(fdroidserver.common.is_valid_package_name(name),
103 "{0} should not be a valid package name".format(name))
105 def test_prepare_sources(self):
107 teststr = 'FAKE_STR_FOR_TESTING'
109 tmpdir = os.path.join(os.path.dirname(__file__), '..', '.testfiles')
110 if not os.path.exists(tmpdir):
112 tmptestsdir = tempfile.mkdtemp(prefix='test_prepare_sources', dir=tmpdir)
113 shutil.copytree(os.path.join(os.path.dirname(__file__), 'source-files'),
114 os.path.join(tmptestsdir, 'source-files'))
116 testdir = os.path.join(tmptestsdir, 'source-files', 'fdroid', 'fdroidclient')
119 config['sdk_path'] = os.getenv('ANDROID_HOME')
120 config['ndk_paths'] = {'r10d': os.getenv('ANDROID_NDK_HOME')}
121 config['build_tools'] = 'FAKE_BUILD_TOOLS_VERSION'
122 fdroidserver.common.config = config
123 app = fdroidserver.metadata.App()
124 app.id = 'org.fdroid.froid'
125 build = fdroidserver.metadata.Build()
126 build.commit = 'master'
127 build.forceversion = True
128 build.forcevercode = True
129 build.gradle = ['yes']
130 build.target = 'android-' + str(testint)
131 build.versionName = teststr
132 build.versionCode = testint
135 # no need to change to the correct commit here
136 def gotorevision(self, rev, refresh=True):
139 # no srclib info needed, but it could be added...
143 fdroidserver.common.prepare_source(FakeVcs(), app, build, testdir, testdir, testdir)
145 with open(os.path.join(testdir, 'build.gradle'), 'r') as f:
147 self.assertIsNotNone(re.search("\s+compileSdkVersion %s\s+" % testint, filedata))
149 with open(os.path.join(testdir, 'AndroidManifest.xml')) as f:
151 self.assertIsNone(re.search('android:debuggable', filedata))
152 self.assertIsNotNone(re.search('android:versionName="%s"' % build.versionName, filedata))
153 self.assertIsNotNone(re.search('android:versionCode="%s"' % build.versionCode, filedata))
155 def test_fdroid_popen_stderr_redirect(self):
156 commands = ['sh', '-c', 'echo stdout message && echo stderr message 1>&2']
158 p = fdroidserver.common.FDroidPopen(commands)
159 self.assertEqual(p.output, 'stdout message\nstderr message\n')
161 p = fdroidserver.common.FDroidPopen(commands, stderr_to_stdout=False)
162 self.assertEqual(p.output, 'stdout message\n')
164 def test_signjar(self):
165 fdroidserver.common.config = None
166 config = fdroidserver.common.read_config(fdroidserver.common.options)
167 config['jarsigner'] = fdroidserver.common.find_sdk_tools_cmd('jarsigner')
168 fdroidserver.common.config = config
169 fdroidserver.signindex.config = config
171 basedir = os.path.dirname(__file__)
172 tmpdir = os.path.join(basedir, '..', '.testfiles')
173 if not os.path.exists(tmpdir):
175 sourcedir = os.path.join(basedir, 'signindex')
176 testsdir = tempfile.mkdtemp(prefix='test_signjar', dir=tmpdir)
177 for f in ('testy.jar', 'guardianproject.jar',):
178 sourcefile = os.path.join(sourcedir, f)
179 testfile = os.path.join(testsdir, f)
180 shutil.copy(sourcefile, testsdir)
181 fdroidserver.signindex.sign_jar(testfile)
182 # these should be resigned, and therefore different
183 self.assertNotEqual(open(sourcefile, 'rb').read(), open(testfile, 'rb').read())
185 def test_verify_apk_signature(self):
186 fdroidserver.common.config = None
187 config = fdroidserver.common.read_config(fdroidserver.common.options)
188 config['jarsigner'] = fdroidserver.common.find_sdk_tools_cmd('jarsigner')
189 fdroidserver.common.config = config
191 self.assertTrue(fdroidserver.common.verify_apk_signature('urzip.apk'))
192 self.assertFalse(fdroidserver.common.verify_apk_signature('urzip-badcert.apk'))
193 self.assertFalse(fdroidserver.common.verify_apk_signature('urzip-badsig.apk'))
194 self.assertTrue(fdroidserver.common.verify_apk_signature('urzip-release.apk'))
195 self.assertFalse(fdroidserver.common.verify_apk_signature('urzip-release-unsigned.apk'))
197 def test_verify_apks(self):
198 fdroidserver.common.config = None
199 config = fdroidserver.common.read_config(fdroidserver.common.options)
200 config['jarsigner'] = fdroidserver.common.find_sdk_tools_cmd('jarsigner')
201 fdroidserver.common.config = config
203 basedir = os.path.dirname(__file__)
204 sourceapk = os.path.join(basedir, 'urzip.apk')
206 tmpdir = os.path.join(basedir, '..', '.testfiles')
207 if not os.path.exists(tmpdir):
209 testdir = tempfile.mkdtemp(prefix='test_verify_apks', dir=tmpdir)
210 print('testdir', testdir)
212 copyapk = os.path.join(testdir, 'urzip-copy.apk')
213 shutil.copy(sourceapk, copyapk)
214 self.assertTrue(fdroidserver.common.verify_apk_signature(copyapk))
215 self.assertIsNone(fdroidserver.common.verify_apks(sourceapk, copyapk, tmpdir))
217 unsignedapk = os.path.join(testdir, 'urzip-unsigned.apk')
218 with ZipFile(sourceapk, 'r') as apk:
219 with ZipFile(unsignedapk, 'w') as testapk:
220 for info in apk.infolist():
221 if not info.filename.startswith('META-INF/'):
222 testapk.writestr(info, apk.read(info.filename))
223 self.assertIsNone(fdroidserver.common.verify_apks(sourceapk, unsignedapk, tmpdir))
225 twosigapk = os.path.join(testdir, 'urzip-twosig.apk')
226 otherapk = ZipFile(os.path.join(basedir, 'urzip-release.apk'), 'r')
227 with ZipFile(sourceapk, 'r') as apk:
228 with ZipFile(twosigapk, 'w') as testapk:
229 for info in apk.infolist():
230 testapk.writestr(info, apk.read(info.filename))
231 if info.filename.startswith('META-INF/'):
232 testapk.writestr(info, otherapk.read(info.filename))
234 self.assertFalse(fdroidserver.common.verify_apk_signature(twosigapk))
235 self.assertIsNone(fdroidserver.common.verify_apks(sourceapk, twosigapk, tmpdir))
237 def test_write_to_config(self):
238 with tempfile.TemporaryDirectory() as tmpPath:
239 cfgPath = os.path.join(tmpPath, 'config.py')
240 with open(cfgPath, 'w', encoding='utf-8') as f:
241 f.write(textwrap.dedent("""\
243 # test = 'example value'
247 do_not_touch = "good value"
250 key="123" # inline"""))
252 cfg = {'key': '111', 'default_me_orig': 'orig'}
253 fdroidserver.common.write_to_config(cfg, 'key', config_file=cfgPath)
254 fdroidserver.common.write_to_config(cfg, 'default_me', config_file=cfgPath)
255 fdroidserver.common.write_to_config(cfg, 'test', value='test value', config_file=cfgPath)
256 fdroidserver.common.write_to_config(cfg, 'new_key', value='new', config_file=cfgPath)
258 with open(cfgPath, 'r', encoding='utf-8') as f:
259 self.assertEqual(f.read(), textwrap.dedent("""\
265 do_not_touch = "good value"
272 def test_write_to_config_when_empty(self):
273 with tempfile.TemporaryDirectory() as tmpPath:
274 cfgPath = os.path.join(tmpPath, 'config.py')
275 with open(cfgPath, 'w') as f:
277 fdroidserver.common.write_to_config({}, 'key', 'val', cfgPath)
278 with open(cfgPath, 'r', encoding='utf-8') as f:
279 self.assertEqual(f.read(), textwrap.dedent("""\
285 if __name__ == "__main__":
286 parser = optparse.OptionParser()
287 parser.add_option("-v", "--verbose", action="store_true", default=False,
288 help="Spew out even more information than normal")
289 (fdroidserver.common.options, args) = parser.parse_args(['--verbose'])
291 newSuite = unittest.TestSuite()
292 newSuite.addTest(unittest.makeSuite(CommonTest))