import logging
from unittest.mock import patch
from pathlib import Path

import pytest

from haystack.preview.dataclasses import ByteStream
from haystack.preview.components.converters.txt import TextFileToDocument


class TestTextfileToDocument:
    @pytest.mark.unit
    def test_run(self, preview_samples_path):
        """
        Test if the component runs correctly.
        """
        bytestream = ByteStream.from_file_path(preview_samples_path / "txt" / "doc_3.txt")
        bytestream.metadata["file_path"] = str(preview_samples_path / "txt" / "doc_3.txt")
        bytestream.metadata["key"] = "value"
        files = [
            str(preview_samples_path / "txt" / "doc_1.txt"),
            preview_samples_path / "txt" / "doc_2.txt",
            bytestream,
        ]
        converter = TextFileToDocument()
        output = converter.run(sources=files)
        docs = output["documents"]
        assert len(docs) == 3
        assert "Some text for testing." in docs[0].content
        assert "This is a test line." in docs[1].content
        assert "That's yet another file!" in docs[2].content
        assert docs[0].meta["file_path"] == str(files[0])
        assert docs[1].meta["file_path"] == str(files[1])
        assert docs[2].meta == bytestream.metadata

    @pytest.mark.unit
    def test_run_error_handling(self, preview_samples_path, caplog):
        """
        Test if the component correctly handles errors.
        """
        paths = [
            preview_samples_path / "txt" / "doc_1.txt",
            "non_existing_file.txt",
            preview_samples_path / "txt" / "doc_3.txt",
        ]
        converter = TextFileToDocument()
        with caplog.at_level(logging.WARNING):
            output = converter.run(sources=paths)
            assert "non_existing_file.txt" in caplog.text
        docs = output["documents"]
        assert len(docs) == 2
        assert docs[0].meta["file_path"] == str(paths[0])
        assert docs[1].meta["file_path"] == str(paths[2])

    @pytest.mark.unit
    def test_encoding_override(self, preview_samples_path):
        """
        Test if the encoding metadata field is used properly
        """
        bytestream = ByteStream.from_file_path(preview_samples_path / "txt" / "doc_1.txt")
        bytestream.metadata["key"] = "value"

        converter = TextFileToDocument(encoding="utf-16")
        output = converter.run(sources=[bytestream])
        assert "Some text for testing." not in output["documents"][0].content

        bytestream.metadata["encoding"] = "utf-8"
        output = converter.run(sources=[bytestream])
        assert "Some text for testing." in output["documents"][0].content