#
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
#


import json
import subprocess
import sys

import pytest
from airbyte_cdk.exception_handler import assemble_uncaught_exception
from airbyte_cdk.models import (
    AirbyteErrorTraceMessage,
    AirbyteLogMessage,
    AirbyteMessage,
    AirbyteMessageSerializer,
    AirbyteTraceMessage,
    FailureType,
    Level,
    TraceType,
)
from airbyte_cdk.models import Type as MessageType
from airbyte_cdk.sources.streams.concurrent.exceptions import ExceptionWithDisplayMessage
from airbyte_cdk.utils.traced_exception import AirbyteTracedException


def test_given_exception_is_traced_exception_when_assemble_uncaught_exception_then_return_same_exception():
    exception = AirbyteTracedException()
    assembled_exception = assemble_uncaught_exception(type(exception), exception)
    assert exception == assembled_exception


def test_given_exception_not_traced_exception_when_assemble_uncaught_exception_then_return_traced_exception():
    exception = ValueError("any error")
    assembled_exception = assemble_uncaught_exception(type(exception), exception)
    assert isinstance(assembled_exception, AirbyteTracedException)


def test_given_exception_with_display_message_when_assemble_uncaught_exception_then_internal_message_contains_display_message():
    display_message = "some display message"
    exception = ExceptionWithDisplayMessage(display_message)
    assembled_exception = assemble_uncaught_exception(type(exception), exception)
    assert display_message in assembled_exception.internal_message


def test_uncaught_exception_handler():
    cmd = "from airbyte_cdk.logger import init_logger; from airbyte_cdk.exception_handler import init_uncaught_exception_handler; logger = init_logger('airbyte'); init_uncaught_exception_handler(logger); raise 1"
    exception_message = "exceptions must derive from BaseException"
    exception_trace = (
        "Traceback (most recent call last):\n"
        '  File "<string>", line 1, in <module>\n'
        "TypeError: exceptions must derive from BaseException"
    )

    expected_log_message = AirbyteMessage(
        type=MessageType.LOG, log=AirbyteLogMessage(level=Level.FATAL, message=f"{exception_message}\n{exception_trace}")
    )

    expected_trace_message = AirbyteMessage(
        type=MessageType.TRACE,
        trace=AirbyteTraceMessage(
            type=TraceType.ERROR,
            emitted_at=0.0,
            error=AirbyteErrorTraceMessage(
                failure_type=FailureType.system_error,
                message="Something went wrong in the connector. See the logs for more details.",
                internal_message=exception_message,
                stack_trace=f"{exception_trace}\n",
            ),
        ),
    )

    with pytest.raises(subprocess.CalledProcessError) as err:
        subprocess.check_output([sys.executable, "-c", cmd], stderr=subprocess.STDOUT)

    assert not err.value.stderr, "nothing on the stderr"

    stdout_lines = err.value.output.decode("utf-8").strip().split("\n")
    assert len(stdout_lines) == 2

    log_output, trace_output = stdout_lines

    out_log_message = AirbyteMessageSerializer.load(json.loads(log_output))
    assert out_log_message == expected_log_message, "Log message should be emitted in expected form"

    out_trace_message = AirbyteMessageSerializer.load(json.loads(trace_output))
    assert out_trace_message.trace.emitted_at > 0
    out_trace_message.trace.emitted_at = 0.0  # set a specific emitted_at value for testing
    assert out_trace_message == expected_trace_message, "Trace message should be emitted in expected form"