|  | #!/usr/bin/env python | 
|  | # | 
|  | # Copyright (C) 2018 The Android Open Source Project | 
|  | # | 
|  | # Licensed under the Apache License, Version 2.0 (the "License"); | 
|  | # you may not use this file except in compliance with the License. | 
|  | # You may obtain a copy of the License at | 
|  | # | 
|  | #      http://www.apache.org/licenses/LICENSE-2.0 | 
|  | # | 
|  | # Unless required by applicable law or agreed to in writing, software | 
|  | # distributed under the License is distributed on an "AS IS" BASIS, | 
|  | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | # See the License for the specific language governing permissions and | 
|  | # limitations under the License. | 
|  | # | 
|  | """Unit tests for manifest_fixer.py.""" | 
|  |  | 
|  | import sys | 
|  | import unittest | 
|  | from xml.dom import minidom | 
|  |  | 
|  | import manifest_check | 
|  |  | 
|  | sys.dont_write_bytecode = True | 
|  |  | 
|  |  | 
|  | def uses_library_xml(name, attr=''): | 
|  | return '<uses-library android:name="%s"%s />' % (name, attr) | 
|  |  | 
|  |  | 
|  | def required_xml(value): | 
|  | return ' android:required="%s"' % ('true' if value else 'false') | 
|  |  | 
|  |  | 
|  | def uses_library_apk(name, sfx=''): | 
|  | return "uses-library%s:'%s'" % (sfx, name) | 
|  |  | 
|  |  | 
|  | def required_apk(value): | 
|  | return '' if value else '-not-required' | 
|  |  | 
|  |  | 
|  | class EnforceUsesLibrariesTest(unittest.TestCase): | 
|  | """Unit tests for add_extract_native_libs function.""" | 
|  |  | 
|  | def run_test(self, xml, apk, uses_libraries=[], optional_uses_libraries=[], | 
|  | missing_optional_uses_libraries=[]): #pylint: disable=dangerous-default-value | 
|  | doc = minidom.parseString(xml) | 
|  | try: | 
|  | relax = False | 
|  | manifest_check.enforce_uses_libraries( | 
|  | doc, uses_libraries, optional_uses_libraries, missing_optional_uses_libraries, | 
|  | relax, False, 'path/to/X/AndroidManifest.xml') | 
|  | manifest_check.enforce_uses_libraries(apk, uses_libraries, | 
|  | optional_uses_libraries, | 
|  | missing_optional_uses_libraries, | 
|  | relax, True, | 
|  | 'path/to/X/X.apk') | 
|  | return True | 
|  | except manifest_check.ManifestMismatchError: | 
|  | return False | 
|  |  | 
|  | xml_tmpl = ( | 
|  | '<?xml version="1.0" encoding="utf-8"?>\n<manifest ' | 
|  | 'xmlns:android="http://schemas.android.com/apk/res/android">\n    ' | 
|  | '<application>\n    %s\n    </application>\n</manifest>\n') | 
|  |  | 
|  | apk_tmpl = ( | 
|  | "package: name='com.google.android.something' versionCode='100'\n" | 
|  | "sdkVersion:'29'\n" | 
|  | "targetSdkVersion:'29'\n" | 
|  | "uses-permission: name='android.permission.ACCESS_NETWORK_STATE'\n" | 
|  | '%s\n' | 
|  | "densities: '160' '240' '320' '480' '640' '65534") | 
|  |  | 
|  | def test_uses_library(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo')) | 
|  | apk = self.apk_tmpl % (uses_library_apk('foo')) | 
|  | matches = self.run_test(xml, apk, uses_libraries=['foo']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_uses_library_required(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo', required_xml(True))) | 
|  | apk = self.apk_tmpl % (uses_library_apk('foo', required_apk(True))) | 
|  | matches = self.run_test(xml, apk, uses_libraries=['foo']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo', required_xml(False))) | 
|  | apk = self.apk_tmpl % (uses_library_apk('foo', required_apk(False))) | 
|  | matches = self.run_test(xml, apk, optional_uses_libraries=['foo']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_expected_uses_library(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo', required_xml(False))) | 
|  | apk = self.apk_tmpl % (uses_library_apk('foo', required_apk(False))) | 
|  | matches = self.run_test(xml, apk, uses_libraries=['foo']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_expected_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo')) | 
|  | apk = self.apk_tmpl % (uses_library_apk('foo')) | 
|  | matches = self.run_test(xml, apk, optional_uses_libraries=['foo']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_expected_missing_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % ( | 
|  | uses_library_xml('foo') + uses_library_xml('missing') + uses_library_xml('bar')) | 
|  | apk = self.apk_tmpl % ( | 
|  | uses_library_apk('foo') + uses_library_apk('missing') + uses_library_apk('bar')) | 
|  | matches = self.run_test(xml, apk, optional_uses_libraries=['foo', 'bar'], | 
|  | missing_optional_uses_libraries=['missing']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_missing_uses_library(self): | 
|  | xml = self.xml_tmpl % ('') | 
|  | apk = self.apk_tmpl % ('') | 
|  | matches = self.run_test(xml, apk, uses_libraries=['foo']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_missing_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % ('') | 
|  | apk = self.apk_tmpl % ('') | 
|  | matches = self.run_test(xml, apk, optional_uses_libraries=['foo']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_extra_uses_library(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo')) | 
|  | apk = self.apk_tmpl % (uses_library_xml('foo')) | 
|  | matches = self.run_test(xml, apk) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_extra_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % (uses_library_xml('foo', required_xml(False))) | 
|  | apk = self.apk_tmpl % (uses_library_apk('foo', required_apk(False))) | 
|  | matches = self.run_test(xml, apk) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_multiple_uses_library(self): | 
|  | xml = self.xml_tmpl % ('\n'.join( | 
|  | [uses_library_xml('foo'), | 
|  | uses_library_xml('bar')])) | 
|  | apk = self.apk_tmpl % ('\n'.join( | 
|  | [uses_library_apk('foo'), | 
|  | uses_library_apk('bar')])) | 
|  | matches = self.run_test(xml, apk, uses_libraries=['foo', 'bar']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_multiple_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % ('\n'.join([ | 
|  | uses_library_xml('foo', required_xml(False)), | 
|  | uses_library_xml('bar', required_xml(False)) | 
|  | ])) | 
|  | apk = self.apk_tmpl % ('\n'.join([ | 
|  | uses_library_apk('foo', required_apk(False)), | 
|  | uses_library_apk('bar', required_apk(False)) | 
|  | ])) | 
|  | matches = self.run_test( | 
|  | xml, apk, optional_uses_libraries=['foo', 'bar']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_order_uses_library(self): | 
|  | xml = self.xml_tmpl % ('\n'.join( | 
|  | [uses_library_xml('foo'), | 
|  | uses_library_xml('bar')])) | 
|  | apk = self.apk_tmpl % ('\n'.join( | 
|  | [uses_library_apk('foo'), | 
|  | uses_library_apk('bar')])) | 
|  | matches = self.run_test(xml, apk, uses_libraries=['bar', 'foo']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_order_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % ('\n'.join([ | 
|  | uses_library_xml('foo', required_xml(False)), | 
|  | uses_library_xml('bar', required_xml(False)) | 
|  | ])) | 
|  | apk = self.apk_tmpl % ('\n'.join([ | 
|  | uses_library_apk('foo', required_apk(False)), | 
|  | uses_library_apk('bar', required_apk(False)) | 
|  | ])) | 
|  | matches = self.run_test( | 
|  | xml, apk, optional_uses_libraries=['bar', 'foo']) | 
|  | self.assertFalse(matches) | 
|  |  | 
|  | def test_duplicate_uses_library(self): | 
|  | xml = self.xml_tmpl % ('\n'.join( | 
|  | [uses_library_xml('foo'), | 
|  | uses_library_xml('foo')])) | 
|  | apk = self.apk_tmpl % ('\n'.join( | 
|  | [uses_library_apk('foo'), | 
|  | uses_library_apk('foo')])) | 
|  | matches = self.run_test(xml, apk, uses_libraries=['foo']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_duplicate_optional_uses_library(self): | 
|  | xml = self.xml_tmpl % ('\n'.join([ | 
|  | uses_library_xml('foo', required_xml(False)), | 
|  | uses_library_xml('foo', required_xml(False)) | 
|  | ])) | 
|  | apk = self.apk_tmpl % ('\n'.join([ | 
|  | uses_library_apk('foo', required_apk(False)), | 
|  | uses_library_apk('foo', required_apk(False)) | 
|  | ])) | 
|  | matches = self.run_test(xml, apk, optional_uses_libraries=['foo']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_mixed(self): | 
|  | xml = self.xml_tmpl % ('\n'.join([ | 
|  | uses_library_xml('foo'), | 
|  | uses_library_xml('bar', required_xml(False)) | 
|  | ])) | 
|  | apk = self.apk_tmpl % ('\n'.join([ | 
|  | uses_library_apk('foo'), | 
|  | uses_library_apk('bar', required_apk(False)) | 
|  | ])) | 
|  | matches = self.run_test( | 
|  | xml, apk, uses_libraries=['foo'], optional_uses_libraries=['bar']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  | def test_mixed_with_namespace(self): | 
|  | xml = self.xml_tmpl % ('\n'.join([ | 
|  | uses_library_xml('foo'), | 
|  | uses_library_xml('bar', required_xml(False)) | 
|  | ])) | 
|  | apk = self.apk_tmpl % ('\n'.join([ | 
|  | uses_library_apk('foo'), | 
|  | uses_library_apk('bar', required_apk(False)) | 
|  | ])) | 
|  | matches = self.run_test( | 
|  | xml, | 
|  | apk, | 
|  | uses_libraries=['//x/y/z:foo'], | 
|  | optional_uses_libraries=['//x/y/z:bar']) | 
|  | self.assertTrue(matches) | 
|  |  | 
|  |  | 
|  | class ExtractTargetSdkVersionTest(unittest.TestCase): | 
|  |  | 
|  | def run_test(self, xml, apk, version): | 
|  | doc = minidom.parseString(xml) | 
|  | v = manifest_check.extract_target_sdk_version(doc, is_apk=False) | 
|  | self.assertEqual(v, version) | 
|  | v = manifest_check.extract_target_sdk_version(apk, is_apk=True) | 
|  | self.assertEqual(v, version) | 
|  |  | 
|  | xml_tmpl = ( | 
|  | '<?xml version="1.0" encoding="utf-8"?>\n<manifest ' | 
|  | 'xmlns:android="http://schemas.android.com/apk/res/android">\n    ' | 
|  | '<uses-sdk android:minSdkVersion="28" android:targetSdkVersion="%s" ' | 
|  | '/>\n</manifest>\n') | 
|  |  | 
|  | apk_tmpl = ( | 
|  | "package: name='com.google.android.something' versionCode='100'\n" | 
|  | "sdkVersion:'28'\n" | 
|  | "targetSdkVersion:'%s'\n" | 
|  | "uses-permission: name='android.permission.ACCESS_NETWORK_STATE'\n") | 
|  |  | 
|  | def test_targert_sdk_version_28(self): | 
|  | xml = self.xml_tmpl % '28' | 
|  | apk = self.apk_tmpl % '28' | 
|  | self.run_test(xml, apk, '28') | 
|  |  | 
|  | def test_targert_sdk_version_29(self): | 
|  | xml = self.xml_tmpl % '29' | 
|  | apk = self.apk_tmpl % '29' | 
|  | self.run_test(xml, apk, '29') | 
|  |  | 
|  |  | 
|  | if __name__ == '__main__': | 
|  | unittest.main(verbosity=2) |