test_mangler.py

import unittest
import time

try:
    import xml.etree.cElementTree as ET
except:
    import xml.etree.ElementTree as ET

#from newsmangler import asyncnntp

from newsmangler.article import Article
from newsmangler.common import *


class DummyFileWrapper:
    def __init__(self):
        self.set_returnStr()
    
    def set_returnStr(self, newByteString = b'Hello world'):
        self.returnStr = newByteString
    
    def read_part(self, begin, end):
        return self.returnStr

class TestArticle(unittest.TestCase):    
    def setUp(self):
        self.fileWrapper = DummyFileWrapper()
        
        fileinfo = {
            'dirname': 'post_title',
            'filename': 'real_filename',
            'filepath': 'filepath',
            'filesize': 123456,
            'parts': 2,
        }
        
        # setup enviroment variable
        self.conf = {}
        self.conf['server'] = {'hostname':'test.Host'}
        self.conf['posting'] = {'from' : 'testFrom@Account'}
        self.newsgroup = 'test.Newsgroup'
        subject = 'testSubject [1/3] - "testFile" yEnc (1/1)'
        partnum = fileinfo['parts']
        
        self.art = Article(self.fileWrapper, 0, 3, fileinfo, subject, partnum)
        
        self.art.headers['From'] = self.conf['posting']['from']
        self.art.headers['Newsgroups'] = self.newsgroup
        self.art.headers['Subject'] = subject #% (partnum)
        self.art.headers['Message-ID'] = '<%.5f.%d@%s>' % (time.time(), partnum, self.conf['server']['hostname'])
        self.art.headers['X-Newsposter'] = 'newsmangler %s (%s) - https://github.com/madcowfred/newsmangler\r\n' % \
                ('0.0.1', 'yenc.yEncMode()')

    
    def test_article(self):
        self.art.prepare()
        
        ioRepresentation = self.art.postfile
        print(ioRepresentation.getvalue())
    
    def test_test(self):
        print(self.fileWrapper.read_part(1,2))
        
if __name__ == '__main__':
    unittest.main(verbosity=2)