From 2bf8bb3720992c16725330f94be573d8aee1bcae Mon Sep 17 00:00:00 2001
From: Neha P <144308844+neha00290@users.noreply.github.com>
Date: Fri, 10 Jan 2025 19:17:47 +0530
Subject: [PATCH] test: add test function pytest cases (#5667)

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

## Summary by CodeRabbit

- **Tests**
	- Added comprehensive test cases for API function creation
- Included tests for invalid payloads, text-only inputs, and special
character scenarios
	- Verified successful function creation with valid payloads

- **Chores**
- Updated `.gitignore` to exclude temporary directories, configuration
files, and build artifacts

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
---
 .gitignore                                |  14 +++
 tests/api-testing/tests/test_functions.py | 147 ++++++++++++++++++++++
 2 files changed, 161 insertions(+)

diff --git a/.gitignore b/.gitignore
index 73ee640edd..7a9e677154 100644
--- a/.gitignore
+++ b/.gitignore
@@ -21,6 +21,20 @@ flamegraph.html
 
 
 
+/tmp/
+/data/
+/.idea/
+/.vscode
+/.env
+/.env*
+/integration-tests
+/download/
+
+#web distribution
+/web/dist
+/web/.env
+
+
 /tmp/
 /data/
 /.idea/
diff --git a/tests/api-testing/tests/test_functions.py b/tests/api-testing/tests/test_functions.py
index b05d58fe8f..39492f0d2c 100644
--- a/tests/api-testing/tests/test_functions.py
+++ b/tests/api-testing/tests/test_functions.py
@@ -202,3 +202,150 @@ def test_e2e_addDeleteStreamFunction(create_session, base_url):
     assert (
         resp_delete_function.status_code == 200
     ), f"Deleting this function, but got {resp_delete_function.status_code} {resp_delete_function.content}"
+
+
+def test_e2e_testinvalidfunction(create_session, base_url):
+    """Running an E2E test for invalid test function"""
+
+    session = create_session
+    org_id = "default"
+    payload = {
+        "function": ".test=2jehwkhhe\n.",
+        "events": [
+            {
+                "_timestamp": 1735128523652186,
+                "job": "test",
+                "level": "info",
+                "log": "test message for openobserve"
+            },
+            {
+                "_timestamp": 1735128522644223,
+                "job": "test",
+                "level": "info",
+                "log": "test message for openobserve"
+            }
+        ]
+    }
+
+    resp_test_function = session.post(
+        f"{base_url}api/{org_id}/functions/test", json=payload
+    )
+
+    print(resp_test_function.content)
+    assert (
+        resp_test_function.status_code == 400
+    ), f"Invalid function creation, but got {resp_test_function.status_code} {resp_test_function.content}"
+
+
+def test_e2e_onlytextfunction(create_session, base_url):
+    """Running an E2E test for invalid test function"""
+
+    session = create_session
+    org_id = "default"
+    payload = {
+        "function": "test",
+        "events": [
+            {
+            "_timestamp": 1735128523652186,
+            "job": "test",
+            "level": "info",
+            "log": "test message for openobserve"
+            },
+            {
+            "_timestamp": 1735128522644223,
+            "job": "test",
+            "level": "info",
+            "log": "test message for openobserve"
+            }
+        ]
+        }
+
+    resp_test_function = session.post(
+        f"{base_url}api/{org_id}/functions/test", json=payload
+    )
+
+    print(resp_test_function.content)
+    assert (
+        resp_test_function.status_code == 400
+    ), f"Invalid function creation, but got {resp_test_function.status_code} {resp_test_function.content}"
+
+
+
+def test_e2e_testonlyspecialcharfunction(create_session, base_url):
+    """Running an E2E test for invalid test function"""
+
+    session = create_session
+    org_id = "default"
+    payload = {
+        "function": "=====",
+        "events": [
+            {
+                "_timestamp": 1735128523652186,
+                "job": "test",
+                "level": "info",
+                "log": "test message for openobserve"
+            },
+            {
+                "_timestamp": 1735128522644223,
+                "job": "test",
+                "level": "info",
+                "log": "test message for openobserve"
+            }
+        ]
+    }
+
+    resp_test_function = session.post(
+        f"{base_url}api/{org_id}/functions/test", json=payload
+    )
+
+    print(resp_test_function.content)
+    assert (
+        resp_test_function.status_code == 400
+    ), f"Invalid function creation, but got {resp_test_function.status_code} {resp_test_function.content}"
+
+def test_e2e_testfunction(create_session, base_url):
+    """Running an E2E test for the test function"""
+
+    session = create_session
+    # Define organization ID and payload
+    org_id = "default"
+    payload = {
+        "function": ".test=2\n.",
+        "events": [
+            {
+                "_timestamp": 1735128523652186,
+                "job": "test",
+                "level": "info",
+                "log": "test message for openobserve"
+            },
+            {
+                "_timestamp": 1735128522644223,
+                "job": "test",
+                "level": "info",
+                "log": "test message for openobserve"
+            }
+        ]
+    }
+
+    # Send a POST request to the API
+    resp_test_function = session.post(
+        f"{base_url}api/{org_id}/functions/test", json=payload
+    )
+
+    # Print the response content for debugging
+    print(resp_test_function.content)
+
+    # Assert that the response status code is 200
+    assert resp_test_function.status_code == 200, (
+        f"Expected status code 200, but got {resp_test_function.status_code}. "
+        f"Response content: {resp_test_function.content}"
+    )
+
+    # Parse the response JSON
+    response_json = resp_test_function.json()
+
+    # Assert that `test=2` is present in each event in the response
+    for result in response_json.get("results", []):
+        assert result["event"].get("test") == 2, (
+            f"Expected 'test=2' in the response event, but got {result}."
+        )