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_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)