58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
import os
|
|
|
|
import infra.basetest
|
|
|
|
|
|
class TestPv(infra.basetest.BRTest):
|
|
config = infra.basetest.BASIC_TOOLCHAIN_CONFIG + \
|
|
"""
|
|
BR2_PACKAGE_PV=y
|
|
BR2_TARGET_ROOTFS_CPIO=y
|
|
# BR2_TARGET_ROOTFS_TAR is not set
|
|
"""
|
|
|
|
def test_run(self):
|
|
cpio_file = os.path.join(self.builddir, "images", "rootfs.cpio")
|
|
self.emulator.boot(arch="armv5",
|
|
kernel="builtin",
|
|
options=["-initrd", cpio_file])
|
|
self.emulator.login()
|
|
|
|
# We check the program can run.
|
|
self.assertRunOk("pv --version")
|
|
|
|
# We check that "pv" works like the "cat" command. We print a
|
|
# message on its standard input and redirect its output to a
|
|
# file. We also force the pv terminal width to 80, since
|
|
# "infra.emulator" is setting a large width to prevent
|
|
# line wrapping.
|
|
msg = "Hello Buildroot!"
|
|
out_file = "/tmp/out.txt"
|
|
cmd = f"echo '{msg}' | pv -w80 > {out_file}"
|
|
self.assertRunOk(cmd)
|
|
|
|
# We check the pv output file contains exactly our message.
|
|
cmd = f"cat {out_file}"
|
|
out, ret = self.emulator.run(cmd)
|
|
self.assertEqual(ret, 0)
|
|
self.assertEqual(out[0], msg)
|
|
|
|
# Finally, we check that "pv" correctly shows a progress
|
|
# bar. We print few lines, one per second into "pv" setup in
|
|
# line mode. We check the last pv status line contains the
|
|
# correct count and a "100%" string that shows completion.
|
|
lines = 5
|
|
print_ln_cmd = f"( for X in $(seq {lines}) ; do echo $X ; sleep 1 ; done )"
|
|
cmd = f"{print_ln_cmd} | pv -s{lines} -l -w80 > /dev/null"
|
|
out, ret = self.emulator.run(cmd, timeout=10)
|
|
self.assertEqual(ret, 0)
|
|
# pv updates status may contain extra spaces, and is updated
|
|
# with carriage return characters. We strip lines and filter
|
|
# out empty remaining lines, to make sure we get the last
|
|
# meaningful status line.
|
|
pv_out = [ln.strip() for ln in out]
|
|
pv_out = [ln for ln in pv_out if ln]
|
|
last_line = pv_out[-1]
|
|
self.assertTrue(last_line.startswith(str(lines)))
|
|
self.assertTrue(last_line.endswith("100%"))
|