from __future__ import annotations

from litegram.types import MessageEntity


class TestMessageEntity:
    def test_extract_from(self):
        entity = MessageEntity(type="hashtag", length=4, offset=5)
        assert entity.extract_from("#foo #bar #baz") == "#bar"
