tests: add tests for assert and string_code

Author: lepture

src/otpauth/core.py

@@ -11,6 +11,7 @@ class OTP(object, metaclass=ABCMeta):
     ALGORITHMS = ["SHA1", "SHA256", "SHA512"]
 
     def __init__(self, secret: bytes, digit: int = 6, algorithm: str = "SHA1"):
+        assert 0 < digit < 11
         assert algorithm in self.ALGORITHMS
 
         self.secret = secret

tests/test_misc.py

@@ -0,0 +1,19 @@
+import unittest
+from otpauth import HOTP, TOTP
+
+
+class TestMisc(unittest.TestCase):
+    def test_invalid_digit(self):
+        self.assertRaises(AssertionError, HOTP, b"secret", 0)
+        self.assertRaises(AssertionError, TOTP, b"secret", 0)
+
+        self.assertRaises(AssertionError, HOTP, b"secret", 11)
+        self.assertRaises(AssertionError, TOTP, b"secret", 11)
+
+    def test_invalid_algorithm(self):
+        self.assertRaises(AssertionError, HOTP, b"secret", algorithm="SHA2")
+        self.assertRaises(AssertionError, TOTP, b"secret", algorithm="SHA2")
+
+    def test_string_code(self):
+        htop = HOTP(b"secret")
+        self.assertEqual(htop.string_code(123), "000123")