summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--pym/portage/tests/process/test_poll.py12
1 files changed, 6 insertions, 6 deletions
diff --git a/pym/portage/tests/process/test_poll.py b/pym/portage/tests/process/test_poll.py
index 579acf907..9df2e88ae 100644
--- a/pym/portage/tests/process/test_poll.py
+++ b/pym/portage/tests/process/test_poll.py
@@ -3,7 +3,7 @@
from portage import os
from portage.tests import TestCase
-from _emerge.TaskScheduler import TaskScheduler
+from _emerge.PollScheduler import PollScheduler
from _emerge.PipeReader import PipeReader
from _emerge.SpawnProcess import SpawnProcess
@@ -18,22 +18,22 @@ class PipeReaderTestCase(TestCase):
test_string = 2 * "blah blah blah\n"
- task_scheduler = TaskScheduler()
+ scheduler = PollScheduler().sched_iface
master_fd, slave_fd = os.pipe()
master_file = os.fdopen(master_fd, 'rb')
slave_file = os.fdopen(slave_fd, 'wb')
producer = SpawnProcess(
args=["bash", "-c", "echo -n '%s'" % test_string],
env=os.environ, fd_pipes={1:slave_fd},
- scheduler=task_scheduler.sched_iface)
+ scheduler=scheduler)
producer.start()
slave_file.close()
consumer = PipeReader(
input_files={"producer" : master_file},
- scheduler=task_scheduler.sched_iface)
+ scheduler=scheduler)
- task_scheduler.add(consumer)
- task_scheduler.run()
+ consumer.start()
+ consumer.wait()
output = consumer.getvalue().decode('ascii', 'replace')
self.assertEqual(test_string, output)