From f94224d330ee4204eb9cd6a3225e7f08787cf614 Mon Sep 17 00:00:00 2001 From: Bob Halley Date: Sun, 31 Aug 2014 16:59:30 -0700 Subject: [PATCH] Fix some issues with dns.zone.to_file(). --- dns/zone.py | 2 +- tests/zone.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dns/zone.py b/dns/zone.py index c1b02236..a5f4f9c1 100644 --- a/dns/zone.py +++ b/dns/zone.py @@ -507,7 +507,7 @@ class Zone(object): LF on POSIX, CRLF on Windows, CR on Macintosh). @type nl: string or None """ - temp_buffer = io.StringIO() + temp_buffer = io.BytesIO() self.to_file(temp_buffer, sorted, relativize, nl) return_value = temp_buffer.getvalue() temp_buffer.close() diff --git a/tests/zone.py b/tests/zone.py index a1346098..61f41480 100644 --- a/tests/zone.py +++ b/tests/zone.py @@ -133,7 +133,7 @@ class ZoneTestCase(unittest.TestCase): finally: if not _keep_output: os.unlink('example3.out') - self.failUnless(ok) + self.assertTrue(ok) def testFromText(self): z = dns.zone.from_text(example_text, 'example.', relativize=True) -- 2.47.3