Skip to content

Commit 606bd6f

Browse files
committed
add test for passing of conversation id
1 parent 9b0e8e8 commit 606bd6f

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

tests/cli/test_rasa_interactive.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import argparse
22
import pytest
33
from typing import Callable, Text
4-
from unittest.mock import Mock
4+
from unittest.mock import Mock, ANY
55

66
from _pytest.monkeypatch import MonkeyPatch
77
from _pytest.pytester import RunResult
@@ -158,3 +158,22 @@ def test_no_interactive_without_core_data(
158158

159159
mock.train_model.assert_not_called()
160160
mock.perform_interactive_learning.assert_not_called()
161+
162+
163+
def test_pass_conversation_id_to_interactive_learning(monkeypatch: MonkeyPatch):
164+
from rasa.core.train import do_interactive_learning
165+
from rasa.core.training import interactive as interactive_learning
166+
167+
parser = argparse.ArgumentParser()
168+
sub_parser = parser.add_subparsers()
169+
interactive.add_subparser(sub_parser, [])
170+
171+
expected_conversation_id = "🎁"
172+
args = parser.parse_args(["interactive", "--conversation-id", expected_conversation_id, "--skip-visualization", ])
173+
174+
_serve_application = Mock()
175+
monkeypatch.setattr(interactive_learning, "_serve_application", _serve_application)
176+
177+
do_interactive_learning(args, Mock())
178+
179+
_serve_application.assert_called_once_with(ANY, ANY, True, expected_conversation_id)

0 commit comments

Comments
 (0)