OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	03/31/2022 06:22:39 AM	rwxr-xr-x
📄 __init__.py	14 bytes	09/08/2017 10:38:36 AM	rw-r--r--
📁 __pycache__	-	03/31/2022 06:22:40 AM	rwxr-xr-x
📄 test_basechat.py	2.45 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_basesupport.py	2.99 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_domish.py	19.7 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_irc.py	100.97 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_irc_service.py	9.41 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_ircsupport.py	10.45 KB	09/08/2017 10:38:35 AM	rw-r--r--
📄 test_jabberclient.py	16.47 KB	03/22/2022 11:03:56 AM	rw-r--r--
📄 test_jabbercomponent.py	13.87 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_jabbererror.py	11.25 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_jabberjid.py	7.04 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_jabberjstrports.py	996 bytes	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_jabbersasl.py	9.21 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_jabbersaslmechanisms.py	5.62 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_jabberxmlstream.py	44.65 KB	03/22/2022 11:03:56 AM	rw-r--r--
📄 test_jabberxmppstringprep.py	5.42 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_service.py	27.79 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_tap.py	2.13 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_xishutil.py	9.18 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_xmlstream.py	6.13 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_xmpproutertap.py	2.34 KB	09/08/2017 10:38:36 AM	rw-r--r--
📄 test_xpath.py	10.74 KB	09/08/2017 10:38:36 AM	rw-r--r--

Editing: test_basesupport.py

# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

from twisted.trial import unittest
from twisted.words.im import basesupport
from twisted.internet import error, defer

class DummyAccount(basesupport.AbstractAccount):
    """
    An account object that will do nothing when asked to start to log on.
    """

loginHasFailed = False
    loginCallbackCalled = False

def _startLogOn(self, *args):
        """
        Set self.loginDeferred to the same as the deferred returned, allowing a
        testcase to .callback or .errback.
        
        @return: A deferred.
        """
        self.loginDeferred = defer.Deferred()
        return self.loginDeferred

def _loginFailed(self, result):
        self.loginHasFailed = True
        return basesupport.AbstractAccount._loginFailed(self, result)

def _cb_logOn(self, result):
        self.loginCallbackCalled = True
        return basesupport.AbstractAccount._cb_logOn(self, result)

class DummyUI(object):
    """
    Provide just the interface required to be passed to AbstractAccount.logOn.
    """
    clientRegistered = False

def registerAccountClient(self, result): 
        self.clientRegistered = True

class ClientMsgTests(unittest.TestCase):
    def makeUI(self):
        return DummyUI()

def makeAccount(self):
        return DummyAccount('la', False, 'la', None, 'localhost', 6667)

def test_connect(self):
        """
        Test that account.logOn works, and it calls the right callback when a
        connection is established.
        """
        account = self.makeAccount()
        ui = self.makeUI()
        d = account.logOn(ui)
        account.loginDeferred.callback(None)

def check(result):
            self.assertFalse(account.loginHasFailed,
                    "Login shouldn't have failed")
            self.assertTrue(account.loginCallbackCalled,
                    "We should be logged in")
        d.addCallback(check)
        return d

def test_failedConnect(self):
        """
        Test that account.logOn works, and it calls the right callback when a
        connection is established.
        """
        account = self.makeAccount()
        ui = self.makeUI()
        d = account.logOn(ui)
        account.loginDeferred.errback(Exception())

def err(reason):
            self.assertTrue(account.loginHasFailed, "Login should have failed")
            self.assertFalse(account.loginCallbackCalled,
                    "We shouldn't be logged in")
            self.assertTrue(not ui.clientRegistered,
                    "Client shouldn't be registered in the UI")
        cb = lambda r: self.assertTrue(False, "Shouldn't get called back")
        d.addCallbacks(cb, err)
        return d

def test_alreadyConnecting(self):
        """
        Test that it can fail sensibly when someone tried to connect before
        we did. 
        """
        account = self.makeAccount()
        ui = self.makeUI()
        account.logOn(ui)
        self.assertRaises(error.ConnectError, account.logOn, ui)