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()