def test_delete_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.delete_project() mock_input.assert_called_once() mock_menu().draw_menu.assert_called_once() mock_project_list.assert_called_once()
def test_delete_project(self, mock_print, mock_project_list, mock_menu, mock_input, mock_project_deletion): mock_project_list.return_value = {'name': 'id'} mock_menu().draw_menu.return_value = 'name' mock_input.return_value = '' bdo = BDO() bdo.delete_project() mock_input.assert_called_once() mock_menu().draw_menu.assert_called_once() mock_project_list.assert_called_once() mock_project_deletion.assert_called_once() self.assertEqual(mock_print.call_count, 2)