summaryrefslogtreecommitdiff
path: root/tests/test_streamtester.py
blob: 8512f40112a6f00b776d592b3e5b575a41fdb314 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from . sleektest import *
import sleekxmpp.plugins.xep_0033 as xep_0033


class TestStreamTester(SleekTest):
    """
    Test that we can simulate and test a stanza stream.
    """

    def tearDown(self):
        self.streamClose()

    def testClientEcho(self):
        """Test that we can interact with a ClientXMPP instance."""
        self.streamStart(mode='client')

        def echo(msg):
            msg.reply('Thanks for sending: %(body)s' % msg).send()
        
        self.xmpp.add_event_handler('message', echo)
        
        self.streamRecv("""
          <message to="tester@localhost" from="user@localhost">
            <body>Hi!</body>
          </message>
        """)
        
        self.streamSendMessage("""
          <message to="user@localhost">
            <body>Thanks for sending: Hi!</body>
          </message>
        """)

    def testComponentEcho(self):
        """Test that we can interact with a ComponentXMPP instance."""
        self.streamStart(mode='component')

        def echo(msg):
            msg.reply('Thanks for sending: %(body)s' % msg).send()

        self.xmpp.add_event_handler('message', echo)

        self.streamRecv("""
          <message to="tester.localhost" from="user@localhost">
            <body>Hi!</body>
          </message>
        """)

        self.streamSendMessage("""
          <message to="user@localhost" from="tester.localhost">
            <body>Thanks for sending: Hi!</body>
          </message>
        """)

suite = unittest.TestLoader().loadTestsFromTestCase(TestStreamTester)