From b28332c5950940ef867532057b30c854286b68fe Mon Sep 17 00:00:00 2001
From: Petr Shumilov <p.shumilov@vkteam.ru>
Date: Tue, 31 Mar 2026 17:46:09 +0300
Subject: [PATCH] Add tests for get_running_fork_id

Signed-off-by: Petr Shumilov <p.shumilov@vkteam.ru>
---
 tests/python/tests/coroutine/__init__.py      |  0
 .../coroutine/php/data/component-config.yaml  |  7 +++
 tests/python/tests/coroutine/php/index.php    | 54 +++++++++++++++++++
 .../coroutine/test_get_running_fork_id.py     | 17 ++++++
 4 files changed, 78 insertions(+)
 create mode 100644 tests/python/tests/coroutine/__init__.py
 create mode 100644 tests/python/tests/coroutine/php/data/component-config.yaml
 create mode 100644 tests/python/tests/coroutine/php/index.php
 create mode 100644 tests/python/tests/coroutine/test_get_running_fork_id.py

diff --git a/tests/python/tests/coroutine/__init__.py b/tests/python/tests/coroutine/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
diff --git a/tests/python/tests/coroutine/php/data/component-config.yaml b/tests/python/tests/coroutine/php/data/component-config.yaml
new file mode 100644
index 0000000000..5ec7d22c4e
--- /dev/null
+++ b/tests/python/tests/coroutine/php/data/component-config.yaml
@@ -0,0 +1,7 @@
+entry: script
+components:
+  script:
+    image: KPHP
+    scope: Request
+    args: {}
+    links: {}
diff --git a/tests/python/tests/coroutine/php/index.php b/tests/python/tests/coroutine/php/index.php
new file mode 100644
index 0000000000..6f50bd4853
--- /dev/null
+++ b/tests/python/tests/coroutine/php/index.php
@@ -0,0 +1,54 @@
+<?php
+
+function test_get_running_fork_id() {
+  $f = function() {
+    // 1. Non-"main" fork. Before std function which is coroutine
+    $before = (int)get_running_fork_id();
+
+    sched_yield_sleep(0.01);
+
+    $nested_f = function () {
+      // 2. Yet another non-"main" fork. Before std function which is coroutine
+      $before = (int)get_running_fork_id();
+      sched_yield_sleep(0.01);
+      assert_fork_id($before);
+      return null;
+    };
+
+    assert_fork_id($before);
+    $fut = fork($nested_f());
+    assert_fork_id($before);
+    wait($fut);
+    assert_fork_id($before);
+    return null;
+  };
+  $fut = fork($f());
+
+  // 3. "main" fork. Before std function which is coroutine
+  $before = (int)get_running_fork_id();
+  sched_yield_sleep(0.01);
+  assert_fork_id($before);
+  wait($fut);
+  assert_fork_id($before);
+}
+
+
+function assert_fork_id(int $expected_fork_id) {
+  $current_fork_id = (int)get_running_fork_id();
+  if ($current_fork_id != $expected_fork_id) {
+    critical_error("unexpected fork id, expected: " . $expected_fork_id . "  got: " . $current_fork_id);
+  }
+}
+
+function main() {
+  switch ($_SERVER["PHP_SELF"]) {
+    case "/test_get_running_fork_id": {
+        test_get_running_fork_id();
+        return;
+      }
+  }
+
+  critical_error("unknown test");
+}
+
+main();
diff --git a/tests/python/tests/coroutine/test_get_running_fork_id.py b/tests/python/tests/coroutine/test_get_running_fork_id.py
new file mode 100644
index 0000000000..cae4c43011
--- /dev/null
+++ b/tests/python/tests/coroutine/test_get_running_fork_id.py
@@ -0,0 +1,17 @@
+import socket
+import pytest
+
+from python.lib.testcase import WebServerAutoTestCase
+from python.lib.http_client import RawResponse
+
+class TestErrors(WebServerAutoTestCase):
+    @classmethod
+    def extra_class_setup(cls):
+        if not cls.should_use_k2():
+            cls.web_server.update_options({
+                "--workers-num": 1
+            })
+
+    def test_get_running_fork_id(self):
+        response = self.web_server.http_request(uri="/test_get_running_fork_id", method='GET')
+        self.assertEqual(200, response.status_code)