Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 1 | import unittest |
| 2 | import os |
| 3 | import logging |
| 4 | import re |
| 5 | import shutil |
| 6 | import datetime |
| 7 | |
| 8 | import oeqa.utils.ftools as ftools |
| 9 | from oeqa.selftest.base import oeSelfTest |
| 10 | from oeqa.utils.commands import runCmd, bitbake, get_bb_var |
| 11 | from oeqa.utils.decorators import testcase |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 12 | from oeqa.utils.network import get_free_port |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 13 | |
| 14 | class BitbakePrTests(oeSelfTest): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 15 | |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 16 | def get_pr_version(self, package_name): |
| 17 | pkgdata_dir = get_bb_var('PKGDATA_DIR') |
| 18 | package_data_file = os.path.join(pkgdata_dir, 'runtime', package_name) |
| 19 | package_data = ftools.read_file(package_data_file) |
| 20 | find_pr = re.search("PKGR: r[0-9]+\.([0-9]+)", package_data) |
| 21 | self.assertTrue(find_pr, "No PKG revision found in %s" % package_data_file) |
| 22 | return int(find_pr.group(1)) |
| 23 | |
| 24 | def get_task_stamp(self, package_name, recipe_task): |
| 25 | stampdata = get_bb_var('STAMP', target=package_name).split('/') |
| 26 | prefix = stampdata[-1] |
| 27 | package_stamps_path = "/".join(stampdata[:-1]) |
| 28 | stamps = [] |
| 29 | for stamp in os.listdir(package_stamps_path): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 30 | find_stamp = re.match("%s\.%s\.([a-z0-9]{32})" % (re.escape(prefix), recipe_task), stamp) |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 31 | if find_stamp: |
| 32 | stamps.append(find_stamp.group(1)) |
| 33 | self.assertFalse(len(stamps) == 0, msg="Cound not find stamp for task %s for recipe %s" % (recipe_task, package_name)) |
| 34 | self.assertFalse(len(stamps) > 1, msg="Found multiple %s stamps for the %s recipe in the %s directory." % (recipe_task, package_name, package_stamps_path)) |
| 35 | return str(stamps[0]) |
| 36 | |
| 37 | def increment_package_pr(self, package_name): |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 38 | inc_data = "do_package_append() {\n bb.build.exec_func('do_test_prserv', d)\n}\ndo_test_prserv() {\necho \"The current date is: %s\"\n}" % datetime.datetime.now() |
Patrick Williams | c124f4f | 2015-09-15 14:41:29 -0500 | [diff] [blame] | 39 | self.write_recipeinc(package_name, inc_data) |
| 40 | bitbake("-ccleansstate %s" % package_name) |
| 41 | res = bitbake(package_name, ignore_status=True) |
| 42 | self.delete_recipeinc(package_name) |
| 43 | self.assertEqual(res.status, 0, msg=res.output) |
| 44 | self.assertTrue("NOTE: Started PRServer with DBfile" in res.output, msg=res.output) |
| 45 | |
| 46 | def config_pr_tests(self, package_name, package_type='rpm', pr_socket='localhost:0'): |
| 47 | config_package_data = 'PACKAGE_CLASSES = "package_%s"' % package_type |
| 48 | self.write_config(config_package_data) |
| 49 | config_server_data = 'PRSERV_HOST = "%s"' % pr_socket |
| 50 | self.append_config(config_server_data) |
| 51 | |
| 52 | def run_test_pr_service(self, package_name, package_type='rpm', track_task='do_package', pr_socket='localhost:0'): |
| 53 | self.config_pr_tests(package_name, package_type, pr_socket) |
| 54 | |
| 55 | self.increment_package_pr(package_name) |
| 56 | pr_1 = self.get_pr_version(package_name) |
| 57 | stamp_1 = self.get_task_stamp(package_name, track_task) |
| 58 | |
| 59 | self.increment_package_pr(package_name) |
| 60 | pr_2 = self.get_pr_version(package_name) |
| 61 | stamp_2 = self.get_task_stamp(package_name, track_task) |
| 62 | |
| 63 | bitbake("-ccleansstate %s" % package_name) |
| 64 | self.assertTrue(pr_2 - pr_1 == 1, "Step between same pkg. revision is greater than 1") |
| 65 | self.assertTrue(stamp_1 != stamp_2, "Different pkg rev. but same stamp: %s" % stamp_1) |
| 66 | |
| 67 | def run_test_pr_export_import(self, package_name, replace_current_db=True): |
| 68 | self.config_pr_tests(package_name) |
| 69 | |
| 70 | self.increment_package_pr(package_name) |
| 71 | pr_1 = self.get_pr_version(package_name) |
| 72 | |
| 73 | exported_db_path = os.path.join(self.builddir, 'export.inc') |
| 74 | export_result = runCmd("bitbake-prserv-tool export %s" % exported_db_path, ignore_status=True) |
| 75 | self.assertEqual(export_result.status, 0, msg="PR Service database export failed: %s" % export_result.output) |
| 76 | |
| 77 | if replace_current_db: |
| 78 | current_db_path = os.path.join(get_bb_var('PERSISTENT_DIR'), 'prserv.sqlite3') |
| 79 | self.assertTrue(os.path.exists(current_db_path), msg="Path to current PR Service database is invalid: %s" % current_db_path) |
| 80 | os.remove(current_db_path) |
| 81 | |
| 82 | import_result = runCmd("bitbake-prserv-tool import %s" % exported_db_path, ignore_status=True) |
| 83 | os.remove(exported_db_path) |
| 84 | self.assertEqual(import_result.status, 0, msg="PR Service database import failed: %s" % import_result.output) |
| 85 | |
| 86 | self.increment_package_pr(package_name) |
| 87 | pr_2 = self.get_pr_version(package_name) |
| 88 | |
| 89 | bitbake("-ccleansstate %s" % package_name) |
| 90 | self.assertTrue(pr_2 - pr_1 == 1, "Step between same pkg. revision is greater than 1") |
| 91 | |
| 92 | @testcase(930) |
| 93 | def test_import_export_replace_db(self): |
| 94 | self.run_test_pr_export_import('m4') |
| 95 | |
| 96 | @testcase(931) |
| 97 | def test_import_export_override_db(self): |
| 98 | self.run_test_pr_export_import('m4', replace_current_db=False) |
| 99 | |
| 100 | @testcase(932) |
| 101 | def test_pr_service_rpm_arch_dep(self): |
| 102 | self.run_test_pr_service('m4', 'rpm', 'do_package') |
| 103 | |
| 104 | @testcase(934) |
| 105 | def test_pr_service_deb_arch_dep(self): |
| 106 | self.run_test_pr_service('m4', 'deb', 'do_package') |
| 107 | |
| 108 | @testcase(933) |
| 109 | def test_pr_service_ipk_arch_dep(self): |
| 110 | self.run_test_pr_service('m4', 'ipk', 'do_package') |
| 111 | |
| 112 | @testcase(935) |
| 113 | def test_pr_service_rpm_arch_indep(self): |
| 114 | self.run_test_pr_service('xcursor-transparent-theme', 'rpm', 'do_package') |
| 115 | |
| 116 | @testcase(937) |
| 117 | def test_pr_service_deb_arch_indep(self): |
| 118 | self.run_test_pr_service('xcursor-transparent-theme', 'deb', 'do_package') |
| 119 | |
| 120 | @testcase(936) |
| 121 | def test_pr_service_ipk_arch_indep(self): |
| 122 | self.run_test_pr_service('xcursor-transparent-theme', 'ipk', 'do_package') |
Patrick Williams | d8c66bc | 2016-06-20 12:57:21 -0500 | [diff] [blame] | 123 | |
| 124 | @testcase(1419) |
| 125 | def test_stopping_prservice_message(self): |
| 126 | port = get_free_port() |
| 127 | |
| 128 | runCmd('bitbake-prserv --host localhost --port %s --loglevel=DEBUG --start' % port) |
| 129 | ret = runCmd('bitbake-prserv --host localhost --port %s --loglevel=DEBUG --stop' % port) |
| 130 | |
| 131 | self.assertEqual(ret.status, 0) |
| 132 | |