31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
# -*- coding: utf-8; -*-
|
|
|
|
from io import StringIO
|
|
from unittest import TestCase
|
|
from unittest.mock import patch
|
|
|
|
from rattail import csvutil
|
|
|
|
|
|
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)
|