"""Run integration tests with a speckle server."""
from speckle_automate import (
    AutomationContext,
    AutomationRunData,
    AutomationStatus,
    run_function,
)
from speckle_automate.fixtures import *  # noqa: F401, F403

from data_shield import automate_function, FunctionInputs, SanitizationMode


class TestFunction:
    def test_function_run(self, test_automation_run_data: AutomationRunData, test_automation_token: str) -> None:
        """Run an integration test for the automate function."""
        automation_context = AutomationContext.initialize(test_automation_run_data, test_automation_token)

        automate_sdk = run_function(
            automation_context,
            automate_function,
            FunctionInputs(
                sanitization_mode=SanitizationMode.PREFIX_MATCHING,
                forbidden_parameter_input="Speckle"
            ),
        )

        assert automate_sdk.run_status == AutomationStatus.SUCCEEDED