summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorEli Collins <elic@assurancetechnologies.com>2016-07-17 14:26:13 -0400
committerEli Collins <elic@assurancetechnologies.com>2016-07-17 14:26:13 -0400
commitcb49e92a59400de636f76f0295da232340f1bbf8 (patch)
tree5b7faf98ead517ad4963a0b57cd2948b578bb5e6
parentd889b3a58ce24a247eb18394e2736ae43b9d4929 (diff)
downloadpasslib-cb49e92a59400de636f76f0295da232340f1bbf8.tar.gz
bugfix: test_totp: py3 compat fixes (forgot to mark some byte strings)
-rw-r--r--passlib/tests/test_totp.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/passlib/tests/test_totp.py b/passlib/tests/test_totp.py
index e726235..bdaa28b 100644
--- a/passlib/tests/test_totp.py
+++ b/passlib/tests/test_totp.py
@@ -181,7 +181,7 @@ class OTPContextTest(TestCase):
# ensure ":" allowed in secret
context = OTPContext("1: aaa: bbb \n# comment\n \n2: bbb ")
- self.assertEqual(context._secrets, {"1": b"aaa: bbb", "2": "bbb"})
+ self.assertEqual(context._secrets, {"1": b"aaa: bbb", "2": b"bbb"})
# json dict
context = OTPContext('{"1":"aaa","2":"bbb"}')
@@ -246,17 +246,17 @@ class OTPContextTest(TestCase):
# should sort numerically
context = OTPContext({"1": "one", "02": "two"})
self.assertEqual(context._default_tag, "02")
- self.assertEqual(context._default_secret, "two")
+ self.assertEqual(context._default_secret, b"two")
# should sort alphabetically if non-digit present
context = OTPContext({"1": "one", "02": "two", "A": "aaa"})
self.assertEqual(context._default_tag, "A")
- self.assertEqual(context._default_secret, "aaa")
+ self.assertEqual(context._default_secret, b"aaa")
# should use honor custom tag
context = OTPContext({"1": "one", "02": "two", "A": "aaa"}, default_tag="1")
self.assertEqual(context._default_tag, "1")
- self.assertEqual(context._default_secret, "one")
+ self.assertEqual(context._default_secret, b"one")
# throw error on unknown value
self.assertRaises(KeyError, OTPContext, {"1": "one", "02": "two", "A": "aaa"},