poky/meta/lib/oeqa/selftest/cases/liboe.py
Ross Burton ef6748323f oeqa/selftest: add test case for oeqa.utils.subprocesstweak
This class has a monkey-patched CalledProcessError instance that extends
the __str__ method. Add a test case to ensure that it behaves as
expected.

(From OE-Core rev: 3c0e7b537eac62fced196e82ede781d88e593a69)

Signed-off-by: Ross Burton <ross.burton@arm.com>
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
2025-06-09 17:43:41 +01:00

138 lines
4.8 KiB
Python

#
# Copyright OpenEmbedded Contributors
#
# SPDX-License-Identifier: MIT
#
from oeqa.selftest.case import OESelftestTestCase
from oeqa.utils.commands import get_bb_var, get_bb_vars, bitbake, runCmd
import oe.path
import os
class CopyTreeTests(OESelftestTestCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.tmp_dir = get_bb_var('TMPDIR')
def test_copy_tree_special(self):
"""
Summary: oe.path.copytree() should copy files with special character
Expected: 'test file with sp£c!al @nd spaces' should exist in
copy destination
Product: OE-Core
Author: Joshua Lock <joshua.g.lock@intel.com>
"""
testloc = oe.path.join(self.tmp_dir, 'liboetests')
src = oe.path.join(testloc, 'src')
dst = oe.path.join(testloc, 'dst')
bb.utils.mkdirhier(testloc)
bb.utils.mkdirhier(src)
testfilename = 'test file with sp£c!al @nd spaces'
# create the test file and copy it
open(oe.path.join(src, testfilename), 'w+b').close()
oe.path.copytree(src, dst)
# ensure path exists in dest
fileindst = os.path.isfile(oe.path.join(dst, testfilename))
self.assertTrue(fileindst, "File with spaces doesn't exist in dst")
oe.path.remove(testloc)
def test_copy_tree_xattr(self):
"""
Summary: oe.path.copytree() should preserve xattr on copied files
Expected: testxattr file in destination should have user.oetest
extended attribute
Product: OE-Core
Author: Joshua Lock <joshua.g.lock@intel.com>
"""
testloc = oe.path.join(self.tmp_dir, 'liboetests')
src = oe.path.join(testloc, 'src')
dst = oe.path.join(testloc, 'dst')
bb.utils.mkdirhier(testloc)
bb.utils.mkdirhier(src)
testfilename = 'testxattr'
# ensure we have setfattr available
bitbake("attr-native")
bb_vars = get_bb_vars(['SYSROOT_DESTDIR', 'bindir'], 'attr-native')
destdir = bb_vars['SYSROOT_DESTDIR']
bindir = bb_vars['bindir']
bindir = destdir + bindir
# create a file with xattr and copy it
open(oe.path.join(src, testfilename), 'w+b').close()
runCmd('%s/setfattr -n user.oetest -v "testing liboe" %s' % (bindir, oe.path.join(src, testfilename)))
oe.path.copytree(src, dst)
# ensure file in dest has user.oetest xattr
result = runCmd('%s/getfattr -n user.oetest %s' % (bindir, oe.path.join(dst, testfilename)))
self.assertIn('user.oetest="testing liboe"', result.output, 'Extended attribute not sert in dst')
oe.path.remove(testloc)
def test_copy_hardlink_tree_count(self):
"""
Summary: oe.path.copyhardlinktree() shouldn't miss out files
Expected: src and dst should have the same number of files
Product: OE-Core
Author: Joshua Lock <joshua.g.lock@intel.com>
"""
testloc = oe.path.join(self.tmp_dir, 'liboetests')
src = oe.path.join(testloc, 'src')
dst = oe.path.join(testloc, 'dst')
bb.utils.mkdirhier(testloc)
bb.utils.mkdirhier(src)
testfiles = ['foo', 'bar', '.baz', 'quux']
def touchfile(tf):
open(oe.path.join(src, tf), 'w+b').close()
for f in testfiles:
touchfile(f)
oe.path.copyhardlinktree(src, dst)
dstcnt = len(os.listdir(dst))
srccnt = len(os.listdir(src))
self.assertEqual(dstcnt, len(testfiles), "Number of files in dst (%s) differs from number of files in src(%s)." % (dstcnt, srccnt))
oe.path.remove(testloc)
class SubprocessTests(OESelftestTestCase):
def test_subprocess_tweak(self):
"""
Test that the string representation of
oeqa.utils.subprocesstweak.OETestCalledProcessError includes stdout and
stderr, as expected.
"""
script = """
#! /bin/sh
echo Ivn fgqbhg | tr '[a-zA-Z]' '[n-za-mN-ZA-M]'
echo Ivn fgqree | tr '[a-zA-Z]' '[n-za-mN-ZA-M]' >&2
exit 42
"""
import subprocess
import unittest.mock
from oeqa.utils.subprocesstweak import OETestCalledProcessError
with self.assertRaises(OETestCalledProcessError) as cm:
with unittest.mock.patch("subprocess.CalledProcessError", OETestCalledProcessError):
subprocess.run(["bash", "-"], input=script, text=True, capture_output=True, check=True)
e = cm.exception
self.assertEqual(e.returncode, 42)
self.assertEqual("Via stdout\n", e.stdout)
self.assertEqual("Via stderr\n", e.stderr)
string = str(e)
self.assertIn("exit status 42", string)
self.assertIn("Standard Output: Via stdout", string)
self.assertIn("Standard Error: Via stderr", string)