Files
@ d9b506352a89
Branch filter:
Location: rattail-project/rattail/tests/test_csvutil.py - annotation
d9b506352a89
1.1 KiB
text/x-python
Change behavior of `files.locking_copy()` function.
The test function has been doing its job well, I'm calling it the better
function at this point.
The test function has been doing its job well, I'm calling it the better
function at this point.
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)
|