rattail-project/rattail Files · tests/test

Files @ 08f9a4519954

Branch filter:

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

08f9a4519954 1.1 KiB text/x-python Show Annotation Show as Raw Download as Raw

lance

Added `util` module, for `OrderedDict` convenience.


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)