Fixed broken tests when running with a non-TTY stdin.

This commit is contained in:
Baptiste Mispelon 2014-02-18 00:07:11 +01:00
parent 2a9ee49f3c
commit 6b310bafc5
1 changed files with 13 additions and 2 deletions

View File

@ -62,6 +62,15 @@ def mock_inputs(inputs):
return inner return inner
class MockTTY(object):
"""
A fake stdin object that pretends to be a TTY to be used in conjunction
with mock_inputs.
"""
def isatty(self):
return True
@skipIfCustomUser @skipIfCustomUser
class GetDefaultUsernameTestCase(TestCase): class GetDefaultUsernameTestCase(TestCase):
@ -180,7 +189,8 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
interactive=True, interactive=True,
username="nolocale@somewhere.org", username="nolocale@somewhere.org",
email="nolocale@somewhere.org", email="nolocale@somewhere.org",
verbosity=0 verbosity=0,
stdin=MockTTY(),
) )
except TypeError: except TypeError:
@ -209,7 +219,8 @@ class CreatesuperuserManagementCommandTestCase(TestCase):
call_command( call_command(
"createsuperuser", "createsuperuser",
interactive=True, interactive=True,
stdout=new_io stdout=new_io,
stdin=MockTTY(),
) )
finally: finally:
username_field.verbose_name = old_verbose_name username_field.verbose_name = old_verbose_name