def test_03_class_methods(self):
db_token = Token.query.filter(Token.serial == self.serial1).first()
token = SSHkeyTokenClass(db_token)
info = token.get_class_info()
self.assertTrue(info.get("title") == "SSHkey Token",
"%s" % info.get("title"))
info = token.get_class_info("title")
self.assertTrue(info == "SSHkey Token", info)
def test_03_class_methods(self):
db_token = Token.query.filter(Token.serial == self.serial1).first()
token = SSHkeyTokenClass(db_token)
info = token.get_class_info()
self.assertTrue(info.get("title") == "SSHkey Token",
"{0!s}".format(info.get("title")))
info = token.get_class_info("title")
self.assertTrue(info == "SSHkey Token", info)
def test_01_create_token(self):
db_token = Token(self.serial1, tokentype="sshkey")
db_token.save()
token = SSHkeyTokenClass(db_token)
# An invalid key, raises an exception
self.assertRaises(Exception, token.update, {"sshkey": "InvalidKey"})
# An invalid key, raises an exception
self.assertRaises(Exception, token.update, {"sshkey": self.wrong_sshkey})
# Set valid key
token.update({"sshkey": self.sshkey})
self.assertTrue(token.token.serial == self.serial1, token)
self.assertTrue(token.token.tokentype == "sshkey",
token.token.tokentype)
self.assertTrue(token.type == "sshkey", token)
class_prefix = token.get_class_prefix()
self.assertTrue(class_prefix == "SSHK", class_prefix)
self.assertTrue(token.get_class_type() == "sshkey", token)
def test_04_get_sshkey(self):
db_token = Token.query.filter(Token.serial == self.serial1).first()
token = SSHkeyTokenClass(db_token)
sshkey = token.get_sshkey()
self.assertTrue(sshkey == self.sshkey, sshkey)
self.assertIsInstance(sshkey, six.text_type)
db_token = Token.query.filter(Token.serial == self.serial2).first()
token = SSHkeyTokenClass(db_token)
sshkey = token.get_sshkey()
self.assertTrue(sshkey == self.sshkey_ecdsa, sshkey)
self.assertIsInstance(sshkey, six.text_type)
db_token = Token.query.filter(Token.serial == self.serial3).first()
token = SSHkeyTokenClass(db_token)
sshkey = token.get_sshkey()
self.assertTrue(sshkey == self.sshkey_ed25519, sshkey)
self.assertIsInstance(sshkey, six.text_type)
def test_01_create_token(self):
db_token = Token(self.serial1, tokentype="sshkey")
db_token.save()
token = SSHkeyTokenClass(db_token)
# An invalid key, raises an exception
self.assertRaises(Exception, token.update, {"sshkey": "InvalidKey"})
# An invalid key, raises an exception
self.assertRaises(Exception, token.update, {"sshkey": self.wrong_sshkey})
# Set valid key
token.update({"sshkey": self.sshkey})
self.assertTrue(token.token.serial == self.serial1, token)
self.assertTrue(token.token.tokentype == "sshkey",
token.token.tokentype)
self.assertTrue(token.type == "sshkey", token)
class_prefix = token.get_class_prefix()
self.assertTrue(class_prefix == "SSHK", class_prefix)
self.assertTrue(token.get_class_type() == "sshkey", token)
def test_04_get_sshkey(self):
db_token = Token.query.filter(Token.serial == self.serial1).first()
token = SSHkeyTokenClass(db_token)
sshkey = token.get_sshkey()
self.assertTrue(sshkey == self.sshkey, sshkey)
def test_04_get_sshkey(self):
db_token = Token.query.filter(Token.serial == self.serial1).first()
token = SSHkeyTokenClass(db_token)
sshkey = token.get_sshkey()
self.assertTrue(sshkey == self.sshkey, sshkey)
