From e176c8fd4b84f8c0802e98c872fdbe589e9de259 Mon Sep 17 00:00:00 2001 From: William Grant Date: Sat, 19 Mar 2011 20:08:31 +1100 Subject: [PATCH] Bytesify test_dkim. --- dkim/tests/test_dkim.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/dkim/tests/test_dkim.py b/dkim/tests/test_dkim.py index ee29158..2f2ae47 100644 --- a/dkim/tests/test_dkim.py +++ b/dkim/tests/test_dkim.py @@ -35,13 +35,13 @@ class TestFold(unittest.TestCase): def test_short_line(self): self.assertEqual( - "foo", dkim.fold("foo")) + b"foo", dkim.fold(b"foo")) def DISABLED_test_long_line(self): # The function is terribly broken, not passing even this simple # test. self.assertEqual( - "foo"*24 + "\r\n foo", dkim.fold("foo" * 25)) + b"foo"*24 + b"\r\n foo", dkim.fold(b"foo" * 25)) class TestSignAndVerify(unittest.TestCase): @@ -52,19 +52,19 @@ class TestSignAndVerify(unittest.TestCase): self.key = read_test_data("test.private") def dnsfunc(self, domain): - self.assertEqual('test._domainkey.example.com.', domain) + self.assertEqual(b'test._domainkey.example.com.', domain) return read_test_data("test.txt") def test_verifies(self): # A message verifies after being signed. - sig = dkim.sign(self.message, "test", "example.com", self.key) + sig = dkim.sign(self.message, b"test", b"example.com", self.key) res = dkim.verify(sig + self.message, dnsfunc=self.dnsfunc) self.assertTrue(res) def test_altered_body_fails(self): # An altered body fails verification. - sig = dkim.sign(self.message, "test", "example.com", self.key) - res = dkim.verify(sig + self.message + "foo", dnsfunc=self.dnsfunc) + sig = dkim.sign(self.message, b"test", b"example.com", self.key) + res = dkim.verify(sig + self.message + b"foo", dnsfunc=self.dnsfunc) self.assertFalse(res)