def test_02_set_user(self): db_token = Token.query.filter_by(serial=self.serial1).first() token = TokenClass(db_token) self.assertTrue(token.token.tokentype == "newtype", token.token.tokentype) self.assertTrue(token.type == "newtype", token.type) token.add_user(User(login="******", realm=self.realm1)) user_object = token.user self.assertTrue(user_object.login == "cornelius", user_object) self.assertTrue(user_object.resolver == self.resolvername1, user_object)