rattail-project/rattail Files · tests/test

Files @ 90726f7af75d

Branch filter:

Location: rattail-project/rattail/tests/test_csvutil.py

90726f7af75d 1.1 KiB text/x-python Show Annotation Show as Raw Download as Raw

lance

Add docs for new email system, esp. config syntax.


from unittest import TestCase
from mock import patch

from rattail import csvutil
from cStringIO import StringIO


class TestDictWriter(TestCase):

    def test_writeheader_26(self):
        # Simulate Python 2.6
        with patch('csv.writer'):
            with patch('rattail.csvutil.csv.DictWriter', spec=['writer']) as DictWriter:
                buf = StringIO()
                writer = csvutil.DictWriter(buf, ['field1', 'field2'])
                writer.writeheader()
                buf.close()
                writer.writer.writerow.assert_called_once_with(['field1', 'field2'])

    def test_writeheader_27(self):
        # Simulate Python 2.7+
        with patch('csv.writer'):
            with patch('rattail.csvutil.csv.DictWriter', spec=['writer', 'writeheader']) as DictWriter:
                buf = StringIO()
                writer = csvutil.DictWriter(buf, ['field1', 'field2'])
                writer.writeheader()
                buf.close()
                self.assertFalse(writer.writer.writerow.called)
                DictWriter.writeheader.assert_called_once_with(writer)