← Back to branch summary

~certify-web-dev/twisted/certify-trunk

« back to all changes in this revision

Viewing changes to twisted/news/test/test_news.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2007-01-17 14:52:35 UTC
  • mfrom: (1.1.5 upstream) (2.1.2 etch)
  • Revision ID: james.westby@ubuntu.com-20070117145235-btmig6qfmqfen0om
Tags: 2.5.0-0ubuntu1
New upstream version, compatible with python2.5.

Show diffs side-by-side

added added

removed removed

Lines of Context:
twisted/news/test/test_news.py
 
1
# Copyright (c) 2001-2004 Twisted Matrix Laboratories.
 
2
# See LICENSE for details.
 
3
 
 
4
import sys, types
 
5
from pprint import pformat
 
6
 
 
7
from twisted.trial import unittest
 
8
from twisted.news import database
 
9
from twisted.internet import reactor
 
10
 
 
11
MESSAGE_ID = "f83ba57450ed0fd8ac9a472b847e830e"
 
12
 
 
13
POST_STRING = """Path: not-for-mail
 
14
From: <exarkun@somehost.domain.com>
 
15
Subject: a test
 
16
Newsgroups: alt.test.nntp
 
17
Organization: 
 
18
Summary: 
 
19
Keywords: 
 
20
Message-Id: %s
 
21
User-Agent: tin/1.4.5-20010409 ("One More Nightmare") (UNIX) (Linux/2.4.17 (i686))
 
22
 
 
23
this is a test
 
24
...
 
25
lala
 
26
moo
 
27
-- 
 
28
"One World, one Web, one Program." - Microsoft(R) promotional ad
 
29
"Ein Volk, ein Reich, ein Fuhrer." - Adolf Hitler
 
30
--
 
31
 10:56pm up 4 days, 4:42, 1 user, load average: 0.08, 0.08, 0.12
 
32
""" % (MESSAGE_ID)
 
33
 
 
34
class NewsTestCase(unittest.TestCase):
 
35
    def setUp(self):
 
36
        self.backend = database.NewsShelf(None, 'news2.db')
 
37
        self.backend.addGroup('alt.test.nntp', 'y')
 
38
        self.backend.postRequest(POST_STRING.replace('\n', '\r\n'))
 
39
 
 
40
 
 
41
    def testArticleExists(self):
 
42
        d = self.backend.articleExistsRequest(MESSAGE_ID)
 
43
        d.addCallback(self.failUnless)
 
44
        return d
 
45
 
 
46
 
 
47
    def testArticleRequest(self):
 
48
        d = self.backend.articleRequest(None, None, MESSAGE_ID)
 
49
 
 
50
        def cbArticle(result):
 
51
            self.failUnless(isinstance(result, tuple),
 
52
                            'callback result is wrong type: ' + str(result))
 
53
            self.assertEquals(len(result), 3,
 
54
                              'callback result list should have three entries: ' +
 
55
                              str(result))
 
56
            self.assertEquals(result[1], MESSAGE_ID,
 
57
                              "callback result Message-Id doesn't match: %s vs %s" %
 
58
                              (MESSAGE_ID, result[1]))
 
59
            body = result[2].read()
 
60
            self.failIfEqual(body.find('\r\n\r\n'), -1,
 
61
                             "Can't find \\r\\n\\r\\n between header and body")
 
62
            return result
 
63
 
 
64
        d.addCallback(cbArticle)
 
65
        return d
 
66
 
 
67
 
 
68
    def testHeadRequest(self):
 
69
        d = self.testArticleRequest()
 
70
 
 
71
        def cbArticle(result):
 
72
            index = result[0]
 
73
 
 
74
            d = self.backend.headRequest("alt.test.nntp", index)
 
75
            d.addCallback(cbHead)
 
76
            return d
 
77
 
 
78
        def cbHead(result):
 
79
            self.assertEquals(result[1], MESSAGE_ID,
 
80
                              "callback result Message-Id doesn't match: %s vs %s" %
 
81
                              (MESSAGE_ID, result[1]))
 
82
 
 
83
            self.assertEquals(result[2][-2:], '\r\n',
 
84
                              "headers must be \\r\\n terminated.")
 
85
 
 
86
        d.addCallback(cbArticle)
 
87
        return d
 
88
 
 
89
 
 
90
    def testBodyRequest(self):
 
91
        d = self.testArticleRequest()
 
92
 
 
93
        def cbArticle(result):
 
94
            index = result[0]
 
95
 
 
96
            d = self.backend.bodyRequest("alt.test.nntp", index)
 
97
            d.addCallback(cbBody)
 
98
            return d
 
99
 
 
100
        def cbBody(result):
 
101
            body = result[2].read()
 
102
            self.assertEquals(body[0:4], 'this',
 
103
                              "message body has been altered: " +
 
104
                              pformat(body[0:4]))
 
105
 
 
106
        d.addCallback(cbArticle)
 
107
        return d