|
9 | 9 | from tests.factories.use_case_factory import UseCaseFactory
|
10 | 10 | from tests.factories.use_case_version_factory import UseCaseVersionFactory
|
11 | 11 | from tests.factories.variable_factory import VariableFactory
|
| 12 | +from use_case_executor.adapters.database_repository.models import UseCaseVersion |
12 | 13 | from use_case_executor.adapters.database_repository.models.use_case import UseCase
|
13 | 14 | from use_case_executor.adapters.database_repository.models.variable import Variable
|
14 | 15 | from use_case_executor.domain.flow.edge_data.quick_reply_data import QuickReplyData
|
@@ -260,14 +261,13 @@ def test_clone_use_case_on_a_use_case_with_production_only_setting(
|
260 | 261 | assert new_use_case.service_record_id == payload["service_record_id"]
|
261 | 262 | assert new_use_case.intent_record_id == payload["intent_record_id"]
|
262 | 263 |
|
| 264 | + assert new_use_case.production_version_id is None |
263 | 265 | if has_production_flow:
|
264 | 266 | assert len(new_use_case.versions) == 1
|
265 | 267 | version = new_use_case.versions[0]
|
266 |
| - assert new_use_case.production_version_id == version.id |
267 | 268 | assert version.flow == prod_version.flow
|
268 | 269 | else:
|
269 | 270 | assert len(new_use_case.versions) == 0
|
270 |
| - assert new_use_case.production_version_id is None |
271 | 271 |
|
272 | 272 | assert response.json() == {
|
273 | 273 | "instance_id": payload["instance_id"],
|
@@ -538,10 +538,12 @@ def test_creates_new_variables_if_cloning_to_other_instance( # pylint: disable=
|
538 | 538 | assert new_variable.name == variable.name
|
539 | 539 | assert new_variable.type == variable.type
|
540 | 540 |
|
541 |
| - new_use_case = session.query(UseCase).filter(UseCase.uuid != use_case.uuid).one() |
542 |
| - new_flow = common.dict_flow_to_domain_flow( |
543 |
| - flow=new_use_case.production_version.flow |
| 541 | + new_use_case_version = ( |
| 542 | + session.query(UseCaseVersion) |
| 543 | + .filter(UseCaseVersion.use_case_uuid != use_case.uuid) |
| 544 | + .one() |
544 | 545 | )
|
| 546 | + new_flow = common.dict_flow_to_domain_flow(flow=new_use_case_version.flow) |
545 | 547 |
|
546 | 548 | new_answer_node_data = new_flow.get_node(uuid=answer_node_uuid).data
|
547 | 549 | assert isinstance(new_answer_node_data, AnswerData)
|
|
0 commit comments