chore: import upstream snapshot with attribution

2026-07-13 13:24:47 +08:00
commit dc6079821b
1384 changed files with 261110 additions and 0 deletions
@@ -0,0 +1,116 @@
+import warnings
+from pathlib import Path
+from typing import Optional, Text
+
+import pytest
+
+from rasa.cli import scaffold
+from rasa.shared.importers.importer import TrainingDataImporter
+from tests.conftest import filter_expected_warnings
+
+
+@pytest.mark.flaky
+@pytest.mark.parametrize(
+    "config_file, domain_file, data_folder, raise_slot_warning, msg",
+    [
+        (
+            "examples/concertbot/config.yml",
+            "examples/concertbot/domain.yml",
+            "examples/concertbot/data",
+            True,
+            None,
+        ),
+        (
+            "examples/formbot/config.yml",
+            "examples/formbot/domain.yml",
+            "examples/formbot/data",
+            True,
+            None,
+        ),
+        (
+            "examples/knowledgebasebot/config.yml",
+            "examples/knowledgebasebot/domain.yml",
+            "examples/knowledgebasebot/data",
+            True,
+            "You are using an experimental feature: "
+            "Action 'action_query_knowledge_base'!",
+        ),
+        (
+            "data/test_moodbot/config.yml",
+            "data/test_moodbot/domain.yml",
+            "data/test_moodbot/data",
+            False,
+            None,
+        ),
+        (
+            "examples/reminderbot/config.yml",
+            "examples/reminderbot/domain.yml",
+            "examples/reminderbot/data",
+            True,
+            None,
+        ),
+        (
+            "examples/rules/config.yml",
+            "examples/rules/domain.yml",
+            "examples/rules/data",
+            True,
+            None,
+        ),
+    ],
+)
+def test_example_bot_training_data_raises_only_auto_fill_warning(
+    config_file: Text,
+    domain_file: Text,
+    data_folder: Text,
+    raise_slot_warning: bool,
+    msg: Optional[Text],
+):
+
+    importer = TrainingDataImporter.load_from_config(
+        config_file, domain_file, [data_folder]
+    )
+
+    if raise_slot_warning:
+        with pytest.warns() as record:
+            warnings.simplefilter(action="ignore", category=DeprecationWarning)
+
+            if msg is not None:
+                warnings.filterwarnings(action="ignore", message=msg)
+
+            importer.get_nlu_data()
+            importer.get_stories()
+
+        assert len(record) == 2
+        assert all(
+            [
+                "Slot auto-fill has been removed in 3.0 and replaced with "
+                "a new explicit mechanism to set slots." in r.message.args[0]
+                for r in record
+            ]
+        )
+    else:
+        with warnings.catch_warnings() as record:
+            importer.get_nlu_data()
+            importer.get_stories()
+
+        assert record is None
+
+
+def test_example_bot_training_on_initial_project(tmp_path: Path):
+    # we need to test this one separately, as we can't test it in place
+    # configuration suggestions would otherwise change the initial file
+    scaffold.create_initial_project(str(tmp_path))
+
+    importer = TrainingDataImporter.load_from_config(
+        str(tmp_path / "config.yml"),
+        str(tmp_path / "domain.yml"),
+        str(tmp_path / "data"),
+    )
+
+    with warnings.catch_warnings() as record:
+        importer.get_nlu_data()
+        importer.get_stories()
+
+    if record is not None:
+        records = filter_expected_warnings(record)
+        assert len(records) == 0