From 7a95d8fff940cf6affbfef759ec69a5acf92f773 Mon Sep 17 00:00:00 2001 From: Brendan Abolivier Date: Thu, 29 Aug 2019 15:55:37 +0100 Subject: Fix encoding for account validity HTML files on Python 2 --- tests/rest/client/v2_alpha/test_register.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tests') diff --git a/tests/rest/client/v2_alpha/test_register.py b/tests/rest/client/v2_alpha/test_register.py index af1e600591..b28de3663c 100644 --- a/tests/rest/client/v2_alpha/test_register.py +++ b/tests/rest/client/v2_alpha/test_register.py @@ -20,6 +20,7 @@ import json import os from mock import Mock +from six import ensure_binary import pkg_resources @@ -437,7 +438,7 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase): # Check that the HTML we're getting is the one we expect on a successful renewal. expected_html = self.hs.config.account_validity.account_renewed_html_content self.assertEqual( - channel.result["body"], expected_html.encode("utf8"), channel.result + channel.result["body"], ensure_binary(expected_html), channel.result ) # Move 3 days forward. If the renewal failed, every authed request with @@ -467,7 +468,7 @@ class AccountValidityRenewalByEmailTestCase(unittest.HomeserverTestCase): # invalid/unknown token. expected_html = self.hs.config.account_validity.invalid_token_html_content self.assertEqual( - channel.result["body"], expected_html.encode("utf8"), channel.result + channel.result["body"], ensure_binary(expected_html), channel.result ) def test_manual_email_send(self): -- cgit 1.5.1