tests/metadata.TestCase

   1 #!/usr/bin/env python3
   2
   3 # http://www.drdobbs.com/testing/unit-testing-with-python/240165163
   4
   5 import inspect
   6 import optparse
   7 import os
   8 import sys
   9 import unittest
  10 import yaml
  11 import tempfile
  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.metadata
  21
  22
  23 class MetadataTest(unittest.TestCase):
  24     '''fdroidserver/metadata.py'''
  25
  26     def test_read_metadata(self):
  27
  28         def _build_yaml_representer(dumper, data):
  29             '''Creates a YAML representation of a Build instance'''
  30             return dumper.represent_dict(data)
  31
  32         testsdir = os.path.dirname(__file__)
  33         os.chdir(testsdir)
  34
  35         self.maxDiff = None
  36
  37         # these need to be set to prevent code running on None, only
  38         # 'accepted_formats' is actually used in metadata.py
  39         config = dict()
  40         config['sdk_path'] = '/opt/android-sdk'
  41         config['ndk_paths'] = dict()
  42         config['accepted_formats'] = ['json', 'txt', 'yml']
  43         fdroidserver.common.config = config
  44
  45         apps = fdroidserver.metadata.read_metadata(xref=True)
  46         for appid in ('org.smssecure.smssecure', 'org.adaway',
  47                       'org.videolan.vlc', 'com.politedroid'):
  48             savepath = os.path.join('metadata', 'dump', appid + '.yaml')
  49             frommeta = dict(apps[appid])
  50             self.assertTrue(appid in apps)
  51             with open(savepath, 'r') as f:
  52                 frompickle = yaml.load(f)
  53             self.assertEqual(frommeta, frompickle)
  54             # comment above assert and uncomment below to update test
  55             # files when new metadata fields are added
  56             # with open(savepath, 'w') as f:
  57             #     yaml.add_representer(fdroidserver.metadata.Build, _build_yaml_representer)
  58             #     yaml.dump(frommeta, f, default_flow_style=False)
  59
  60     def test_rewrite_yaml_fakeotaupdate(self):
  61
  62         # setup/reset test dir if necessary and setup params
  63         tmpdir = os.path.join(os.path.dirname(__file__), '..', '.testfiles')
  64         if not os.path.exists(tmpdir):
  65             os.makedirs(tmpdir)
  66         testdir = tempfile.mkdtemp(prefix='test_rewrite_metadata_', dir=tmpdir)
  67         fdroidserver.common.config = {'accepted_formats': ['txt', 'yml']}
  68
  69         # rewrite metadata
  70         allapps = fdroidserver.metadata.read_metadata(xref=True)
  71         for appid, app in allapps.items():
  72             if appid == 'fake.ota.update':
  73                 fdroidserver.metadata.write_metadata(os.path.join(testdir, appid + '.yml'), app)
  74
  75         # assert rewrite result
  76         with open(os.path.join(testdir, 'fake.ota.update.yml'), 'r', encoding='utf-8') as result:
  77             with open('metadata-rewrite-yml/fake.ota.update.yml', 'r', encoding='utf-8') as orig:
  78                 self.maxDiff = None
  79                 self.assertEqual(result.read(), orig.read())
  80
  81     def test_rewrite_yaml_fdroidclient(self):
  82
  83         # setup/reset test dir if necessary and setup params
  84         tmpdir = os.path.join(os.path.dirname(__file__), '..', '.testfiles')
  85         if not os.path.exists(tmpdir):
  86             os.makedirs(tmpdir)
  87         testdir = tempfile.mkdtemp(prefix='test_rewrite_metadata_', dir=tmpdir)
  88         fdroidserver.common.config = {'accepted_formats': ['txt', 'yml']}
  89
  90         # rewrite metadata
  91         allapps = fdroidserver.metadata.read_metadata(xref=True)
  92         for appid, app in allapps.items():
  93             if appid == 'org.fdroid.fdroid':
  94                 fdroidserver.metadata.write_metadata(os.path.join(testdir, appid + '.yml'), app)
  95
  96         # assert rewrite result
  97         with open(os.path.join(testdir, 'org.fdroid.fdroid.yml'), 'r', encoding='utf-8') as result:
  98             with open('metadata-rewrite-yml/org.fdroid.fdroid.yml', 'r', encoding='utf-8') as orig:
  99                 self.maxDiff = None
 100                 self.assertEqual(result.read(), orig.read())
 101
 102     def test_rewrite_yaml_special_build_params(self):
 103
 104         # setup/reset test dir if necessary and setup params
 105         tmpdir = os.path.join(os.path.dirname(__file__), '..', '.testfiles')
 106         if not os.path.exists(tmpdir):
 107             os.makedirs(tmpdir)
 108         testdir = tempfile.mkdtemp(prefix='test_rewrite_metadata_', dir=tmpdir)
 109         fdroidserver.common.config = {'accepted_formats': ['txt', 'yml']}
 110
 111         # rewrite metadata
 112         allapps = fdroidserver.metadata.read_metadata(xref=True)
 113         for appid, app in allapps.items():
 114             if appid == 'app.with.special.build.params':
 115                 fdroidserver.metadata.write_metadata(os.path.join(testdir, appid + '.yml'), app)
 116
 117         # assert rewrite result
 118         with open(os.path.join(testdir, 'app.with.special.build.params.yml'), 'r', encoding='utf-8') as result:
 119             with open('metadata-rewrite-yml/app.with.special.build.params.yml', 'r', encoding='utf-8') as orig:
 120                 self.maxDiff = None
 121                 self.assertEqual(result.read(), orig.read())
 122
 123
 124 if __name__ == "__main__":
 125     parser = optparse.OptionParser()
 126     parser.add_option("-v", "--verbose", action="store_true", default=False,
 127                       help="Spew out even more information than normal")
 128     (fdroidserver.common.options, args) = parser.parse_args(['--verbose'])
 129
 130     newSuite = unittest.TestSuite()
 131     newSuite.addTest(unittest.makeSuite(MetadataTest))
 132     unittest.main()