"""Tests for move_target model. Field display_name in move_target model would be deprecated soon. Field escalation_type has been added and is a replica of display_name. """ from product_review.models import move_target as move_target_model from tests.utils.db_utils import create_test_move_target def test_get_items_found(): """Test get_items found.""" create_test_move_target( moved_to_target_id=10, target_group_id=1, display_name="some name", show_in_ui=0, escalation_type="some name", ) create_test_move_target( moved_to_target_id=14, target_group_id=2, display_name="another name", show_in_ui=0, escalation_type="another name", ) items = move_target_model.get_items(target_group_id=2) assert items == [ { "moved_to_target_id": 14, "target_group_id": 2, "display_name": "another name", "orchadmin_user_id": None, "show_in_ui": False, "escalation_type": "another name", } ] def test_get_items_not_found(): """Test get_items not found.""" create_test_move_target( moved_to_target_id=10, target_group_id=1, display_name="some name", show_in_ui=0, escalation_type="some name", ) mock_move_target_1 = create_test_move_target( moved_to_target_id=14, target_group_id=2, display_name="another name", show_in_ui=0, escalation_type="another name", ) items_0 = move_target_model.get_items(target_group_id=2) items_1 = move_target_model.get_items(target_group_id=4) assert items_0 == [mock_move_target_1.to_dict()] assert items_1 == [] def test_get_move_target_found(): """Test get_move_target found.""" create_test_move_target( moved_to_target_id=1, target_group_id=2, display_name="some display name", escalation_type="some display name", ) move_target_item = move_target_model.get_move_target(1) assert move_target_item == { "moved_to_target_id": 1, "target_group_id": 2, "display_name": "some display name", "orchadmin_user_id": None, "show_in_ui": True, "escalation_type": "some display name", } def test_get_move_target_not_found(): """Test get_move_target not found.""" create_test_move_target( moved_to_target_id=1, target_group_id=2, display_name="some display name", escalation_type="some display name", ) assert not move_target_model.get_move_target(3) def test_delete(): """Test delete.""" create_test_move_target( moved_to_target_id=1, target_group_id=2, display_name="some display name", escalation_type="some display name", ) move_target_item = move_target_model.get_move_target(1) assert move_target_item == { "moved_to_target_id": 1, "target_group_id": 2, "display_name": "some display name", "orchadmin_user_id": None, "show_in_ui": True, "escalation_type": "some display name" } move_target_model.delete(1) assert not move_target_model.get_move_target(1)