def test_create_gpm_invalid_gender(self, mock_print, mock_input): mock_input.side_effect = [ 'name', 'pass@Pas123', 'username', 'state', 'district', '232323', '21', 'male', '' ] bdo = BDO() bdo.create_gpm() mock_print.assert_called_once() self.assertEqual(mock_input.call_count, 9)
def test_create_gpm(self, mock_db, mock_print, mock_input, mock_encrypt): mock_input.side_effect = ['name', 'pass@Pas123', 'username', 'state', 'district', '232323', '21', 'M'] bdo = BDO() bdo.create_gpm() mock_db().add_user.assert_called_once() mock_db().add_personal_details.assert_called_once() mock_db().commit_data.assert_called_once() mock_encrypt.assert_called_once() mock_print.assert_called_once() self.assertEqual(mock_input.call_count, 8)
def test_create_gpm_invalid_password(self, mock_print, mock_input): mock_input.side_effect = ['name', 'pass@', ''] bdo = BDO() bdo.create_gpm() mock_print.assert_called_once() self.assertEqual(mock_input.call_count, 3)
def test_create_gpm_exception(self, mock_print, mock_input, mock_db): bdo = BDO() bdo.create_gpm() mock_input.assert_called_once() mock_print.assert_called_once() mock_db().rollback_data.assert_called_once()