xref: /OK3568_Linux_fs/yocto/bitbake/lib/layerindexlib/tests/common.py (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun# Copyright (C) 2017-2018 Wind River Systems, Inc.
2*4882a593Smuzhiyun#
3*4882a593Smuzhiyun# SPDX-License-Identifier: GPL-2.0-only
4*4882a593Smuzhiyun#
5*4882a593Smuzhiyun
6*4882a593Smuzhiyunimport unittest
7*4882a593Smuzhiyunimport tempfile
8*4882a593Smuzhiyunimport os
9*4882a593Smuzhiyunimport bb
10*4882a593Smuzhiyun
11*4882a593Smuzhiyunimport logging
12*4882a593Smuzhiyun
13*4882a593Smuzhiyunclass LayersTest(unittest.TestCase):
14*4882a593Smuzhiyun
15*4882a593Smuzhiyun    def setUp(self):
16*4882a593Smuzhiyun        self.origdir = os.getcwd()
17*4882a593Smuzhiyun        self.d = bb.data.init()
18*4882a593Smuzhiyun        # At least one variable needs to be set
19*4882a593Smuzhiyun        self.d.setVar('DL_DIR', os.getcwd())
20*4882a593Smuzhiyun
21*4882a593Smuzhiyun        if os.environ.get("BB_SKIP_NETTESTS") == "yes":
22*4882a593Smuzhiyun            self.d.setVar('BB_NO_NETWORK', '1')
23*4882a593Smuzhiyun
24*4882a593Smuzhiyun        self.tempdir = tempfile.mkdtemp()
25*4882a593Smuzhiyun        self.logger = logging.getLogger("BitBake")
26*4882a593Smuzhiyun
27*4882a593Smuzhiyun    def tearDown(self):
28*4882a593Smuzhiyun        os.chdir(self.origdir)
29*4882a593Smuzhiyun        if os.environ.get("BB_TMPDIR_NOCLEAN") == "yes":
30*4882a593Smuzhiyun            print("Not cleaning up %s. Please remove manually." % self.tempdir)
31*4882a593Smuzhiyun        else:
32*4882a593Smuzhiyun            bb.utils.prunedir(self.tempdir)
33*4882a593Smuzhiyun
34