python-feedgen/tests/test_main.py
Lars Kiesow 9026316391
Use isort to sort import
…so that they all have the same structure.
2018-03-04 20:05:29 +01:00

42 lines
987 B
Python

# -*- coding: utf-8 -*-
'''
Tests for feedgen main
'''
import os
import sys
import tempfile
import unittest
from feedgen import __main__
class TestSequenceFunctions(unittest.TestCase):
def test_usage(self):
sys.argv = ['feedgen']
try:
__main__.main()
except BaseException as e:
assert e.code is None
def test_feed(self):
for ftype in 'rss', 'atom', 'podcast', 'torrent', 'dc.rss', 'dc.atom',\
'syndication.rss', 'syndication.atom':
sys.argv = ['feedgen', ftype]
try:
__main__.main()
except:
assert False
def test_file(self):
for extemsion in '.atom', '.rss':
fh, filename = tempfile.mkstemp(extemsion)
sys.argv = ['feedgen', filename]
try:
__main__.main()
except:
assert False
os.close(fh)
os.remove(filename)