def test_no_resize_signal(self):
# Test normal case initialization
stdmgr = _manager.Manager(stream=self.tty.stdout)
self.assertTrue(hasattr(stdmgr, 'sigwinch_orig'))
stdmgr.counters[MockCounter(manager=stdmgr)] = 3
stdmgr.counters[MockCounter(manager=stdmgr)] = 4
# Test no resize signal initialization
with mock.patch.object(_manager, 'RESIZE_SUPPORTED', False):
manager = _manager.Manager(stream=self.tty.stdout)
self.assertFalse(hasattr(manager, 'sigwinch_orig'))
manager.counters[MockCounter(manager=manager)] = 3
manager.counters[MockCounter(manager=manager)] = 4
# Test set_scroll_area()
with mock.patch.object(_manager.signal,
'signal',
wraps=_manager.signal.signal) as mocksignal:
with mock.patch('enlighten._manager.atexit'):
# Test no resize signal set_scroll_area
with mock.patch.object(_manager, 'RESIZE_SUPPORTED', False):
with mock.patch.object(manager.term, 'change_scroll'):
manager._set_scroll_area()
self.assertFalse(mocksignal.called)
# Test normal case set_scroll_area
with mock.patch.object(stdmgr.term, 'change_scroll'):
stdmgr._set_scroll_area()
self.assertTrue(mocksignal.called)
# Test stop()
with mock.patch.object(_manager.signal,
'signal',
wraps=_manager.signal.signal) as mocksignal:
with mock.patch('%s.reset' % TERMINAL):
# Test no resize signal stop
with mock.patch.object(_manager, 'RESIZE_SUPPORTED', False):
manager.stop()
self.assertFalse(mocksignal.called)
# Test normal case stop
stdmgr.stop()
self.assertTrue(mocksignal.called)
def test_init_stderr_companion_hc(self):
# Hard-coded companion stream always wins
manager = _manager.Manager(stream=sys.__stderr__,
companion_stream=OUTPUT)
self.assertIs(manager.companion_stream, OUTPUT)
self.assertIs(manager.companion_term.stream, OUTPUT)
def test_counter_replaced(self):
"""Counter replaces an existing counter"""
manager = _manager.Manager(stream=self.tty.stdout, set_scroll=False)
# Pinned replacement
counter1 = manager.counter(position=2)
self.assertEqual(manager.counters[counter1], 2)
counter2 = manager.counter(replace=counter1)
self.assertEqual(len(manager.counters), 1)
self.assertEqual(manager.counters[counter2], 2)
self.assertTrue(counter2._pinned)
# Unpinned replacement
counter3 = manager.counter()
self.assertEqual(len(manager.counters), 2)
self.assertEqual(manager.counters[counter3], 1)
counter4 = manager.counter(replace=counter3)
self.assertEqual(len(manager.counters), 2)
self.assertEqual(manager.counters[counter4], 1)
self.assertFalse(counter4._pinned)
# Unmanaged counter given
with self.assertRaisesRegex(
ValueError, 'Counter to replace is not currently managed'):
manager.counter(replace=counter1)
def test_stop_position_1(self):
"""
Ensure a line feed is given if there is a counter at position 1
"""
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter)
term = manager.term
manager.counters[MockCounter(manager=manager)] = 3
with mock.patch.object(manager, '_flush_streams'):
manager.stop()
self.assertEqual(manager._buffer,
[term.normal_cursor, term.csr(0, term.height - 1),
term.move(term.height, 0)])
del manager._buffer[:]
manager.enabled = True
manager.counters[MockCounter(manager=manager)] = 1
with mock.patch.object(manager, '_flush_streams'):
manager.stop()
self.assertEqual(manager._buffer,
[term.normal_cursor, term.csr(0, term.height - 1),
term.move(term.height, 0), term.cud1 or '\n'])
def test_resize_threaded(self):
"""
Test a resize event threading behavior
"""
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter, threaded=True)
counter3 = MockCounter(manager=manager)
counter3.last_update = time.time()
manager.counters[counter3] = 3
manager.scroll_offset = 4
term = manager.term
# simulate resize
manager._stage_resize()
self.assertTrue(manager._resize)
self.assertEqual(counter3.last_update, 0)
with mock.patch('%s.width' % TERMINAL, new_callable=mock.PropertyMock) as mockwidth:
mockwidth.return_value = 70
# resize doesn't happen until a write is called
self.assertEqual(manager.width, 80)
with mock.patch('enlighten._manager.Manager._set_scroll_area') as ssa:
manager.write()
self.assertEqual(ssa.call_count, 1)
self.assertEqual(manager.width, 70)
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), term.move(21, 0) + term.clear_eos + 'X\n')
self.assertFalse(manager.resize_lock)
self.assertFalse(manager._resize)
self.assertEqual(counter3.calls, ['refresh(flush=False, elapsed=None)'])
def test_stop_companion(self):
"""
In this case, we have a companion terminal
Just make sure we have an extra reset
"""
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter,
companion_stream=self.tty.stdout)
manager.counters[MockCounter(manager=manager)] = 3
manager.counters[MockCounter(manager=manager)] = 4
term = manager.term
with mock.patch('enlighten._manager.atexit'):
manager._set_scroll_area()
del manager._buffer[:]
del manager._companion_buffer[:]
with mock.patch.object(manager, '_flush_streams'):
manager.stop()
self.assertEqual(manager._buffer,
[term.move(term.height - 2, 0), term.clear_eol,
term.move(term.height - 1, 0), term.clear_eol,
term.normal_cursor, term.csr(0, term.height - 1),
term.move(term.height, 0)])
self.assertEqual(manager._companion_buffer,
[term.normal_cursor, term.csr(0, term.height - 1),
term.move(term.height, 0)])
def test_set_scroll_area_no_change(self):
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter)
manager.counters['dummy'] = 3
manager.scroll_offset = 4
manager._set_scroll_area()
self.assertEqual(manager._buffer, [manager.term.move(21, 0)])
def test_flush_companion_buffer(self):
"""
Output is stored in buffer, but only written in companion stream is defined
"""
manager = _manager.Manager(stream=self.tty.stdout)
msg = u'test message'
manager._companion_buffer = [msg]
manager._flush_streams()
# Companion buffer flushed, but not outputted
self.assertEqual(manager._companion_buffer, [])
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
# set companion stream and test again
manager.companion_stream = OUTPUT
manager._companion_buffer = [msg]
manager._flush_streams()
self.assertEqual(manager._companion_buffer, [])
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
self.assertEqual(OUTPUT.getvalue(), msg)
def test_resize_handler_height_greater_threaded(self):
with mock.patch('%s.height' % TERMINAL, new_callable=mock.PropertyMock) as mockheight:
mockheight.side_effect = [25, 27]
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter,
threaded=True)
counter3 = MockCounter(manager=manager)
manager.counters[counter3] = 3
manager.scroll_offset = 4
term = manager.term
with mock.patch('enlighten._manager.Manager._set_scroll_area') as ssa:
manager._resize_handler()
self.assertEqual(ssa.call_count, 1)
self.assertEqual(manager.height, 27)
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), term.move(27, 0) + '\n')
self.assertEqual(self.tty.stdread.readline(), '\n')
self.assertEqual(self.tty.stdread.readline(), '\n')
self.assertEqual(self.tty.stdread.readline(), term.move(23, 0) + term.clear_eos + 'X\n')
self.assertEqual(counter3.calls, ['refresh(flush=False, elapsed=None)'])
def test_resize_handler_height_only(self):
with mock.patch('%s.height' % TERMINAL,
new_callable=mock.PropertyMock) as mockheight:
mockheight.side_effect = [25, 23, 28, 30, 30]
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
counter3 = MockCounter(manager=manager)
manager.counters[counter3] = 3
manager.scroll_offset = 4
with mock.patch(
'enlighten._manager.Manager._set_scroll_area') as ssa:
manager._resize_handler()
self.assertEqual(ssa.call_count, 1)
# Height is set in _set_scroll_area which is mocked
self.assertEqual(manager.height, 25)
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
self.assertEqual(counter3.calls,
['refresh(flush=False, elapsed=None)'])
def test_stop_no_set_scroll(self):
"""
set_scroll is False
"""
with mock.patch('%s.reset' % TERMINAL) as reset:
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter,
set_scroll=False)
manager.counters[MockCounter(manager=manager)] = 3
manager.counters[MockCounter(manager=manager)] = 4
term = manager.term
with mock.patch('enlighten._manager.atexit'):
with mock.patch.object(term, 'change_scroll'):
manager._set_scroll_area()
self.assertEqual(manager.scroll_offset, 5)
self.assertEqual(signal.getsignal(signal.SIGWINCH), manager._resize_handler)
self.assertTrue(manager.process_exit)
# Stream empty
self.tty.stdout.write('X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
manager.stop()
self.assertEqual(signal.getsignal(signal.SIGWINCH), manager.sigwinch_orig)
self.assertFalse(reset.called)
self.tty.stdout.write('X\n')
self.assertEqual(self.tty.stdread.readline(), term.move(23, 0) + term.clear_eol +
term.move(24, 0) + term.clear_eol + term.move(25, 0) + 'X\n')
self.assertFalse(manager.process_exit)
def test_at_exit(self):
tty = MockTTY()
with mock.patch('%s.reset' % TERMINAL) as reset:
manager = _manager.Manager(stream=tty.stdout, counter_class=MockCounter)
term = manager.term
# process_exit is False
manager._at_exit()
self.assertFalse(reset.called)
# No output
tty.stdout.write('X\n')
self.assertEqual(tty.stdread.readline(), 'X\n')
# process_exit is True, set_scroll False
manager.process_exit = True
manager.set_scroll = False
manager._at_exit()
self.assertFalse(reset.called)
self.assertEqual(tty.stdread.readline(), term.move(25, 0) + term.cud1)
# process_exit is True, set_scroll True
manager.set_scroll = True
manager._at_exit()
self.assertEqual(reset.call_count, 1)
self.assertEqual(tty.stdread.readline(), term.cud1)
tty.close()
manager._at_exit()
def test_resize_handler_height_less(self):
with mock.patch('%s.height' % TERMINAL,
new_callable=mock.PropertyMock) as mockheight:
mockheight.side_effect = [25, 23]
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
counter3 = MockCounter(manager=manager)
manager.counters[counter3] = 3
manager.scroll_offset = 4
with mock.patch(
'enlighten._manager.Manager._set_scroll_area') as ssa:
manager._resize_handler()
self.assertEqual(ssa.call_count, 1)
self.assertEqual(manager.height, 23)
self.assertEqual(self.tty.stdread.readline(),
manager.term.move(19, 0) + '\n')
for _ in range(5):
self.assertEqual(self.tty.stdread.readline(), '\n')
self.assertEqual(counter3.calls,
['refresh(flush=False, elapsed=None)'])
def test_set_scroll_area(self):
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter)
manager.counters['dummy'] = 3
term = manager.term
stdread = self.tty.stdread
self.assertEqual(manager.scroll_offset, 1)
self.assertFalse(manager.process_exit)
self.assertNotEqual(signal.getsignal(signal.SIGWINCH), manager._resize_handler)
with mock.patch('enlighten._manager.atexit') as atexit:
with mock.patch.object(term, 'change_scroll'):
manager._set_scroll_area()
self.assertEqual(term.change_scroll.call_count, 1) # pylint: disable=no-member
self.assertEqual(manager.scroll_offset, 4)
self.assertEqual(signal.getsignal(signal.SIGWINCH), manager._resize_handler)
self.assertEqual(stdread.readline(), term.move(24, 0) + '\n')
self.assertEqual(stdread.readline(), '\n')
self.assertEqual(stdread.readline(), '\n')
self.assertTrue(manager.process_exit)
atexit.register.assert_called_with(manager._at_exit)
self.tty.stdout.write('X\n')
self.assertEqual(stdread.readline(), term.move(21, 0) + 'X\n')
# Run it again and make sure exit handling isn't reset
with mock.patch('enlighten._manager.atexit') as atexit:
with mock.patch.object(term, 'change_scroll'):
manager._set_scroll_area(force=True)
self.assertEqual(term.change_scroll.call_count, 1) # pylint: disable=no-member
self.assertFalse(atexit.register.called)
def test_counter_position_pinned(self):
"""If a position is taken, use next available"""
manager = _manager.Manager(stream=self.tty.stdout, set_scroll=False)
counter1 = manager.counter(position=2)
self.assertEqual(manager.counters[counter1], 2)
counter2 = manager.counter()
self.assertEqual(manager.counters[counter1], 2)
self.assertEqual(manager.counters[counter2], 1)
counter3 = manager.counter()
self.assertEqual(manager.counters[counter1], 2)
self.assertEqual(manager.counters[counter2], 3)
self.assertEqual(manager.counters[counter3], 1)
status1 = manager.status_bar(position=3)
self.assertEqual(manager.counters[counter1], 2)
self.assertEqual(manager.counters[counter2], 4)
self.assertEqual(manager.counters[counter3], 1)
self.assertEqual(manager.counters[status1], 3)
status2 = manager.status_bar()
self.assertEqual(manager.counters[counter1], 2)
self.assertEqual(manager.counters[counter2], 5)
self.assertEqual(manager.counters[counter3], 4)
self.assertEqual(manager.counters[status1], 3)
self.assertEqual(manager.counters[status2], 1)
def test_autorefresh(self):
"""
Ensure auto-refreshed counters are updated when others are
"""
manager = _manager.Manager(stream=self.tty.stdout)
counter1 = manager.counter(count=1,
total=0,
counter_format=u'counter1',
autorefresh=True)
counter2 = manager.counter(count=1,
total=0,
counter_format=u'counter2')
self.tty.clear()
# Counter 1 in auto-refresh list
self.assertIn(counter1, manager.autorefresh)
# If auto-refreshed counter hasn't been refreshed recently refresh
counter1.last_update = 0
counter2.refresh()
self.tty.stdout.write(u'X\n')
output = self.tty.stdread.readline()
self.assertRegex(output, 'counter2.+counter1')
# If auto-refreshed counter has been refreshed recently, skip
counter1.last_update = time.time() + 5
counter2.refresh()
self.tty.stdout.write(u'X\n')
output = self.tty.stdread.readline()
self.assertRegex(output, 'counter2')
self.assertNotRegex(output, 'counter1')
def test_set_scroll_area_no_change(self):
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter)
manager.counters['dummy'] = 3
manager.scroll_offset = 4
manager._set_scroll_area()
self.tty.stdout.write('X\n')
self.assertEqual(self.tty.stdread.readline(), manager.term.move(21, 0) + 'X\n')
def test_set_scroll_area_disabled(self):
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter, set_scroll=False)
manager.counters['dummy'] = 3
manager._set_scroll_area()
self.tty.stdout.write('X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
def test_init_hc(self):
# Nonstandard stream doesn't get a companion stream by default
manager = _manager.Manager(stream=OUTPUT)
self.assertIs(manager.stream, OUTPUT)
self.assertIs(manager.term.stream, OUTPUT)
self.assertIsNone(manager.companion_stream)
self.assertIsNone(manager.companion_term)
def test_init_stderr(self):
# Companion stream is stdout if stream is stderr
manager = _manager.Manager(stream=sys.__stderr__)
self.assertIs(manager.stream, sys.__stderr__)
self.assertIs(manager.term.stream, sys.__stderr__)
# This will fail building rpm packages since stderr is redirected
if sys.__stdout__.isatty():
self.assertIs(manager.companion_stream, sys.__stdout__)
self.assertIs(manager.companion_term.stream, sys.__stdout__)
def test_set_scroll_area(self):
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
manager.counters['dummy'] = 3
term = manager.term
stdread = self.tty.stdread
self.assertEqual(manager.scroll_offset, 1)
self.assertFalse(manager.process_exit)
self.assertNotEqual(signal.getsignal(signal.SIGWINCH),
manager._stage_resize)
old_offset = manager.scroll_offset
with mock.patch('enlighten._manager.atexit') as atexit:
manager._set_scroll_area()
self.assertEqual(manager.scroll_offset, 4)
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager._stage_resize)
self.assertTrue(manager.process_exit)
atexit.register.assert_called_with(manager._at_exit)
offset = manager.scroll_offset
scroll_position = term.height - offset
self.assertEqual(manager._buffer, [
term.move(term.height - old_offset, 0), '\n' *
(offset - old_offset), term.hide_cursor,
term.csr(0, scroll_position),
term.move(scroll_position, 0)
])
# No companion buffer defined
self.assertEqual(manager._companion_buffer, [])
# Make sure nothing was flushed
self.tty.stdout.write(u'X\n')
self.assertEqual(stdread.readline(), 'X\n')
# Run it again and make sure exit handling isn't reset
del manager._buffer[:]
del manager._companion_buffer[:]
with mock.patch('enlighten._manager.atexit') as atexit:
manager._set_scroll_area(force=True)
self.assertFalse(atexit.register.called)
self.assertEqual(manager._buffer, [
term.hide_cursor,
term.csr(0, scroll_position),
term.move(scroll_position, 0)
])
# Set max counter lower and make sure scroll_offset hasn't changed
manager.counters['dummy'] = 1
with mock.patch('enlighten._manager.atexit') as atexit:
manager._set_scroll_area()
self.assertEqual(manager.scroll_offset, 4)
def test_context_manager(self):
mgr = None
with _manager.Manager(stream=self.tty.stdout) as manager:
self.assertIsInstance(manager, _manager.Manager)
self.assertTrue(manager.enabled)
mgr = manager
self.assertFalse(mgr.enabled)
def test_init_stderr_redirect(self):
# If stderr is redirected, but stdout is still a tty, use it for companion
with redirect_output('stderr', OUTPUT):
manager = _manager.Manager(stream=sys.stderr)
self.assertIs(manager.stream, sys.stderr)
self.assertIs(manager.term.stream, sys.stderr)
# This will fail building rpm packages since stderr is redirected
if sys.__stdout__.isatty():
self.assertIs(manager.companion_stream, sys.stdout)
self.assertIs(manager.companion_term.stream, sys.stdout)
def test_counter_position(self):
manager = _manager.Manager(stream=self.tty.stdout, set_scroll=False)
counter1 = manager.counter(position=4)
self.assertEqual(manager.counters[counter1], 4)
with self.assertRaisesRegex(ValueError, 'Counter position 4 is already occupied'):
manager.counter(position=4)
with self.assertRaisesRegex(ValueError,
'Counter position 200 is greater than terminal height'):
manager.counter(position=200)
def test_write(self):
msg = 'test message'
with mock.patch('enlighten._manager.Manager._set_scroll_area') as ssa:
manager = _manager.Manager(stream=self.tty.stdout)
term = manager.term
manager.write(msg, position=3)
self.tty.stdout.write('X\n')
# Carriage return is getting converted to newline
self.assertEqual(self.tty.stdread.readline(),
term.move(22, 0) + '\r' + term.clear_eol + msg + 'X\n')
self.assertEqual(ssa.call_count, 1)
def test_stop(self):
with mock.patch('%s.reset' % TERMINAL) as reset:
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
manager.counters[MockCounter(manager=manager)] = 3
manager.counters[MockCounter(manager=manager)] = 4
term = manager.term
self.assertIsNone(manager.companion_term)
with mock.patch('enlighten._manager.atexit'):
with mock.patch.object(term, 'change_scroll'):
manager._set_scroll_area()
self.assertEqual(manager.scroll_offset, 5)
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager._resize_handler)
self.assertTrue(manager.process_exit)
# Clear stream
self.tty.stdout.write(u'X\n')
for num in range(4 + 1): # pylint: disable=unused-variable
self.tty.stdread.readline()
self.assertFalse(reset.called)
manager.enabled = False
manager.stop()
# No output, No changes
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager._resize_handler)
self.assertTrue(manager.process_exit)
manager.enabled = True
manager.stop()
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager.sigwinch_orig)
self.assertEqual(reset.call_count, 1)
self.tty.stdout.write(u'X\n')
self.assertEqual(
self.tty.stdread.readline(),
term.move(23, 0) + term.clear_eol + term.move(24, 0) +
term.clear_eol + 'X\n')
self.assertFalse(manager.process_exit)
self.assertFalse(manager.enabled)
for counter in manager.counters:
self.assertFalse(counter.enabled)
def test_inherit_kwargs(self):
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter,
unit='knights', not_real=True, desc='Default')
self.assertTrue('unit' in manager.defaults)
self.assertTrue('desc' in manager.defaults)
self.assertTrue('not_real' in manager.defaults)
with mock.patch.object(manager, '_set_scroll_area'):
ctr = manager.counter(desc='Huzzah')
self.assertEqual(ctr.unit, 'knights')
self.assertEqual(ctr.desc, 'Huzzah')
self.assertFalse(hasattr(ctr, 'not_real'))
def test_set_scroll_area_companion(self):
"""
Ensure when no change is made, a term.move is still called for the companion stream
"""
manager = _manager.Manager(stream=self.tty.stdout, counter_class=MockCounter,
companion_stream=self.tty.stdout)
manager.counters['dummy'] = 3
manager.scroll_offset = 4
term = manager.term
manager._set_scroll_area()
self.tty.stdout.write('X\n')
self.assertEqual(self.tty.stdread.readline(),
term.move(21, 0) + term.move(21, 0) + 'X\n')
def test_stop(self):
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
manager.counters[MockCounter(manager=manager)] = 3
manager.counters[MockCounter(manager=manager)] = 4
term = manager.term
self.assertIsNone(manager.companion_term)
with mock.patch('enlighten._manager.atexit'):
manager._set_scroll_area()
self.assertEqual(manager.scroll_offset, 5)
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager._stage_resize)
self.assertTrue(manager.process_exit)
# Clear buffer
del manager._buffer[:]
manager.enabled = False
manager.stop()
# No output, No changes
self.tty.stdout.write(u'X\n')
self.assertEqual(self.tty.stdread.readline(), 'X\n')
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager._stage_resize)
self.assertTrue(manager.process_exit)
manager.enabled = True
manager.stop()
self.assertEqual(signal.getsignal(signal.SIGWINCH),
manager.sigwinch_orig)
self.tty.stdout.write(u'X\n')
self.assertEqual(
self.tty.stdread.readline(),
term.move(term.height - 2, 0) + term.clear_eol +
term.move(term.height - 1, 0) + term.clear_eol +
term.normal_cursor + term.csr(0, term.height - 1) +
term.move(term.height, 0) + 'X\n')
self.assertFalse(manager.process_exit)
self.assertFalse(manager.enabled)
for counter in manager.counters:
self.assertFalse(counter.enabled)
def test_threaded_eval(self):
"""
Dynamic value for threaded determined when scroll area is first set
"""
# Not dynamic if explicitly True
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter,
threaded=True)
self.assertTrue(manager.threaded)
with mock.patch('threading.active_count', return_value=4):
manager.counter()
self.assertTrue(manager.threaded)
# Not dynamic if explicitly False
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter,
threaded=False)
self.assertFalse(manager.threaded)
with mock.patch('threading.active_count', return_value=4):
manager.counter()
self.assertFalse(manager.threaded)
# False by default
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
self.assertIsNone(manager.threaded)
manager.counter()
self.assertFalse(manager.threaded)
# True if threaded
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
self.assertIsNone(manager.threaded)
with mock.patch('threading.active_count', return_value=4):
manager.counter()
self.assertTrue(manager.threaded)
# True if has child processes
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
self.assertIsNone(manager.threaded)
with mock.patch('multiprocessing.active_children', return_value=[1,
2]):
manager.counter()
self.assertTrue(manager.threaded)
# True if is child processes
manager = _manager.Manager(stream=self.tty.stdout,
counter_class=MockCounter)
self.assertIsNone(manager.threaded)
with mock.patch('multiprocessing.current_process') as c_process:
c_process.name = 'Process1'
manager.counter()
self.assertTrue(manager.threaded)