from omniff.integrations.llamaindex import OmniFFReader


def test_reader_init():
    reader = OmniFFReader()
    assert reader.base_url == "http://localhost:8000"
    assert ".jpg" in reader.supported_extensions


def test_can_read_image():
    reader = OmniFFReader()
    assert reader.can_read("photo.jpg")
    assert reader.can_read("doc.pdf")
    assert not reader.can_read("script.py")


def test_custom_extensions():
    reader = OmniFFReader(supported_extensions=[".csv"])
    assert reader.can_read("data.csv")
    assert not reader.can_read("image.png")