def test_show_request_exception(self, mock_db, mock_print):
mock_db().get_requests.side_effect = Exception
gpm = GPM()
gpm.show_requests()
mock_print.assert_called_once()
mock_db().get_requests.assert_called_once()
mock_db().rollback_data.assert_called_once()
def test_show_request_no_request(self, mock_db, mock_print):
mock_db().get_requests.return_value.fetchall.return_value = []
mock_db().get_requests.return_value.description.return_value = ()
gpm = GPM()
gpm.show_requests()
mock_db().get_requests.assert_called_once()
mock_db().get_requests().fetchall.assert_called_once()
self.assertEqual(mock_print.call_count, 1)
def test_show_request_invalid_index(self, mock_menu, mock_db, mock_print,
mock_input, mock_table):
mock_db().get_requests.return_value.description.return_value = [
'dummy', 'dummy', 'dummy', 'dummy', 'dummy'
]
mock_db().get_requests.return_value.fetchall.return_value = [[
'ISSUES', 'dummy', 'request message', 'dummy', 'dummy'
]]
mock_menu().draw_menu.return_value = 'APPROVED'
mock_input.side_effect = ['invalid_row', '']
gpm = GPM()
gpm.show_requests()
mock_table.assert_called_once()
mock_db().get_requests.assert_called_once()
mock_db().get_requests().fetchall.assert_called_once()
self.assertEqual(mock_print.call_count, 3)
self.assertEqual(mock_input.call_count, 2)
def test_show_request_rejected_issue(self, mock_menu, mock_db, mock_print,
mock_input, mock_table):
mock_db().get_requests.return_value.description.return_value = [
'dummy', 'dummy', 'dummy', 'dummy', 'dummy'
]
mock_db().get_requests.return_value.fetchall.return_value = [[
'ISSUES', 'dummy', 'member request msg', 'dummy', 'dummy'
]]
mock_menu().draw_menu.return_value = 'REJECTED'
mock_input.side_effect = ['0', '']
gpm = GPM()
gpm.show_requests()
mock_menu().draw_menu.assert_called_once()
mock_table.assert_called_once()
mock_db().get_requests.assert_called_once()
mock_db().resolve_request.assert_called_once()
mock_db().commit_data.assert_called_once()
mock_db().get_requests().fetchall.assert_called_once()
self.assertEqual(mock_print.call_count, 3)
self.assertEqual(mock_input.call_count, 2)
