1*4882a593Smuzhiyun# 2*4882a593Smuzhiyun# SPDX-License-Identifier: MIT 3*4882a593Smuzhiyun# 4*4882a593Smuzhiyun 5*4882a593Smuzhiyunimport datetime 6*4882a593Smuzhiyunimport os 7*4882a593Smuzhiyunimport re 8*4882a593Smuzhiyun 9*4882a593Smuzhiyunfrom oeqa.selftest.case import OESelftestTestCase 10*4882a593Smuzhiyunfrom oeqa.utils.commands import get_bb_vars 11*4882a593Smuzhiyun 12*4882a593Smuzhiyun 13*4882a593Smuzhiyunclass SStateBase(OESelftestTestCase): 14*4882a593Smuzhiyun 15*4882a593Smuzhiyun def setUpLocal(self): 16*4882a593Smuzhiyun super(SStateBase, self).setUpLocal() 17*4882a593Smuzhiyun self.temp_sstate_location = None 18*4882a593Smuzhiyun needed_vars = ['SSTATE_DIR', 'NATIVELSBSTRING', 'TCLIBC', 'TUNE_ARCH', 19*4882a593Smuzhiyun 'TOPDIR', 'TARGET_VENDOR', 'TARGET_OS'] 20*4882a593Smuzhiyun bb_vars = get_bb_vars(needed_vars) 21*4882a593Smuzhiyun self.sstate_path = bb_vars['SSTATE_DIR'] 22*4882a593Smuzhiyun self.hostdistro = bb_vars['NATIVELSBSTRING'] 23*4882a593Smuzhiyun self.tclibc = bb_vars['TCLIBC'] 24*4882a593Smuzhiyun self.tune_arch = bb_vars['TUNE_ARCH'] 25*4882a593Smuzhiyun self.topdir = bb_vars['TOPDIR'] 26*4882a593Smuzhiyun self.target_vendor = bb_vars['TARGET_VENDOR'] 27*4882a593Smuzhiyun self.target_os = bb_vars['TARGET_OS'] 28*4882a593Smuzhiyun self.distro_specific_sstate = os.path.join(self.sstate_path, self.hostdistro) 29*4882a593Smuzhiyun 30*4882a593Smuzhiyun # Creates a special sstate configuration with the option to add sstate mirrors 31*4882a593Smuzhiyun def config_sstate(self, temp_sstate_location=False, add_local_mirrors=[]): 32*4882a593Smuzhiyun self.temp_sstate_location = temp_sstate_location 33*4882a593Smuzhiyun 34*4882a593Smuzhiyun if self.temp_sstate_location: 35*4882a593Smuzhiyun temp_sstate_path = os.path.join(self.builddir, "temp_sstate_%s" % datetime.datetime.now().strftime('%Y%m%d%H%M%S')) 36*4882a593Smuzhiyun config_temp_sstate = "SSTATE_DIR = \"%s\"" % temp_sstate_path 37*4882a593Smuzhiyun self.append_config(config_temp_sstate) 38*4882a593Smuzhiyun self.track_for_cleanup(temp_sstate_path) 39*4882a593Smuzhiyun bb_vars = get_bb_vars(['SSTATE_DIR', 'NATIVELSBSTRING']) 40*4882a593Smuzhiyun self.sstate_path = bb_vars['SSTATE_DIR'] 41*4882a593Smuzhiyun self.hostdistro = bb_vars['NATIVELSBSTRING'] 42*4882a593Smuzhiyun self.distro_specific_sstate = os.path.join(self.sstate_path, self.hostdistro) 43*4882a593Smuzhiyun 44*4882a593Smuzhiyun if add_local_mirrors: 45*4882a593Smuzhiyun config_set_sstate_if_not_set = 'SSTATE_MIRRORS ?= ""' 46*4882a593Smuzhiyun self.append_config(config_set_sstate_if_not_set) 47*4882a593Smuzhiyun for local_mirror in add_local_mirrors: 48*4882a593Smuzhiyun self.assertFalse(os.path.join(local_mirror) == os.path.join(self.sstate_path), msg='Cannot add the current sstate path as a sstate mirror') 49*4882a593Smuzhiyun config_sstate_mirror = "SSTATE_MIRRORS += \"file://.* file:///%s/PATH\"" % local_mirror 50*4882a593Smuzhiyun self.append_config(config_sstate_mirror) 51*4882a593Smuzhiyun 52*4882a593Smuzhiyun # Returns a list containing sstate files 53*4882a593Smuzhiyun def search_sstate(self, filename_regex, distro_specific=True, distro_nonspecific=True): 54*4882a593Smuzhiyun result = [] 55*4882a593Smuzhiyun for root, dirs, files in os.walk(self.sstate_path): 56*4882a593Smuzhiyun if distro_specific and re.search(r"%s/%s/[a-z0-9]{2}/[a-z0-9]{2}$" % (self.sstate_path, self.hostdistro), root): 57*4882a593Smuzhiyun for f in files: 58*4882a593Smuzhiyun if re.search(filename_regex, f): 59*4882a593Smuzhiyun result.append(f) 60*4882a593Smuzhiyun if distro_nonspecific and re.search(r"%s/[a-z0-9]{2}/[a-z0-9]{2}$" % self.sstate_path, root): 61*4882a593Smuzhiyun for f in files: 62*4882a593Smuzhiyun if re.search(filename_regex, f): 63*4882a593Smuzhiyun result.append(f) 64*4882a593Smuzhiyun return result 65