ContextManagersTest

Complexity

Total Complexity

27

Size/Duplication

Total Lines	127
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

10 Methods

Rating	Name	Duplication	Size	Complexity
B	test_prepare_file()	0	31	5
A	test_suppress_stdout()	0	17	4
A	test_make_temp()	0	8	3
A	test_simulate_console_inputs()	0	13	3
A	test_retrieve_stderr()	0	4	2
B	test_subprocess_timeout()	0	34	5
A	no_print_func()	0	4	2
A	print_func()	0	3	1
A	test_retrieve_stdout()	0	4	2
A	test_change_directory()	0	7	3

import os
import subprocess
import sys
from tempfile import TemporaryDirectory
import unittest

from coalib.misc.ContextManagers import (
    change_directory, make_temp, prepare_file, retrieve_stdout,
    retrieve_stderr, simulate_console_inputs, subprocess_timeout,
    suppress_stdout)
from coalib.processes.Processing import create_process_group


process_group_timeout_test_code = """
import time, subprocess, sys;
p = subprocess.Popen([sys.executable,
                     "-c",
                     "import time; time.sleep(100)"]);
time.sleep(100);
"""


class ContextManagersTest(unittest.TestCase):

The variable unittest does not seem to be defined.

    def test_subprocess_timeout(self):
        p = subprocess.Popen([sys.executable,

The variable sys does not seem to be defined.

                              "-c",
                              "import time; time.sleep(0.5);"],
                             stderr=subprocess.PIPE)

The variable subprocess does not seem to be defined.

        with subprocess_timeout(p, 0.2) as timedout:

The variable timedout does not seem to be defined.

The variable p does not seem to be defined.

            retval = p.wait()
            p.stderr.close()
            self.assertEqual(timedout.value, True)
        self.assertNotEqual(retval, 0)

The variable retval does not seem to be defined.

        p = create_process_group([sys.executable,
                                  "-c",
                                  process_group_timeout_test_code])

The variable process_group_timeout_test_code does not seem to be defined.

        with subprocess_timeout(p, 0.5, kill_pg=True):
            retval = p.wait()
            self.assertEqual(timedout.value, True)
        self.assertNotEqual(retval, 0)

        p = subprocess.Popen([sys.executable,
                              "-c",
                              "import time"])
        with subprocess_timeout(p, 0.5) as timedout:
            retval = p.wait()
            self.assertEqual(timedout.value, False)
        self.assertEqual(retval, 0)

        p = subprocess.Popen([sys.executable,
                              "-c",
                              "import time"])
        with subprocess_timeout(p, 0) as timedout:
            retval = p.wait()
            self.assertEqual(timedout.value, False)
        self.assertEqual(retval, 0)

    def test_suppress_stdout(self):
        def print_func():
            print("func")
            raise NotImplementedError

The variable NotImplementedError does not seem to be defined.

        def no_print_func():
            with suppress_stdout():
                print("func")
                raise NotImplementedError

The variable NotImplementedError does not seem to be defined.

        old_stdout = sys.stdout

The variable sys does not seem to be defined.

        sys.stdout = False

        self.assertRaises(AttributeError, print_func)

The variable AttributeError does not seem to be defined.

The variable print_func does not seem to be defined.

        self.assertRaises(NotImplementedError, no_print_func)

The variable no_print_func does not seem to be defined.

        sys.stdout = old_stdout

The variable old_stdout does not seem to be defined.

    def test_retrieve_stdout(self):
        with retrieve_stdout() as sio:

The variable sio does not seem to be defined.

            print("test", file=sys.stdout)

The variable sys does not seem to be defined.

            self.assertEqual(sio.getvalue(), "test\n")

    def test_retrieve_stderr(self):
        with retrieve_stderr() as sio:

The variable sio does not seem to be defined.

            print("test", file=sys.stderr)

The variable sys does not seem to be defined.

            self.assertEqual(sio.getvalue(), "test\n")

    def test_simulate_console_inputs(self):
        with simulate_console_inputs(0, 1, 2) as generator:

The variable generator does not seem to be defined.

            self.assertEqual(input(), 0)
            self.assertEqual(generator.last_input, 0)
            generator.inputs.append(3)
            self.assertEqual(input(), 1)
            self.assertEqual(input(), 2)
            self.assertEqual(input(), 3)
            self.assertEqual(generator.last_input, 3)

        with simulate_console_inputs("test"), self.assertRaises(ValueError):

The variable ValueError does not seem to be defined.

            self.assertEqual(input(), "test")
            input()

    def test_make_temp(self):
        with make_temp() as f_a:

The variable f_a does not seem to be defined.

            self.assertTrue(os.path.isfile(f_a))
            self.assertTrue(os.path.basename(f_a).startswith("tmp"))

        with make_temp(suffix=".orig", prefix="pre") as f_b:

The variable f_b does not seem to be defined.

            self.assertTrue(f_b.endswith(".orig"))
            self.assertTrue(os.path.basename(f_b).startswith("pre"))

    def test_prepare_file(self):
        with prepare_file(['line1', 'line2\n'],
                          "/file/name",
                          force_linebreaks=True,
                          create_tempfile=True) as (lines, filename):

The variable filename does not seem to be defined.

The variable lines does not seem to be defined.

            self.assertEqual(filename, '/file/name')
            self.assertEqual(lines, ['line1\n', 'line2\n'])

        with prepare_file(['line1', 'line2\n'],
                          None,
                          force_linebreaks=False,
                          create_tempfile=True) as (lines, filename):
            self.assertTrue(os.path.isfile(filename))
            self.assertEqual(lines, ['line1', 'line2\n'])

        with prepare_file(['line1', 'line2\n'],
                          None,
                          tempfile_kwargs={"suffix": ".test",
                                           "prefix": "test_"},
                          force_linebreaks=False,
                          create_tempfile=True) as (lines, filename):
            self.assertTrue(os.path.isfile(filename))
            basename = os.path.basename(filename)
            self.assertTrue(basename.endswith(".test"))
            self.assertTrue(basename.startswith("test_"))

        with prepare_file(['line1', 'line2\n'],
                          None,
                          force_linebreaks=False,
                          create_tempfile=False) as (lines, filename):
            self.assertEqual(filename, "dummy_file_name")

    def test_change_directory(self):
        old_dir = os.getcwd()
        with TemporaryDirectory("temp") as tempdir:

The variable tempdir does not seem to be defined.

            tempdir = os.path.realpath(tempdir)
            with change_directory(tempdir):
                self.assertEqual(os.getcwd(), tempdir)
        self.assertEqual(os.getcwd(), old_dir)

The variable old_dir does not seem to be defined.