| OLD | NEW |
| (Empty) |
| 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 | |
| OLD | NEW |