root / trunk / twisted / news / test / test_news.py

Revision 15496, 3.1 kB (checked in by exarkun, 4 years ago)

Remove reactor.iterate() usage; resolves 1187

Line 
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
Note: See TracBrowser for help on using the browser.