Files
openobserve/tests/api-testing/tests/test_pipeline.py
T
+2 0d9f47966b feat: Organization revamp (#5579)
Fixes #5131 
Addresses #5408 

### Revamp
- Created a new table to store organizations in the `organizations`
table. Earlier, we never stored organizations in any table, rather we
used to store the organizations in the meta table as part of the `user`
json.
- The users and organization membership relations are now moved to its
own separate tables - `users` and `org_users`.

### New changes

**API changes** -
- `/organizations` - `POST` api to create an org and save it in the meta
table. In non-enterprise version, currently only the root user can
create the org. For enterprise version with rbac enabled, anyone with
`CREATE` access for organizations can create the organization and
becomes an admin. Root user can always create an organization.
- `/{org_id}/rename` - `PUT` api to rename the existing org. Only root
user and the admins of the org can access this API.

The payload for following APIs are changed -
- `POST api/{org_id}/roles` - `{"name":"new_role"}` is changed to
`{"role":"new_role"}
- `POST api/{org_id}/users/{user}` -
`{"role":"viewer","custom_role":["role_name"]}` is the new payload where
`custom_role` is optional and is array of custom roles (only works in
enterprise with rbac enabled) which will be assigned to the user. The
value of `role` can be standard role or as well as any custom role (when
it is custom role, the standard role assigned to the user is decided
based on the enterprise env variable - `O2_DEX_DEFAULT_ROLE`).
- `POST api/{org_id}/users`- the payload -
```
{"email":"user@user.com","role":"viewer","custom_role":["create_org_permission"],"password":"12345678", "first_name": "", "last_name":""}
```

- `PUT api//{org_id}/users/{email_id}` the payload -
```
{"first_name":"Hello","last_name":"j","role":"viewer","isExternal":false,"change_password":false,"password":"","custom_role":["create_org_permission"]}
```


A new ENV variable is introduced - `ZO_CREATE_ORG_THROUGH_INGESTION`
(default `false`) which if true, an org will automatically created when
ingesting data to a new org. If false, user can not create org by
ingesting data to new org, they need to first call the `POST
/organizations` API.

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

## Release Notes

**New Features**
- Added a `cloud` feature for enhanced organizational management
capabilities.
- Introduced new methods for managing organizational users and
asynchronous message processing.
- Enhanced user invitation and membership features, including improved
role handling.
- Implemented configurable ingestion URL support across various
components.
- Added new functionality for user management and organization handling
in the super cluster architecture.

**Improvements**
- Refined user authentication and authorization processes, including
error handling.
- Updated database schema to support multi-organization architecture.
- Enhanced pagination settings and UI components for better user
experience.
- Improved handling of organization names and user roles in
notifications and alerts.
- Streamlined service account token management with visibility
enhancements.
- Enhanced the `AddUpdateOrganization` component with improved user
input handling and validation.
- Updated the `AppOrganizations` component to dynamically load the
appropriate organization management interface.
- Improved the `ServiceAccountsList` component with updated pagination
options and token handling logic.

**Bug Fixes**
- Resolved issues with user role assignments and organization management
inconsistencies.
- Corrected routing and navigation logic for login and logout processes.
- Addressed various error handling scenarios in user and organization
processes.
- Fixed issues with the visibility of service account options in the
`EditGroup` and `EditRole` components based on the cloud configuration.

**Breaking Changes**
- Significant restructuring of user and organization data models.
- Changes to authentication workflows and configuration handling.
- Removal of certain components related to user and organization
management.
- Updated the logic for determining the ingestion URL across various
components.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Signed-off-by: Yashodhan Joshi <yjdoc2@gmail.com>
Co-authored-by: Bhargav <BJP232004@GMAIL.COM>
Co-authored-by: Taiming Liu <liutaiming3@gmail.com>
Co-authored-by: hengfei yang <hengfei.yang@gmail.com>
Co-authored-by: harshiv-7 <harshiv@openobserve.ai>
Co-authored-by: Huaijin <haohuaijin@gmail.com>
Co-authored-by: Zhao.wei <weizhao029@gmail.com>
Co-authored-by: ShyamOOAI <shyam@openobserve.ai>
Co-authored-by: oasisk <ashish.j.kolhe@gmail.com>
Co-authored-by: sai nikhil kethe <nikhil@openobserve.ai>
Co-authored-by: Yashodhan Joshi <yashodhan@openobserve.ai>
Co-authored-by: Yashodhan Joshi <yjdoc2@gmail.com>
2025-05-25 21:53:55 -07:00

1088 lines
38 KiB
Python

import pytest
@pytest.mark.parametrize(
"pipeline_name, stream_name, action, expected_status",
[
("pipeline12222", "stream_pytest_data", None, 200), # No action (default behavior)
("pipeline12345", "stream_test_data", "enable", 200), # Action: Enable
("pipeline67890", "stream_other_data", "disable", 200), # Action: Disable
]
)
def test_pipeline_creation_and_action_realtime(create_session, base_url, pipeline_name, stream_name, action, expected_status):
"""Running an E2E test for pipeline creation and enabling/disabling the pipeline."""
session = create_session
url = base_url
org_id = "default"
# Payload for pipeline creation
pipeline_payload = {
"name": pipeline_name,
"description": "",
"source": {"source_type": "realtime"},
"nodes": [
{
"id": "1e1fa129-cfbd-4082-8d39-c4dc418a670f",
"type": "input",
"data": {
"node_type": "stream",
"stream_name": stream_name,
"stream_type": "logs",
"org_id": org_id,
},
"position": {"x": 169.83333, "y": 55},
"io_type": "input",
},
{
"id": "8e0ed123-7737-4801-9466-bd909c425c72",
"type": "output",
"data": {
"node_type": "stream",
"stream_name": stream_name,
"stream_type": "logs",
"org_id": org_id,
},
"position": {"x": 305.33334, "y": 276},
"io_type": "output",
},
],
"edges": [
{
"id": "e1e1fa129-cfbd-4082-8d39-c4dc418a670f-8e0ed123-7737-4801-9466-bd909c425c72",
"source": "1e1fa129-cfbd-4082-8d39-c4dc418a670f",
"target": "8e0ed123-7737-4801-9466-bd909c425c72",
}
],
"org": org_id,
}
# Create a pipeline
resp_create_pipeline = session.post(f"{url}api/{org_id}/pipelines", json=pipeline_payload)
assert resp_create_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_create_pipeline.status_code}"
print("Create pipeline response:", resp_create_pipeline.json())
# Retrieve the pipeline list
resp_list_pipelines = session.get(f"{url}api/{org_id}/pipelines")
assert resp_list_pipelines.status_code == 200, f"Expected status code 200 but got {resp_list_pipelines.status_code}"
# Parse the pipeline list response
pipelines_list = resp_list_pipelines.json().get("list", [])
assert pipelines_list, "No pipelines found in the list response."
# Extract the pipeline_id for the created pipeline
pipeline_id = None
for pipeline in pipelines_list:
if pipeline["name"] == pipeline_name:
pipeline_id = pipeline["pipeline_id"]
break
assert pipeline_id, "Pipeline ID not found for the created pipeline."
print(f"Pipeline ID: {pipeline_id}")
# Enable or disable the pipeline based on the action parameter
if action == "enable":
resp_enable_pipeline = session.put(f"{url}api/{org_id}/pipelines/{pipeline_id}/enable?value=true")
assert resp_enable_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_enable_pipeline.status_code}"
print(f"Pipeline {pipeline_id} enabled successfully.")
elif action == "disable":
resp_enable_pipeline = session.put(f"{url}api/{org_id}/pipelines/{pipeline_id}/enable?value=false")
assert resp_enable_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_enable_pipeline.status_code}"
print(f"Pipeline {pipeline_id} disabled successfully.")
resp_get_pipeline = session.get(f"{url}api/{org_id}/pipelines")
print(resp_get_pipeline.json())
assert resp_get_pipeline.status_code == 200
pipeline_list = resp_get_pipeline.json()["list"]
target_pipeline = next((p for p in pipeline_list if p["pipeline_id"] == pipeline_id), None)
assert target_pipeline is not None, f"Pipeline {pipeline_id} not found in the list"
if action == "enable":
assert target_pipeline["enabled"] is True
if action == "disable":
assert target_pipeline["enabled"] is False
# Delete the pipeline
resp_delete_pipeline = session.delete(f"{url}api/{org_id}/pipelines/{pipeline_id}")
assert resp_delete_pipeline.status_code == 200, f"Expected status code 200 but got {resp_delete_pipeline.status_code}"
# Verify pipeline is deleted
resp_verify_deletion = session.get(f"{url}api/{org_id}/pipelines")
assert resp_verify_deletion.status_code == 200
deleted_pipeline = next((p for p in resp_verify_deletion.json()["list"] if p["pipeline_id"] == pipeline_id), None)
assert deleted_pipeline is None, f"Pipeline {pipeline_id} still exists after deletion"
@pytest.mark.parametrize(
"pipeline_name, stream_name, action, expected_status, schedule_enabled",
[
("pipeline12222", "stream_pytest_data", None, 200, False), # No action (default behavior)
("scheduled_pipeline", "stream_scheduled_data", "enable", 200, True), # Action: Enable with schedule
("scheduled_pipeline", "stream_scheduled_data", "disable", 200, False), # Action: Disable with schedule
]
)
def test_pipeline_creation_and_action_with_schedule(create_session, base_url, pipeline_name, stream_name, action, expected_status, schedule_enabled):
"""Running an E2E test for pipeline creation with scheduling and enabling/disabling the pipeline."""
session = create_session
url = base_url
org_id = "default"
# Payload for scheduled pipeline creation
pipeline_payload = {
"name": pipeline_name,
"description": "",
"source": {"source_type": "scheduled"},
"nodes": [
{
"id": "8b03771c-2e94-4c2f-b902-8fb8b8e66df2",
"type": "input",
"dimensions": {"width": 140, "height": 42},
"computedPosition": {"x": 212.1047849020086, "y": 93.00563781047852, "z": 0},
"handleBounds": {
"source": [{"id": "output", "position": "bottom", "x": 55.03916432328342, "y": 36.00006521998462, "width": 30, "height": 10}],
"target": []
},
"position": {"x": 212.1047849020086, "y": 93.00563781047852},
"data": {
"label": "8b03771c-2e94-4c2f-b902-8fb8b8e66df2",
"node_type": "query",
"stream_type": "logs",
"org_id": org_id,
"query_condition": {
"type": "sql",
"conditions": None,
"sql": "select * from default",
"promql": None,
"promql_condition": None,
"aggregation": None,
"vrl_function": None,
"search_event_type": "DerivedStream"
},
"trigger_condition": {
"period": 15,
"operator": "=",
"threshold": 0,
"frequency": 15,
"cron": "",
"frequency_type": "minutes",
"silence": 0
}
},
"io_type": "input"
},
{
"id": "cfa476a0-24f3-4ca3-a14f-b56b04729922",
"type": "output",
"dimensions": {"width": 250, "height": 42},
"computedPosition": {"x": 269.2781219770897, "y": 161.6932759313432, "z": 0},
"handleBounds": {
"source": [],
"target": [{"id": "input", "position": "top", "x": 110.11741720560622, "y": -4.000002219844093, "width": 30, "height": 10}]
},
"position": {"x": 269.2781219770897, "y": 161.6932759313432},
"data": {
"label": "cfa476a0-24f3-4ca3-a14f-b56b04729922",
"node_type": "stream",
"stream_type": "logs",
"stream_name": stream_name,
"org_id": org_id
},
"io_type": "output"
}
],
"edges": [
{
"id": "e8b03771c-2e94-4c2f-b902-8fb8b8e66df2-cfa476a0-24f3-4ca3-a14f-b56b04729922",
"type": "custom",
"source": "8b03771c-2e94-4c2f-b902-8fb8b8e66df2",
"target": "cfa476a0-24f3-4ca3-a14f-b56b04729922",
"data": {},
"markerEnd": {"type": "arrowclosed", "width": 20, "height": 20},
"style": {"strokeWidth": 2},
"animated": True
}
],
"org": org_id,
"schedule": {
"enabled": schedule_enabled,
"frequency": "daily", # Example frequency, adjust as needed
"time": "02:00", # Example time, adjust as needed
}
}
# Create a pipeline
resp_create_pipeline = session.post(f"{url}api/{org_id}/pipelines", json=pipeline_payload)
assert resp_create_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_create_pipeline.status_code}"
print("Create pipeline response:", resp_create_pipeline.json())
# Retrieve the pipeline list
resp_list_pipelines = session.get(f"{url}api/{org_id}/pipelines")
assert resp_list_pipelines.status_code == 200, f"Expected status code 200 but got {resp_list_pipelines.status_code}"
# Extract the pipeline_id for the created pipeline
pipelines_list = resp_list_pipelines.json().get("list", [])
pipeline_id = next((p["pipeline_id"] for p in pipelines_list if p["name"] == pipeline_name), None)
assert pipeline_id, "Pipeline ID not found for the created pipeline."
print(f"Pipeline ID: {pipeline_id}")
# Enable/disable pipeline based on the action
value = "True" if action == "enable" else "False"
resp_toggle_pipeline = session.put(f"{url}api/{org_id}/pipelines/{pipeline_id}/enable?value={value}")
assert resp_toggle_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_toggle_pipeline.status_code}"
print(f"Pipeline {pipeline_id} {'enabled' if action == 'enable' else 'disabled'} successfully.")
# Verify the pipeline state
resp_verify_pipeline = session.get(f"{url}api/{org_id}/pipelines")
pipeline_state = next((p["enabled"] for p in resp_verify_pipeline.json()["list"] if p["pipeline_id"] == pipeline_id), None)
assert pipeline_state == (action == "enable"), f"Pipeline state mismatch after {action} action."
# Delete the pipeline
resp_delete_pipeline = session.delete(f"{url}api/{org_id}/pipelines/{pipeline_id}")
assert resp_delete_pipeline.status_code == 200, f"Expected status code 200 but got {resp_delete_pipeline.status_code}"
print(f"Pipeline {pipeline_id} deleted successfully.")
@pytest.mark.parametrize(
"pipeline_name, stream_name, action, expected_status, schedule_enabled",
[
("pipeline12222", "stream_pytest_data", None, 400, False), # No action (default behavior)
]
)
def test_pipeline_creation_and_action_with_schedule(create_session, base_url, pipeline_name, stream_name, action, expected_status, schedule_enabled):
"""Running an E2E test for pipeline creation with scheduling and enabling/disabling the pipeline."""
session = create_session
url = base_url
org_id = "default"
# Payload for scheduled pipeline creation
pipeline_payload = {
"name": pipeline_name,
"description": "",
"source": {"source_type": "scheduled"},
"nodes": [
{
"id": "8b03771c-2e94-4c2f-b902-8fb8b8e66df2",
"type": "input",
"dimensions": {"width": 140, "height": 42},
"computedPosition": {"x": 212.1047849020086, "y": 93.00563781047852, "z": 0},
"handleBounds": {
"source": [{"id": "output", "position": "bottom", "x": 55.03916432328342, "y": 36.00006521998462, "width": 30, "height": 10}],
"target": []
},
"position": {"x": 212.1047849020086, "y": 93.00563781047852},
"data": {
"label": "8b03771c-2e94-4c2f-b902-8fb8b8e66df2",
"node_type": "query",
"stream_type": "logs",
"org_id": org_id,
"query_condition": {
"type": "sql",
"conditions": None,
"sql": "",
"promql": None,
"promql_condition": None,
"aggregation": None,
"vrl_function": None,
"search_event_type": "DerivedStream"
},
"trigger_condition": {
"period": 15,
"operator": "=",
"threshold": 0,
"frequency": 15,
"cron": "",
"frequency_type": "minutes",
"silence": 0
}
},
"io_type": "input"
},
{
"id": "cfa476a0-24f3-4ca3-a14f-b56b04729922",
"type": "output",
"dimensions": {"width": 250, "height": 42},
"computedPosition": {"x": 269.2781219770897, "y": 161.6932759313432, "z": 0},
"handleBounds": {
"source": [],
"target": [{"id": "input", "position": "top", "x": 110.11741720560622, "y": -4.000002219844093, "width": 30, "height": 10}]
},
"position": {"x": 269.2781219770897, "y": 161.6932759313432},
"data": {
"label": "cfa476a0-24f3-4ca3-a14f-b56b04729922",
"node_type": "stream",
"stream_type": "logs",
"stream_name": stream_name,
"org_id": org_id
},
"io_type": "output"
}
],
"edges": [
{
"id": "e8b03771c-2e94-4c2f-b902-8fb8b8e66df2-cfa476a0-24f3-4ca3-a14f-b56b04729922",
"type": "custom",
"source": "8b03771c-2e94-4c2f-b902-8fb8b8e66df2",
"target": "cfa476a0-24f3-4ca3-a14f-b56b04729922",
"data": {},
"markerEnd": {"type": "arrowclosed", "width": 20, "height": 20},
"style": {"strokeWidth": 2},
"animated": True
}
],
"org": org_id,
"schedule": {
"enabled": schedule_enabled,
"frequency": "daily", # Example frequency, adjust as needed
"time": "02:00", # Example time, adjust as needed
}
}
# Create a pipeline
resp_create_pipeline = session.post(f"{url}api/{org_id}/pipelines", json=pipeline_payload)
assert resp_create_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_create_pipeline.status_code}"
print("Create pipeline response:", resp_create_pipeline.json())
@pytest.mark.parametrize(
"pipeline_name, stream_name, action, expected_status",
[
("pipeline12222", "stream_pytest_data", None, 400), # No action (default behavior)
]
)
def test_pipeline_creation_and_action_flatten(create_session, base_url, pipeline_name, stream_name, action, expected_status):
"""Running an E2E test for pipeline creation and enabling/disabling the pipeline."""
session = create_session
url = base_url
org_id = "default"
# Payload for pipeline creation
pipeline_payload = {
"name": pipeline_name,
"description": "",
"source": {
"source_type": "realtime"
},
"nodes": [
{
"id": "b6c04033-6492-44f9-a365-00d6c6425193",
"type": "input",
"dimensions": {
"width": 271,
"height": 42
},
"computedPosition": {
"x": 124.5,
"y": 63,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 120.515625,
"y": 36,
"width": 30,
"height": 10
}
],
"target": []
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 124.5,
"y": 63
},
"data": {
"label": "b6c04033-6492-44f9-a365-00d6c6425193",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "stream_pytest_data",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "input"
},
{
"id": "60aea6df-ec59-452e-ab52-4d1ec18391a1",
"type": "output",
"dimensions": {
"width": 271,
"height": 42
},
"computedPosition": {
"x": 260,
"y": 284,
"z": 0
},
"handleBounds": {
"source": [],
"target": [
{
"id": "input",
"position": "top",
"x": 120.515625,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 260,
"y": 284
},
"data": {
"label": "60aea6df-ec59-452e-ab52-4d1ec18391a1",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "stream_pytest_data",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "output"
},
{
"id": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"type": "default",
"dimensions": {
"width": 234,
"height": 54
},
"computedPosition": {
"x": 285.66666666666674,
"y": 125.00000000000003,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 102.21875,
"y": 48,
"width": 30,
"height": 10
}
],
"target": [
{
"id": "input",
"position": "top",
"x": 102.21875,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 285.66666666666674,
"y": 125.00000000000003
},
"data": {
"label": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"node_type": "function",
"name": "pytest",
"after_flatten": True
},
"events": {},
"io_type": "default"
},
{
"id": "4278cc25-1699-4532-b4ef-705c63d455ee",
"type": "default",
"dimensions": {
"width": 244,
"height": 54
},
"computedPosition": {
"x": 53.33333333333334,
"y": 160.33333333333334,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 107.1171875,
"y": 48,
"width": 30,
"height": 10
}
],
"target": [
{
"id": "input",
"position": "top",
"x": 107.1171875,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 53.33333333333334,
"y": 160.33333333333334
},
"data": {
"label": "4278cc25-1699-4532-b4ef-705c63d455ee",
"node_type": "function",
"name": "pytest2",
"after_flatten": False
},
"events": {},
"io_type": "default"
},
{
"id": "9d2e796a-d424-48b8-a0b0-0d8b2143648c",
"type": "output",
"dimensions": {
"width": 238,
"height": 42
},
"computedPosition": {
"x": -21.666666666666686,
"y": 313,
"z": 1000
},
"handleBounds": {
"source": [],
"target": [
{
"id": "input",
"position": "top",
"x": 103.828125,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": True,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": -21.666666666666686,
"y": 313
},
"data": {
"label": "9d2e796a-d424-48b8-a0b0-0d8b2143648c",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "pyteststream2",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "output"
}
],
"edges": [
{
"id": "eb6c04033-6492-44f9-a365-00d6c6425193-f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"type": "custom",
"source": "b6c04033-6492-44f9-a365-00d6c6425193",
"target": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"data": {},
"events": {},
"label": "",
"markerEnd": {
"type": "arrowclosed",
"width": 20,
"height": 20
},
"style": {
"strokeWidth": 2
},
"animated": True,
"sourceNode": {
"id": "b6c04033-6492-44f9-a365-00d6c6425193",
"type": "input",
"dimensions": {
"width": 271,
"height": 42
},
"computedPosition": {
"x": 124.5,
"y": 63,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 120.515625,
"y": 36,
"width": 30,
"height": 10
}
],
"target": []
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 124.5,
"y": 63
},
"data": {
"label": "b6c04033-6492-44f9-a365-00d6c6425193",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "stream_pytest_data",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "input"
},
"targetNode": {
"id": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"type": "default",
"dimensions": {
"width": 234,
"height": 54
},
"computedPosition": {
"x": 285.66666666666674,
"y": 125.00000000000003,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 102.21875,
"y": 48,
"width": 30,
"height": 10
}
],
"target": [
{
"id": "input",
"position": "top",
"x": 102.21875,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 285.66666666666674,
"y": 125.00000000000003
},
"data": {
"label": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"node_type": "function",
"name": "pytest",
"after_flatten": True
},
"events": {},
"io_type": "default"
},
"sourceX": 260.015625,
"sourceY": 109,
"targetX": 402.88541666666674,
"targetY": 121.00000000000003
},
{
"id": "ef912e999-0f19-4b9a-9f9b-34c4552cdaa4-60aea6df-ec59-452e-ab52-4d1ec18391a1",
"type": "custom",
"source": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"target": "60aea6df-ec59-452e-ab52-4d1ec18391a1",
"data": {},
"events": {},
"label": "",
"markerEnd": {
"type": "arrowclosed",
"width": 20,
"height": 20
},
"style": {
"strokeWidth": 2
},
"animated": True,
"sourceNode": {
"id": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"type": "default",
"dimensions": {
"width": 234,
"height": 54
},
"computedPosition": {
"x": 285.66666666666674,
"y": 125.00000000000003,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 102.21875,
"y": 48,
"width": 30,
"height": 10
}
],
"target": [
{
"id": "input",
"position": "top",
"x": 102.21875,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 285.66666666666674,
"y": 125.00000000000003
},
"data": {
"label": "f912e999-0f19-4b9a-9f9b-34c4552cdaa4",
"node_type": "function",
"name": "pytest",
"after_flatten": True
},
"events": {},
"io_type": "default"
},
"targetNode": {
"id": "60aea6df-ec59-452e-ab52-4d1ec18391a1",
"type": "output",
"dimensions": {
"width": 271,
"height": 42
},
"computedPosition": {
"x": 260,
"y": 284,
"z": 0
},
"handleBounds": {
"source": [],
"target": [
{
"id": "input",
"position": "top",
"x": 120.515625,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 260,
"y": 284
},
"data": {
"label": "60aea6df-ec59-452e-ab52-4d1ec18391a1",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "stream_pytest_data",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "output"
},
"sourceX": 402.88541666666674,
"sourceY": 183.00000000000003,
"targetX": 395.515625,
"targetY": 280
},
{
"id": "eb6c04033-6492-44f9-a365-00d6c6425193-4278cc25-1699-4532-b4ef-705c63d455ee",
"type": "custom",
"source": "b6c04033-6492-44f9-a365-00d6c6425193",
"target": "4278cc25-1699-4532-b4ef-705c63d455ee",
"data": {},
"events": {},
"label": "",
"markerEnd": {
"type": "arrowclosed",
"width": 20,
"height": 20
},
"style": {
"strokeWidth": 2
},
"animated": True,
"sourceNode": {
"id": "b6c04033-6492-44f9-a365-00d6c6425193",
"type": "input",
"dimensions": {
"width": 271,
"height": 42
},
"computedPosition": {
"x": 124.5,
"y": 63,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 120.515625,
"y": 36,
"width": 30,
"height": 10
}
],
"target": []
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 124.5,
"y": 63
},
"data": {
"label": "b6c04033-6492-44f9-a365-00d6c6425193",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "stream_pytest_data",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "input"
},
"targetNode": {
"id": "4278cc25-1699-4532-b4ef-705c63d455ee",
"type": "default",
"dimensions": {
"width": 244,
"height": 54
},
"computedPosition": {
"x": 53.33333333333334,
"y": 160.33333333333334,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 107.1171875,
"y": 48,
"width": 30,
"height": 10
}
],
"target": [
{
"id": "input",
"position": "top",
"x": 107.1171875,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 53.33333333333334,
"y": 160.33333333333334
},
"data": {
"label": "4278cc25-1699-4532-b4ef-705c63d455ee",
"node_type": "function",
"name": "pytest2",
"after_flatten": False
},
"events": {},
"io_type": "default"
},
"sourceX": 260.015625,
"sourceY": 109,
"targetX": 175.45052083333334,
"targetY": 156.33333333333334
},
{
"id": "e4278cc25-1699-4532-b4ef-705c63d455ee-9d2e796a-d424-48b8-a0b0-0d8b2143648c",
"type": "custom",
"source": "4278cc25-1699-4532-b4ef-705c63d455ee",
"target": "9d2e796a-d424-48b8-a0b0-0d8b2143648c",
"data": {},
"events": {},
"label": "",
"markerEnd": {
"type": "arrowclosed",
"width": 20,
"height": 20
},
"style": {
"strokeWidth": 2
},
"animated": True,
"sourceNode": {
"id": "4278cc25-1699-4532-b4ef-705c63d455ee",
"type": "default",
"dimensions": {
"width": 244,
"height": 54
},
"computedPosition": {
"x": 53.33333333333334,
"y": 160.33333333333334,
"z": 0
},
"handleBounds": {
"source": [
{
"id": "output",
"position": "bottom",
"x": 107.1171875,
"y": 48,
"width": 30,
"height": 10
}
],
"target": [
{
"id": "input",
"position": "top",
"x": 107.1171875,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": False,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": 53.33333333333334,
"y": 160.33333333333334
},
"data": {
"label": "4278cc25-1699-4532-b4ef-705c63d455ee",
"node_type": "function",
"name": "pytest2",
"after_flatten": False
},
"events": {},
"io_type": "default"
},
"targetNode": {
"id": "9d2e796a-d424-48b8-a0b0-0d8b2143648c",
"type": "output",
"dimensions": {
"width": 238,
"height": 42
},
"computedPosition": {
"x": -21.666666666666686,
"y": 313,
"z": 1000
},
"handleBounds": {
"source": [],
"target": [
{
"id": "input",
"position": "top",
"x": 103.828125,
"y": -4,
"width": 30,
"height": 10
}
]
},
"selected": True,
"dragging": False,
"resizing": False,
"initialized": False,
"isParent": False,
"position": {
"x": -21.666666666666686,
"y": 313
},
"data": {
"label": "9d2e796a-d424-48b8-a0b0-0d8b2143648c",
"node_type": "stream",
"stream_type": "logs",
"stream_name": "pyteststream2",
"org_id": "org_pytest_data"
},
"events": {},
"io_type": "output"
},
"sourceX": 175.45052083333334,
"sourceY": 218.33333333333334,
"targetX": 97.16145833333331,
"targetY": 309
}
],
"org": org_id
}
# Create a pipeline
resp_create_pipeline = session.post(f"{url}api/{org_id}/pipelines", json=pipeline_payload)
assert resp_create_pipeline.status_code == expected_status, f"Expected status code {expected_status} but got {resp_create_pipeline.status_code}"
print("Create pipeline response:", resp_create_pipeline.json())
assert resp_create_pipeline.status_code == 400, "Invalid pipeline: After Flatten must be checked if a previous FunctionNode already checked it in the same branch."