def testGetAuthToken(self):
"""Test successful GetAuthToken with defaults."""
self.StubSetup()
getauth.getpass.getpass('Password: '******'password')
self.mox.ReplayAll()
self.assertEqual('password', getauth.GetAuthToken())
self.mox.VerifyAll()
def testGetAuthTokenWithValidation(self):
"""Test successful GetAuthToken with a validator."""
validator = re.compile(r'\d{6}')
self.StubSetup()
getauth.getpass.getpass('Password: '******'123456')
self.mox.ReplayAll()
self.assertEqual('123456', getauth.GetAuthToken(validator=validator))
self.mox.VerifyAll()
def testGetAuthTokenGUIEmptyData(self):
"""Test successful GetAuthToken with GUI with no data."""
mock_cocoainput = self.mox.CreateMockAnything()
self.StubSetup()
getauth.cocoadialog.Standard_InputBox().AndReturn(mock_cocoainput)
mock_cocoainput.SetPasswordBox().AndReturn(None)
mock_cocoainput.Show().AndReturn('button\n\n')
self.mox.ReplayAll()
self.assertEqual('', getauth.GetAuthToken(gui=True))
self.mox.VerifyAll()
def testGetAuthTokenGUIWithValidation(self):
"""Test successful GetAuthToken with GUI with a validator."""
validator = re.compile(r'\d{6}')
mock_cocoainput = self.mox.CreateMockAnything()
self.StubSetup()
getauth.cocoadialog.Standard_InputBox().AndReturn(mock_cocoainput)
mock_cocoainput.SetPasswordBox().AndReturn(None)
mock_cocoainput.Show().AndReturn('button\npassword\n')
getauth.cocoadialog.Standard_InputBox().AndReturn(mock_cocoainput)
mock_cocoainput.SetPasswordBox().AndReturn(None)
mock_cocoainput.Show().AndReturn('button\n123456\n')
self.mox.ReplayAll()
self.assertEqual('123456',
getauth.GetAuthToken(gui=True, validator=validator))
self.mox.VerifyAll()