Files
@ 9343ba3fb2c8
Branch filter:
Location: rattail-project/rattail/tests/test_csvutil.py - annotation
9343ba3fb2c8
1.1 KiB
text/x-python
Tweak `locking_copy_test()` to assume destination is always a folder.
Also add constant for "file exists" error.
Also add constant for "file exists" error.
92c03f5d8db0 92c03f5d8db0 92c03f5d8db0 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 ee7d0f121060 | # -*- coding: utf-8 -*-
from __future__ import unicode_literals
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)
|