tests/update.TestCase

   1 #!/usr/bin/env python3
   2
   3 # http://www.drdobbs.com/testing/unit-testing-with-python/240165163
   4
   5 import inspect
   6 import logging
   7 import optparse
   8 import os
   9 import sys
  10 import unittest
  11 from binascii import unhexlify
  12
  13 localmodule = os.path.realpath(
  14     os.path.join(os.path.dirname(inspect.getfile(inspect.currentframe())), '..'))
  15 print('localmodule: ' + localmodule)
  16 if localmodule not in sys.path:
  17     sys.path.insert(0, localmodule)
  18
  19 import fdroidserver.common
  20 import fdroidserver.update
  21 from fdroidserver.common import FDroidPopen
  22
  23
  24 class UpdateTest(unittest.TestCase):
  25     '''fdroid update'''
  26
  27     def javagetsig(self, apkfile):
  28         getsig_dir = os.path.join(os.path.dirname(__file__), 'getsig')
  29         if not os.path.exists(getsig_dir + "/getsig.class"):
  30             logging.critical("getsig.class not found. To fix: cd '%s' && ./make.sh" % getsig_dir)
  31             sys.exit(1)
  32         # FDroidPopen needs some config to work
  33         config = dict()
  34         fdroidserver.common.fill_config_defaults(config)
  35         fdroidserver.common.config = config
  36         p = FDroidPopen(['java', '-cp', os.path.join(os.path.dirname(__file__), 'getsig'),
  37                          'getsig', os.path.join(os.getcwd(), apkfile)])
  38         sig = None
  39         for line in p.output.splitlines():
  40             if line.startswith('Result:'):
  41                 sig = line[7:].strip()
  42                 break
  43         if p.returncode == 0:
  44             return sig
  45         else:
  46             return None
  47
  48     def testGoodGetsig(self):
  49         # config needed to use jarsigner and keytool
  50         config = dict()
  51         fdroidserver.common.fill_config_defaults(config)
  52         fdroidserver.update.config = config
  53         apkfile = os.path.join(os.path.dirname(__file__), 'urzip.apk')
  54         sig = self.javagetsig(apkfile)
  55         self.assertIsNotNone(sig, "sig is None")
  56         pysig = fdroidserver.update.getsig(apkfile)
  57         self.assertIsNotNone(pysig, "pysig is None")
  58         self.assertEqual(sig, fdroidserver.update.getsig(apkfile),
  59                          "python sig not equal to java sig!")
  60         self.assertEqual(len(sig), len(pysig),
  61                          "the length of the two sigs are different!")
  62         try:
  63             self.assertEqual(unhexlify(sig), unhexlify(pysig),
  64                              "the length of the two sigs are different!")
  65         except TypeError as e:
  66             print(e)
  67             self.assertTrue(False, 'TypeError!')
  68
  69     def testBadGetsig(self):
  70         # config needed to use jarsigner and keytool
  71         config = dict()
  72         fdroidserver.common.fill_config_defaults(config)
  73         fdroidserver.update.config = config
  74         apkfile = os.path.join(os.path.dirname(__file__), 'urzip-badsig.apk')
  75         sig = self.javagetsig(apkfile)
  76         self.assertIsNone(sig, "sig should be None: " + str(sig))
  77         pysig = fdroidserver.update.getsig(apkfile)
  78         self.assertIsNone(pysig, "python sig should be None: " + str(sig))
  79
  80         apkfile = os.path.join(os.path.dirname(__file__), 'urzip-badcert.apk')
  81         sig = self.javagetsig(apkfile)
  82         self.assertIsNone(sig, "sig should be None: " + str(sig))
  83         pysig = fdroidserver.update.getsig(apkfile)
  84         self.assertIsNone(pysig, "python sig should be None: " + str(sig))
  85
  86     def testScanApksAndObbs(self):
  87         os.chdir(os.path.dirname(__file__))
  88         if os.path.basename(os.getcwd()) != 'tests':
  89             raise Exception('This test must be run in the "tests/" subdir')
  90
  91         config = dict()
  92         fdroidserver.common.fill_config_defaults(config)
  93         config['ndk_paths'] = dict()
  94         config['accepted_formats'] = ['json', 'txt', 'xml', 'yml']
  95         fdroidserver.common.config = config
  96         fdroidserver.update.config = config
  97
  98         fdroidserver.update.options = type('', (), {})()
  99         fdroidserver.update.options.clean = True
 100         fdroidserver.update.options.delete_unknown = True
 101
 102         apps = fdroidserver.metadata.read_metadata(xref=True)
 103         knownapks = fdroidserver.common.KnownApks()
 104         apks, cachechanged = fdroidserver.update.scan_apks(apps, {}, 'repo', knownapks, False)
 105         self.assertEqual(len(apks), 6)
 106         apk = apks[0]
 107         self.assertEqual(apk['minSdkVersion'], '4')
 108         self.assertEqual(apk['targetSdkVersion'], '18')
 109         self.assertFalse('maxSdkVersion' in apk)
 110
 111         fdroidserver.update.insert_obbs('repo', apps, apks)
 112         for apk in apks:
 113             if apk['id'] == 'obb.mainpatch.current':
 114                 self.assertEqual(apk.get('obbMainFile'), 'main.1619.obb.mainpatch.current.obb')
 115                 self.assertEqual(apk.get('obbPatchFile'), 'patch.1619.obb.mainpatch.current.obb')
 116             elif apk['id'] == 'obb.main.oldversion':
 117                 self.assertEqual(apk.get('obbMainFile'), 'main.1434483388.obb.main.oldversion.obb')
 118                 self.assertIsNone(apk.get('obbPatchFile'))
 119             elif apk['id'] == 'obb.main.twoversions':
 120                 self.assertIsNone(apk.get('obbPatchFile'))
 121                 if apk['versioncode'] == 1101613:
 122                     self.assertEqual(apk.get('obbMainFile'), 'main.1101613.obb.main.twoversions.obb')
 123                 elif apk['versioncode'] == 1101615:
 124                     self.assertEqual(apk.get('obbMainFile'), 'main.1101615.obb.main.twoversions.obb')
 125                 elif apk['versioncode'] == 1101617:
 126                     self.assertEqual(apk.get('obbMainFile'), 'main.1101615.obb.main.twoversions.obb')
 127                 else:
 128                     self.assertTrue(False)
 129             elif apk['id'] == 'info.guardianproject.urzip':
 130                 self.assertIsNone(apk.get('obbMainFile'))
 131                 self.assertIsNone(apk.get('obbPatchFile'))
 132
 133
 134 if __name__ == "__main__":
 135     parser = optparse.OptionParser()
 136     parser.add_option("-v", "--verbose", action="store_true", default=False,
 137                       help="Spew out even more information than normal")
 138     (fdroidserver.common.options, args) = parser.parse_args(['--verbose'])
 139
 140     newSuite = unittest.TestSuite()
 141     newSuite.addTest(unittest.makeSuite(UpdateTest))
 142     unittest.main()