[library/python/pytest/yatest_tools] Fixed module discovering for tests with CONFTEST_LOAD_POLICY_LOCAL()

1 files changed, 37 insertions, 7 deletions

diff --git a/library/python/pytest/ut/test_tools.py b/library/python/pytest/ut/test_tools.py
index 6368b9c2fa..de85b1a6e0 100644
--- a/library/python/pytest/ut/test_tools.py
+++ b/library/python/pytest/ut/test_tools.py
@@ -1,7 +1,11 @@
 import pytest
 import sys
 
-from library.python.pytest.yatest_tools import split_node_id
+from library.python.pytest import config
+from library.python.pytest import yatest_tools
+
+
+config.set_test_mode()
 
 
 @pytest.fixture(params=["", "[param1,param2]"])
@@ -23,7 +27,7 @@ def parameters(request):
     ),
 )
 def test_split_node_id_without_path(parameters, node_id, expected_class_name, expected_test_name):
-    got = split_node_id(node_id + parameters)
+    got = yatest_tools.split_node_id(node_id + parameters)
     assert (expected_class_name, expected_test_name + parameters) == got
 
 
@@ -42,16 +46,20 @@ def test_split_node_id_without_path(parameters, node_id, expected_class_name, ex
             "package.test_script.py::class_name",
             "test_name",
         ),
-        # If module is not found in sys.extra_modules use basename as a class name
-        ("/arcadia/root/package/test_script2.py::test_name", "test_script2.py", "test_name"),
     ),
 )
 def test_split_node_id_with_path(mocker, parameters, node_id, expected_class_name, expected_test_name):
     mocker.patch.object(sys, 'extra_modules', sys.extra_modules | {'__tests__.package.test_script'})
-    got = split_node_id(node_id + parameters)
+    got = yatest_tools.split_node_id(node_id + parameters)
     assert (expected_class_name, expected_test_name + parameters) == got
 
 
+def test_missing_module(parameters):
+    # An errno must be raised if module is not found in sys.extra_modules
+    with pytest.raises(yatest_tools.MissingTestModule):
+        yatest_tools.split_node_id("/arcadia/root/package/test_script2.py::test_name" + parameters)
+
+
 @pytest.mark.parametrize(
     "node_id,expected_class_name,expected_test_name",
     (
@@ -61,6 +69,28 @@ def test_split_node_id_with_path(mocker, parameters, node_id, expected_class_nam
         ("package.test_script.py::class_name::subclass_name::test_name", "package.test_script.py", "test_suffix"),
     ),
 )
-def test_split_node_id_with_test_suffix(mocker, parameters, node_id, expected_class_name, expected_test_name):
-    got = split_node_id(node_id + parameters, "test_suffix")
+def test_split_node_id_with_test_suffix(parameters, node_id, expected_class_name, expected_test_name):
+    got = yatest_tools.split_node_id(node_id + parameters, "test_suffix")
+    assert (expected_class_name, expected_test_name + parameters) == got
+
+
+@pytest.mark.parametrize(
+    "node_id,expected_class_name,expected_test_name",
+    [
+        ("/arcadia/data/b/a/test.py::test_b_a", "b.a.test.py", "test_b_a"),
+        ("/arcadia/data/a/test.py::test_a", "a.test.py", "test_a"),
+        ("/arcadia/data/test.py::test", "test.py", "test"),
+    ],
+)
+def test_path_resolving_for_local_conftest_load_policy(
+    mocker, parameters, node_id, expected_class_name, expected_test_name
+):
+    # Order matters
+    extra_modules = [
+        '__tests__.b.a.test',
+        '__tests__.test',
+        '__tests__.a.test',
+    ]
+    mocker.patch.object(sys, 'extra_modules', extra_modules)
+    got = yatest_tools.split_node_id(node_id + parameters)
     assert (expected_class_name, expected_test_name + parameters) == got