Esempio n. 1
0
 def setUp(self):
     topic.http = mockhttp
     self.host = "kittywalk.sourceforge.jp"
     self.path = "test"
     self.id = "0000000001"
     self.info_file_path = os.path.join(config.CACHEDIR,
                                        self.host,
                                        self.path,
                                        self.id)
     self.topic = Topic(self.host, self.path, self.id)
Esempio n. 2
0
class TopicTestCase(unittest.TestCase):
    def setUp(self):
        topic.http = mockhttp
        self.host = "kittywalk.sourceforge.jp"
        self.path = "test"
        self.id = "0000000001"
        self.info_file_path = os.path.join(config.CACHEDIR,
                                           self.host,
                                           self.path,
                                           self.id)
        self.topic = Topic(self.host, self.path, self.id)

    def tearDown(self):
        topic.http = http
        if os.path.exists(self.info_file_path):
            os.remove(self.info_file_path)
        if os.path.exists(self.topic._data_file_path):
            os.remove(self.topic._data_file_path)

    def test_title(self):
        """Topic.title() extracts HTML entity"""
        self.topic._title = ">"
        self.assertEqual(self.topic.title(), u">")

    def test_summary(self):
        """Summary is modified after update() call"""
        size = 23
        content = "name<>mail<>time<>body<>\n" * size
        mockhttp.set_data([{"status":200,
                            "content":content,
                            "last-modified":"now"}])
        self.topic.update()
        s = board.Summary(self.topic)
        self.assertEqual(s.get(self.topic,"size"), size)

    def test_equality(self):
        """topic1==topic2 if url match"""
        t1 = Topic(self.host, self.path, self.id, title="foo")
        t2 = Topic(self.host, self.path, self.id, title="bar")
        self.assert_(t1==t2)
    
    def test_messages_empty(self):
        """Topic.messages() returns [] if local file does not exist"""
        if os.path.exists(self.topic._data_file_path):
            os.remove(self.topic._data_file_path)
        self.assertEqual([m for m in self.topic.messages()], [])

    def test_messages_non_empty(self):
        """Topic.messages() returns a list of Message instances"""
        name1,mail1,time1,body1 = "1","2","3","4"
        name2,mail2,time2,body2 = "5","6","7","8"
        data = "%s<>%s<>%s<>%s<>title\n" % (name1,mail1,time1,body1) +\
               "%s<>%s<>%s<>%s<>" % (name2,mail2,time2,body2)
        file(self.topic._data_file_path,"w").write(data)
        m1 = Message(1, name1,mail1,time1,body1)
        m2 = Message(2, name2,mail2,time2,body2)
        self.assertEqual([m for m in self.topic.messages()], [m1, m2])
        
    def test_update_with_old_data_full(self):
        """Topic.update() downloads and save data"""
        old_data = "old"
        new_data = "new"
        file(self.topic._data_file_path,"w").write(old_data)
        mockhttp.set_data([{"status":http.OK,"content":"new"}])
        self.topic.update()
        self.assertEqual(file(self.topic._data_file_path,"r").read(),
                         new_data)

    def test_update_with_old_data_partial(self):
        """Topic.update() downloads partial data and save data"""
        old_data = "old"
        new_data = "new"
        file(self.topic._data_file_path,"w").write(old_data)
        mockhttp.set_data([{"status":http.PARTIAL_CONTENT,"content":"new"}])
        self.topic.update()
        self.assertEqual(file(self.topic._data_file_path,"r").read(),
                         old_data+new_data)

    def test_update_with_latest_data(self):
        """Topic.update() does not download if data file is up to date"""
        current_data = "current_data"
        file(self.topic._data_file_path,"w").write(current_data)
        mockhttp.set_data([{"status":http.NOT_MODIFIED,"content":"wrong"}])
        self.topic.update()
        self.assertEqual(file(self.topic._data_file_path,"r").read(),
                         current_data)

    def test_update_without_datafile(self):
        """Topic.update() downloads from bbs if data file does not exist"""
        f = self.topic._data_file_path
        if os.path.exists(f):
            os.remove(f)
        mockhttp.set_data([{"status":200,
                            "content":"hello",
                            "last-modified":"now"}])
        self.topic.update()
        self.assert_(os.path.exists(f))

    def test_update_error(self):
        """Topic.update() raises DownloadError when bad status"""
        mockhttp.set_data([{"status":301,
                            "content":"foo",
                            "last-modified":"now"}])
        self.assertRaises(topic.DownloadError, self.topic.update)
  
    def test_save_info(self):
        """Topic._save_info(info)"""
        d = {"last-modified":"now"}
        if os.path.exists(self.info_file_path):
            os.remove(self.info_file_path)
        self.topic._save_info(d)
        self.assertEqual(marshal.load(file(self.info_file_path,"r")), d)

    def test_save_info_empty(self):
        """Topic._save_info({}) does not create file"""
        if os.path.exists(self.info_file_path):
            os.remove(self.info_file_path)
        self.topic._save_info({})
        self.failIf(os.path.exists(self.info_file_path))

    def test_load_info(self):
        """Topic._load_info() returns saved info"""
        info = {"key":"val"}
        marshal.dump(info, file(self.topic._info_file_path,"w"))
        self.assertEqual(self.topic._load_info(),info)

    def test_load_info_empty(self):
        """Topic._load_info() returns {} if info is not saved"""
        if os.path.exists(self.topic._info_file_path):
            os.remove(self.topic._info_file_path)
        self.assertEqual(self.topic._load_info(),{})