27 lines
953 B
Python
27 lines
953 B
Python
#!/usr/bin/env python
|
|
|
|
import unittest
|
|
from planet.splice import splice, config
|
|
|
|
configfile = 'tests/data/splice/config.ini'
|
|
|
|
class SpliceTest(unittest.TestCase):
|
|
|
|
def test_splice(self):
|
|
config.load(configfile)
|
|
doc = splice()
|
|
self.assertEqual(12,len(doc.getElementsByTagName('entry')))
|
|
self.assertEqual(4,len(doc.getElementsByTagName('planet:source')))
|
|
self.assertEqual(16,len(doc.getElementsByTagName('planet:name')))
|
|
|
|
self.assertEqual('test planet',
|
|
doc.getElementsByTagName('title')[0].firstChild.nodeValue)
|
|
|
|
def test_splice_unsub(self):
|
|
config.load(configfile)
|
|
config.parser.remove_section('tests/data/spider/testfeed2.atom')
|
|
doc = splice()
|
|
self.assertEqual(8,len(doc.getElementsByTagName('entry')))
|
|
self.assertEqual(3,len(doc.getElementsByTagName('planet:source')))
|
|
self.assertEqual(11,len(doc.getElementsByTagName('planet:name')))
|