Coverage for ingadhoc-odoo-saas-adhoc / saas_provider_upgrade / tests / test_run_priority.py: 100%

1from unittest.mock import patch 

2 

3from .test_provider_upgrade_base import TestProviderUpgradeCommon 

4 

5 

6class TestRunPriority(TestProviderUpgradeCommon): 

7 @classmethod 

8 def setUpClass(cls): 

9 super().setUpClass() 

10 request_model = cls.env["helpdesk.ticket.upgrade.request"] 

11 cls.test_request = request_model._create_request(cls.upgrade_ticket.id, "test") 

12 cls.prod_request = request_model._create_request(cls.upgrade_ticket.id, "production") 

13 

14 Run = cls.env["saas.upgrade.line.request.run"] 

15 cls.test_run = Run.create( 

16 { 

17 "upgrade_line_id": cls.upgrade_line_02.id, 

18 "request_id": cls.test_request.id, 

19 } 

20 ) 

21 cls.prod_run = Run.create( 

22 { 

23 "upgrade_line_id": cls.upgrade_line_02.id, 

24 "request_id": cls.prod_request.id, 

25 } 

26 ) 

27 

28 def _assert_priority(self, run, expected_priority): 

29 RunModel = type(self.env["saas.upgrade.line.request.run"]) 

30 JobClass = self.env["base.bg"] 

31 with patch.object(RunModel, "bg_enqueue", return_value=({"dummy": True}, JobClass)) as mock_enqueue: 

32 run.enqueue() 

33 mock_enqueue.assert_called_once() 

34 _, kwargs = mock_enqueue.call_args 

35 self.assertEqual(kwargs.get("priority"), expected_priority) 

36 

37 def test_production_run_has_priority_5(self): 

38 self._assert_priority(self.prod_run, 5) 

39 

40 def test_test_run_has_priority_10(self): 

41 self._assert_priority(self.test_run, 10)