← Back to branch summary

~justin-fathomdb/nova/justinsb-openstack-api-volumes

« back to all changes in this revision

Viewing changes to vendor/Twisted-10.0.0/twisted/test/test_postfix.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
vendor/Twisted-10.0.0/twisted/test/test_postfix.py
 
1
# Copyright (c) 2001-2009 Twisted Matrix Laboratories.
 
2
# See LICENSE for details.
 
3
 
 
4
"""
 
5
Test cases for twisted.protocols.postfix module.
 
6
"""
 
7
 
 
8
from twisted.trial import unittest
 
9
from twisted.protocols import postfix
 
10
from twisted.test.proto_helpers import StringTransport
 
11
 
 
12
 
 
13
class PostfixTCPMapQuoteTestCase(unittest.TestCase):
 
14
    data = [
 
15
        # (raw, quoted, [aliasQuotedForms]),
 
16
        ('foo', 'foo'),
 
17
        ('foo bar', 'foo%20bar'),
 
18
        ('foo\tbar', 'foo%09bar'),
 
19
        ('foo\nbar', 'foo%0Abar', 'foo%0abar'),
 
20
        ('foo\r\nbar', 'foo%0D%0Abar', 'foo%0D%0abar', 'foo%0d%0Abar', 'foo%0d%0abar'),
 
21
        ('foo ', 'foo%20'),
 
22
        (' foo', '%20foo'),
 
23
        ]
 
24
 
 
25
    def testData(self):
 
26
        for entry in self.data:
 
27
            raw = entry[0]
 
28
            quoted = entry[1:]
 
29
 
 
30
            self.assertEquals(postfix.quote(raw), quoted[0])
 
31
            for q in quoted:
 
32
                self.assertEquals(postfix.unquote(q), raw)
 
33
 
 
34
class PostfixTCPMapServerTestCase:
 
35
    data = {
 
36
        # 'key': 'value',
 
37
        }
 
38
 
 
39
    chat = [
 
40
        # (input, expected_output),
 
41
        ]
 
42
 
 
43
    def test_chat(self):
 
44
        """
 
45
        Test that I{get} and I{put} commands are responded to correctly by
 
46
        L{postfix.PostfixTCPMapServer} when its factory is an instance of
 
47
        L{postifx.PostfixTCPMapDictServerFactory}.
 
48
        """
 
49
        factory = postfix.PostfixTCPMapDictServerFactory(self.data)
 
50
        transport = StringTransport()
 
51
 
 
52
        protocol = postfix.PostfixTCPMapServer()
 
53
        protocol.service = factory
 
54
        protocol.factory = factory
 
55
        protocol.makeConnection(transport)
 
56
 
 
57
        for input, expected_output in self.chat:
 
58
            protocol.lineReceived(input)
 
59
            self.assertEquals(
 
60
                transport.value(), expected_output,
 
61
                'For %r, expected %r but got %r' % (
 
62
                    input, expected_output, transport.value()))
 
63
            transport.clear()
 
64
        protocol.setTimeout(None)
 
65
 
 
66
 
 
67
    def test_deferredChat(self):
 
68
        """
 
69
        Test that I{get} and I{put} commands are responded to correctly by
 
70
        L{postfix.PostfixTCPMapServer} when its factory is an instance of
 
71
        L{postifx.PostfixTCPMapDeferringDictServerFactory}.
 
72
        """
 
73
        factory = postfix.PostfixTCPMapDeferringDictServerFactory(self.data)
 
74
        transport = StringTransport()
 
75
 
 
76
        protocol = postfix.PostfixTCPMapServer()
 
77
        protocol.service = factory
 
78
        protocol.factory = factory
 
79
        protocol.makeConnection(transport)
 
80
 
 
81
        for input, expected_output in self.chat:
 
82
            protocol.lineReceived(input)
 
83
            self.assertEquals(
 
84
                transport.value(), expected_output,
 
85
                'For %r, expected %r but got %r' % (
 
86
                    input, expected_output, transport.value()))
 
87
            transport.clear()
 
88
        protocol.setTimeout(None)
 
89
 
 
90
 
 
91
 
 
92
class Valid(PostfixTCPMapServerTestCase, unittest.TestCase):
 
93
    data = {
 
94
        'foo': 'ThisIs Foo',
 
95
        'bar': ' bar really is found\r\n',
 
96
        }
 
97
    chat = [
 
98
        ('get', "400 Command 'get' takes 1 parameters.\n"),
 
99
        ('get foo bar', "500 \n"),
 
100
        ('put', "400 Command 'put' takes 2 parameters.\n"),
 
101
        ('put foo', "400 Command 'put' takes 2 parameters.\n"),
 
102
        ('put foo bar baz', "500 put is not implemented yet.\n"),
 
103
        ('put foo bar', '500 put is not implemented yet.\n'),
 
104
        ('get foo', '200 ThisIs%20Foo\n'),
 
105
        ('get bar', '200 %20bar%20really%20is%20found%0D%0A\n'),
 
106
        ('get baz', '500 \n'),
 
107
        ('foo', '400 unknown command\n'),
 
108
        ]