summaryrefslogtreecommitdiffstats
path: root/tests/test_install_progress_exec.py
blob: 064eb81a21626703eb5a549a92905f952b8f0704 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/usr/bin/python3
#
# Copyright (C) 2019 Colomban Wendling <cwendling@hypra.fr>
#
# Copying and distribution of this file, with or without modification,
# are permitted in any medium without royalty provided the copyright
# notice and this notice are preserved.
"""Unit tests for verifying InstallProgress works when spawning dpkg directly
(i.e. when installing standalone .debs)."""
import os
import sys
import unittest

from test_all import get_library_dir

libdir = get_library_dir()
if libdir:
    sys.path.insert(0, libdir)
import testcommon

from apt.progress.base import InstallProgress


class RunHelper:
    def __init__(self):
        self.script = "helper_install_progress_run.py"

    def do_install(self, fd):
        return os.spawnl(os.P_WAIT, self.script, self.script, str(fd))


class TestInstallProgressExec(testcommon.TestCase):
    """test that InstallProgress.run() passes a valid file descriptor to
    a child process"""

    def test_run(self):
        with InstallProgress() as prog:
            self.assertEqual(prog.run(RunHelper()), 0)


if __name__ == "__main__":
    os.chdir(os.path.dirname(__file__))
    unittest.main()