Jonathon Broughton
3db077d150
- Removed unnecessary imports from the main module. - Consolidated import statements for `FunctionInputs` and `automate_function`. - Updated test file to reflect new import paths.
28 lines
959 B
Python
28 lines
959 B
Python
"""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
|