/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/per_uifactory/__init__.py

  • Committer: Jelmer Vernooij
  • Date: 2018-03-24 17:48:04 UTC
  • mfrom: (6921 work)
  • mto: This revision was merged to the branch mainline in revision 6923.
  • Revision ID: jelmer@jelmer.uk-20180324174804-xf22o05byoj12x1q
Merge trunk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/per_uifactory/__init__.py
44
44
from ..ui_testing import (
45
45
    StringIOWithEncoding,
46
46
    StringIOAsTTY,
 
47
    TextUIFactory,
47
48
    )
48
49
 
49
50
 
124
125
 
125
126
    def setUp(self):
126
127
        super(TestTextUIFactory, self).setUp()
127
 
        self.stdin = StringIOWithEncoding()
128
 
        self.stdout = StringIOWithEncoding()
129
 
        self.stderr = StringIOWithEncoding()
130
 
        self.factory = ui.text.TextUIFactory(self.stdin, self.stdout,
131
 
            self.stderr)
 
128
        self.factory = self._create_ui_factory()
 
129
        self.factory.__enter__()
 
130
        self.addCleanup(self.factory.__exit__, None, None, None)
 
131
        self.stdin = self.factory.stdin
 
132
        self.stdout = self.factory.stdout
 
133
        self.stderr = self.factory.stderr
 
134
 
 
135
    def _create_ui_factory(self):
 
136
        return TextUIFactory(u'')
132
137
 
133
138
    def _check_note(self, note_text):
134
139
        self.assertEqual("%s\n" % note_text,
172
177
 
173
178
class TestTTYTextUIFactory(TestTextUIFactory):
174
179
 
175
 
    def setUp(self):
176
 
        super(TestTTYTextUIFactory, self).setUp()
177
 
 
 
180
    def _create_ui_factory(self):
178
181
        # Remove 'TERM' == 'dumb' which causes us to *not* treat output as a
179
182
        # real terminal, even though isatty returns True
180
183
        self.overrideEnv('TERM', None)
181
 
        self.stderr = StringIOAsTTY()
182
 
        self.stdout = StringIOAsTTY()
183
 
        self.factory = ui.text.TextUIFactory(self.stdin, self.stdout,
184
 
            self.stderr)
 
184
        return TextUIFactory(u'', StringIOAsTTY(), StringIOAsTTY())
185
185
 
186
186
    def _check_log_transport_activity_display(self):
187
187
        self.assertEqual('', self.stdout.getvalue())