def test_handle_message_with_authentication_started_and_verification_fails(self): msg = self.mocker.mock() msg['body'] self.mocker.result("123456") msg.getFrom().jid self.mocker.result("[email protected]/Adium123") account = self.mocker.mock() account.verified self.mocker.result(False) account.authenticating self.mocker.result(True) account.verify("123456") self.mocker.result(False) manager = self.mocker.mock() manager.get_or_create_account("[email protected]/Adium123") self.mocker.result(account) send_message = self.mocker.mock() send_message("[email protected]/Adium123", "Invalid verification code") self.mocker.replay() handler = MessageHandler() handler.manager = manager handler.send_message = send_message handler.handle(msg) self.mocker.verify()
def test_handle_message_from_authenticated_user(self): msg = self.mocker.mock() msg['body'] self.mocker.result("timeline") msg.getFrom().jid self.mocker.result("[email protected]/Adium123") account = self.mocker.mock() account.verified self.mocker.result(True) manager = self.mocker.mock() manager.get_or_create_account("[email protected]/Adium123") self.mocker.result(account) handler = MessageHandler() handler.manager = manager handler.execute_command = self.mocker.mock() handler.execute_command(account, "timeline") self.mocker.replay() handler.handle(msg) self.mocker.verify()
def test_handle_message_to_start_authentication(self): msg = self.mocker.mock() msg['body'] self.mocker.result("start") msg.getFrom().jid self.mocker.result("[email protected]/Adium123") account = self.mocker.mock() account.verified self.mocker.result(False) account.authenticating self.mocker.result(False) account.authenticate() self.mocker.result("/authenticate/url") manager = self.mocker.mock() manager.get_or_create_account("[email protected]/Adium123") self.mocker.result(account) send_message = self.mocker.mock() send_message("[email protected]/Adium123", mocker.ARGS) send_message("[email protected]/Adium123", "/authenticate/url") send_message("[email protected]/Adium123", mocker.ARGS) self.mocker.replay() handler = MessageHandler() handler.manager = manager handler.send_message = send_message handler.handle(msg) self.mocker.verify()
def test_handle_message_with_authentication_started_and_verification_succeed(self): msg = self.mocker.mock() msg['body'] self.mocker.result("123456") msg.get_from() self.mocker.result("[email protected]/Adium123") account = self.mocker.mock() account.verified self.mocker.result(False) account.reload_authentication() self.mocker.result(False) account.authenticating self.mocker.result(True) account.verify("123456") self.mocker.result(True) account.save() manager = self.mocker.mock() manager.get_or_create_account("[email protected]/Adium123") self.mocker.result(account) send_message = self.mocker.mock() send_message("[email protected]/Adium123", "Authentication complete!") self.mocker.replay() handler = MessageHandler() handler.manager = manager handler.send_message = send_message handler.handle(msg) self.mocker.verify()