def test_update_project_back(self, mock_project_list, mock_menu, mock_input):
mock_project_list.return_value = {'name1': 'id1'}
mock_menu().draw_menu.side_effect = ['BACK']
bdo = BDO()
bdo.update_project()
mock_input.assert_called_once()
mock_menu().draw_menu.assert_called_once()
mock_project_list.assert_called_once()
def test_update_project_invalid_area(self, mock_print, mock_menu, mock_input, mock_project_list, mock_validation):
mock_input.return_value = ''
mock_project_list.return_value = {'name': 'id'}
mock_menu().draw_menu.side_effect = ['name', 'AREA_OF_PROJECT']
mock_validation.is_int.return_value = False
bdo = BDO()
bdo.update_project()
self.assertEqual(mock_input.call_count, 4)
self.assertEqual(mock_print.call_count, 3)
self.assertEqual(mock_menu().draw_menu.call_count, 2)
mock_project_list.assert_called_once()
mock_validation.is_int.assert_called_once()
def test_update_project_type(self, mock_print, mock_menu, mock_input, mock_db, mock_project_list):
mock_input.return_value = ''
mock_project_list.return_value = {'name': 'id'}
mock_menu().draw_menu.side_effect = ['name', 'PROJECT_TYPE', 'road_project']
bdo = BDO()
bdo.update_project()
self.assertEqual(mock_input.call_count, 3)
self.assertEqual(mock_print.call_count, 4)
self.assertEqual(mock_menu().draw_menu.call_count, 3)
mock_db().update_project.assert_called_once()
mock_db().commit_data.assert_called_once()
mock_project_list.assert_called_once()
def test_update_project_invalid_end_date(self, mock_print, mock_menu, mock_input, mock_project_list,
mock_validation, mock_db):
mock_input.return_value = ''
mock_project_list.return_value = {'name': 'id'}
mock_menu().draw_menu.side_effect = ['name', 'ESTIMATED_END_DATE']
mock_validation.end_date.return_value = False
bdo = BDO()
bdo.update_project()
self.assertEqual(mock_input.call_count, 4)
self.assertEqual(mock_print.call_count, 3)
self.assertEqual(mock_menu().draw_menu.call_count, 2)
mock_project_list.assert_called_once()
mock_validation.end_date.assert_called_once()
def test_update_project_invalid_labour_numbers(self, mock_print, mock_menu, mock_input, mock_project_list
, mock_validation):
mock_input.return_value = ''
mock_project_list.return_value = {'name': 'id'}
mock_menu().draw_menu.side_effect = ['name', 'TOTAL_LABOUR_REQUIRED']
mock_validation.is_int.return_value = False
bdo = BDO()
bdo.update_project()
self.assertEqual(mock_input.call_count, 4)
self.assertEqual(mock_print.call_count, 3)
self.assertEqual(mock_menu().draw_menu.call_count, 2)
mock_project_list.assert_called_once()
mock_validation.is_int.assert_called_once()
