def test_on_instance_delete(self): expense_type = PublicsTravelExpenseTypeFactory() self.assertEqual(expense_type.deleted_at, EPOCH_ZERO) expense_type.delete() self.assertNotEqual(expense_type.deleted_at, EPOCH_ZERO)
def test_single_endpoint(self): expense_type = PublicsTravelExpenseTypeFactory() response = self.forced_auth_req('get', reverse('public:expense_types'), user=self.unicef_staff) response_json = json.loads(response.rendered_content) self.assertEqual(len(response_json), 1) expense_type.delete() response = self.forced_auth_req('get', reverse('public:expense_types'), user=self.unicef_staff) response_json = json.loads(response.rendered_content) self.assertEqual(len(response_json), 0) response = self.forced_auth_req('get', reverse('public:missing_expense_types'), user=self.unicef_staff) self.assertEqual(response.status_code, 400) response_json = json.loads(response.rendered_content) self.assertEqual(response_json, {'values': ['This list may not be empty.']}) response = self.forced_auth_req('get', reverse('public:missing_expense_types'), data={'values': [expense_type.pk]}, user=self.unicef_staff) self.assertEqual(response.status_code, 200)
def test_travel_expense_type(self): instance = PublicsTravelExpenseTypeFactory.build(title='xyz') self.assertEqual(six.text_type(instance), u'xyz') instance = PublicsTravelExpenseTypeFactory.build( title=u'R\xe4dda Barnen') self.assertEqual(six.text_type(instance), u'R\xe4dda Barnen')
def test_personal_number_usage(self): travel = self.prepare_travel() travel_agent_expense_type = PublicsTravelExpenseTypeFactory( title='Travel Agent', vendor_number='trav01') eur = PublicsCurrencyFactory(name='EUR', code='eur') # Add expenses Expense.objects.create(travel=travel, type=travel_agent_expense_type, currency=eur, amount=35) # Generate invoice self.make_invoices(travel) response = self.forced_auth_req('get', reverse('t2f:vision_invoice_export'), user=self.unicef_staff) xml_data = response.content.decode('utf-8') self.assertEqual(xml_data.count('<pernr />'), 1) self.assertEqual( xml_data.count('<pernr>{}</pernr>'.format( self.traveler.profile.staff_id)), 1)
def test_endpoint(self): # This line is duplicated on purpose. Currency will have always 1+N number of queries # because of the exchange rate factory.build_batch(PublicsCurrencyFactory, 4) # Create one of each model to check if all serializers are working fine PublicsAirlineCompanyFactory() country = PublicsCountryFactory(currency=None) PublicsDSARegionFactory(country=country) PublicsBusinessAreaFactory() PublicsWBSFactory(business_area=None) PublicsGrantFactory() PublicsFundFactory() PublicsTravelExpenseTypeFactory() with self.assertNumQueries(10): response = self.forced_auth_req('get', reverse('public:static'), user=self.unicef_staff) response_json = json.loads(response.rendered_content) self.assertKeysIn([ 'currencies', 'travel_types', 'countries', 'airlines', 'travel_modes', 'expense_types', 'business_areas' ], response_json, exact=True)
def test_cost_summary(self): # Currencies huf = PublicsCurrencyFactory(name='HUF', code='huf') # Expense types et_t_food = PublicsTravelExpenseTypeFactory( title='Food', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER ) et_t_travel = PublicsTravelExpenseTypeFactory( title='Travel', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER ) et_t_other = PublicsTravelExpenseTypeFactory( title='Other', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER ) # Make a travel travel = Travel.objects.create(traveler=self.traveler, supervisor=self.unicef_staff, currency=huf) # Add expenses Expense.objects.create(travel=travel, type=et_t_food, currency=huf, amount=35) Expense.objects.create(travel=travel, type=et_t_travel, currency=huf, amount=50) Expense.objects.create(travel=travel, type=et_t_other, currency=huf, amount=15) Expense.objects.create(travel=travel, type=et_t_travel, currency=huf, amount=None) # This should not raise 500 travel.cost_summary
def test_queryset_delete(self): PublicsTravelExpenseTypeFactory() PublicsTravelExpenseTypeFactory() PublicsTravelExpenseTypeFactory() total_expense_type_count = TravelExpenseType.objects.all().count() self.assertEqual(total_expense_type_count, 3) deleted_at_epoch_zero_count = TravelExpenseType.objects.filter(deleted_at=EPOCH_ZERO).count() self.assertEqual(deleted_at_epoch_zero_count, 3) deleted_at_populated_count = TravelExpenseType.objects.exclude(deleted_at=EPOCH_ZERO).count() self.assertEqual(deleted_at_populated_count, 0) TravelExpenseType.objects.all().delete() deleted_at_epoch_zero_count = TravelExpenseType.objects.filter(deleted_at=EPOCH_ZERO).count() self.assertEqual(deleted_at_epoch_zero_count, 0) deleted_at_populated_count = TravelExpenseType.admin_objects.exclude(deleted_at=EPOCH_ZERO).count() self.assertEqual(deleted_at_populated_count, 3)
def test_expense_type_factory(self): user_1_et = PublicsTravelExpenseTypeFactory( title='User 1', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER) user_2_et = PublicsTravelExpenseTypeFactory( title='User 2', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER) travel_agent_1_et = PublicsTravelExpenseTypeFactory( title='TravelAgent', vendor_number='travel_agent_001') travel_agent_1 = TravelAgentFactory(name='TravelAgent', code='travel_agent_001', expense_type=travel_agent_1_et) travel_agent_1_ba = travel_agent_1.country.business_area travel_agent_2_et = PublicsTravelExpenseTypeFactory( title='TravelAgent 2', vendor_number='travel_agent_002') travel_agent_2 = TravelAgentFactory(name='TravelAgent 2', code='travel_agent_002', expense_type=travel_agent_2_et) travel_agent_2_ba = travel_agent_2.country.business_area self.assertNotEqual(travel_agent_1_ba, travel_agent_2_ba) workspace = self.unicef_staff.profile.country workspace.business_area_code = travel_agent_1_ba.code workspace.save() response = self.forced_auth_req('get', reverse('public:static'), user=self.unicef_staff) response_json = json.loads(response.rendered_content) self.assertEqual(len(response_json['expense_types']), 3) got_expense_type_ids = { et['id'] for et in response_json['expense_types'] } self.assertEqual(got_expense_type_ids, {user_1_et.id, user_2_et.id, travel_agent_1_et.id})
def test_cost_calculation_parking_money(self): """If expense mapping has empty key, allocate to parking money""" itinerary = ItineraryItemFactory( travel=self.travel, arrival_date=datetime(2017, 1, 1, 1, 0, tzinfo=UTC), departure_date=datetime(2017, 1, 2, 4, 0, tzinfo=UTC), dsa_region=self.budapest ) ItineraryItemFactory( travel=self.travel, arrival_date=datetime(2017, 1, 3, 1, 0, tzinfo=UTC), departure_date=datetime(2017, 1, 4, 4, 0, tzinfo=UTC), dsa_region=self.budapest ) parking_money_type = PublicsTravelExpenseTypeFactory( title='Parking money', vendor_number="", ) ExpenseFactory( travel=self.travel, type=parking_money_type, currency=self.currency_usd, amount=100 ) daily_amt = self.budapest.dsa_amount_local last_day_amount = daily_amt * (1 - DSACalculator.LAST_DAY_DEDUCTION) total = daily_amt * 3 + last_day_amount calculator = CostSummaryCalculator(self.travel) cost_summary = calculator.get_cost_summary() self.assertEqual(cost_summary["dsa"], [{ "start_date": date(2017, 1, 1), "end_date": date(2017, 1, 4), "dsa_region": itinerary.dsa_region.pk, "dsa_region_name": itinerary.dsa_region.label, "night_count": 3, "daily_rate": daily_amt, "paid_to_traveler": total, "total_amount": total, "deduction": Decimal("0.0000"), }]) self.assertEqual(cost_summary["expenses_total"], [ {"currency": self.currency_usd, "amount": Decimal("100.0")} ]) self.assertEqual(cost_summary["preserved_expenses"], Decimal("150")) self.assertEqual(cost_summary["dsa_total"], total) self.assertEqual(cost_summary["paid_to_traveler"], total) self.assertEqual(cost_summary["traveler_dsa"], total)
def prepare_travel(self): # Currencies huf = PublicsCurrencyFactory(name='HUF', code='huf') # Add wbs/grant/fund wbs_1 = PublicsWBSFactory(name='WBS #1') grant_1 = PublicsGrantFactory(name='Grant #1') wbs_1.grants.add(grant_1) fund_1 = PublicsFundFactory(name='Fund #1') grant_1.funds.add(fund_1) dsa_region = PublicsDSARegionFactory() PublicsDSARateFactory(region=dsa_region) # Expense types et_t_food = PublicsTravelExpenseTypeFactory( title='Food', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER) # Make a travel travel = Travel.objects.create(traveler=self.traveler, supervisor=self.unicef_staff, currency=huf) # Add expenses Expense.objects.create(travel=travel, type=et_t_food, currency=huf, amount=35) ItineraryItemFactory(travel=travel, departure_date=datetime(2017, 5, 10, tzinfo=UTC), arrival_date=datetime(2017, 5, 11, tzinfo=UTC), dsa_region=dsa_region) ItineraryItemFactory(travel=travel, departure_date=datetime(2017, 5, 20, tzinfo=UTC), arrival_date=datetime(2017, 5, 21, tzinfo=UTC), dsa_region=dsa_region) # Add cost assignments CostAssignment.objects.create(travel=travel, share=100, wbs=wbs_1, grant=grant_1, fund=fund_1) return travel
def test_init(self): currency_usd = PublicsCurrencyFactory(code='USD') travel = TravelFactory(currency=currency_usd) expense_type = PublicsTravelExpenseTypeFactory( title='Train cost', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER ) expense = ExpenseFactory( travel=travel, type=expense_type, currency=currency_usd, amount=100 ) dto = ExpenseDTO("vendor_number", expense) self.assertEqual(dto.vendor_number, "vendor_number") self.assertEqual(dto.label, expense_type.title) self.assertEqual(dto.currency, currency_usd) self.assertEqual(dto.amount, 100)
def setUp(self): super(InvoiceMaking, self).setUp() self.unicef_staff = UserFactory(is_staff=True) self.traveler = UserFactory() profile = self.traveler.profile profile.vendor_number = 'user0001' profile.save() country = profile.country country.business_area_code = '0060' country.save() dsa_region = PublicsDSARegionFactory() PublicsDSARateFactory(region=dsa_region) # Currencies self.huf = PublicsCurrencyFactory(name='HUF', code='huf') self.usd = PublicsCurrencyFactory(name='USD', code='usd') # Add wbs/grant/fund self.wbs_1 = PublicsWBSFactory(name='WBS #1') self.wbs_2 = PublicsWBSFactory(name='WBS #2') self.grant_1 = PublicsGrantFactory(name='Grant #1') self.grant_2 = PublicsGrantFactory(name='Grant #2') self.grant_3 = PublicsGrantFactory(name='Grant #3') self.wbs_1.grants.add(self.grant_1) self.wbs_2.grants.add(self.grant_2, self.grant_3) self.fund_1 = PublicsFundFactory(name='Fund #1') self.fund_2 = PublicsFundFactory(name='Fund #4') self.grant_1.funds.add(self.fund_1) self.grant_3.funds.add(self.fund_2) # Expense types self.et_t_food = PublicsTravelExpenseTypeFactory( title='Food', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER) self.et_t_travel = PublicsTravelExpenseTypeFactory( title='Travel', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER) self.et_t_other = PublicsTravelExpenseTypeFactory( title='Other', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER) self.et_a_nico = PublicsTravelExpenseTypeFactory( title='Nico Travel', vendor_number='a_nico') self.et_a_torben = PublicsTravelExpenseTypeFactory( title='Torben Travel', vendor_number='a_torben') # Make a travel self.travel = Travel.objects.create(traveler=self.traveler, supervisor=self.unicef_staff, currency=self.huf) ItineraryItemFactory(travel=self.travel, departure_date=datetime(2017, 5, 10, tzinfo=UTC), arrival_date=datetime(2017, 5, 11, tzinfo=UTC), dsa_region=dsa_region) ItineraryItemFactory(travel=self.travel, departure_date=datetime(2017, 5, 20, tzinfo=UTC), arrival_date=datetime(2017, 5, 21, tzinfo=UTC), dsa_region=dsa_region)
def test_completed_counts(self): currency = PublicsCurrencyFactory() expense_type = PublicsTravelExpenseTypeFactory() business_area = PublicsBusinessAreaFactory() dsa_region = PublicsDSARegionFactory() wbs = PublicsWBSFactory(business_area=business_area) grant = wbs.grants.first() fund = grant.funds.first() traveler = UserFactory(is_staff=True) traveler.profile.vendor_number = 'usrvend' traveler.profile.save() travel = TravelFactory(reference_number=make_travel_reference_number(), traveler=traveler, status=Travel.CERTIFIED, supervisor=self.unicef_staff) data = {'cost_assignments': [{'wbs': wbs.id, 'grant': grant.id, 'fund': fund.id, 'share': 100}], 'deductions': [{'date': '2016-11-03', 'breakfast': True, 'lunch': True, 'dinner': False, 'accomodation': True}], 'itinerary': [{'origin': 'Berlin', 'destination': 'Budapest', 'departure_date': '2017-04-14T17:06:55.821490', 'arrival_date': '2017-04-15T17:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': []}, {'origin': 'Budapest', 'destination': 'Berlin', 'departure_date': '2017-05-20T12:06:55.821490', 'arrival_date': '2017-05-21T12:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': []}], 'traveler': traveler.id, 'ta_required': True, 'report': 'Some report', 'currency': currency.id, 'supervisor': self.unicef_staff.id, 'expenses': [{'amount': '120', 'type': expense_type.id, 'currency': currency.id, 'document_currency': currency.id}]} act1 = TravelActivityFactory(travel_type=TravelType.PROGRAMME_MONITORING, primary_traveler=traveler) act2 = TravelActivityFactory(travel_type=TravelType.SPOT_CHECK, primary_traveler=traveler) act1.travels.add(travel) act2.travels.add(travel) partner_programmatic_visits = PartnerOrganization.objects.get(id=act1.partner.id) partner_spot_checks = PartnerOrganization.objects.get(id=act2.partner.id) response = self.forced_auth_req('post', reverse('t2f:travels:details:state_change', kwargs={'travel_pk': travel.id, 'transition_name': 'mark_as_completed'}), user=traveler, data=data) response_json = json.loads(response.rendered_content) partner_programmatic_visits_after_complete = PartnerOrganization.objects.get(id=act1.partner.id) partner_spot_checks_after_complete = PartnerOrganization.objects.get(id=act2.partner.id) self.assertEqual(response_json['status'], Travel.COMPLETED) self.assertEqual(partner_programmatic_visits.hact_values['programmatic_visits']['completed']['total'] + 1, partner_programmatic_visits_after_complete.hact_values[ 'programmatic_visits']['completed']['total']) self.assertEqual(partner_spot_checks.hact_values['spot_checks']['completed']['total'] + 1, partner_spot_checks_after_complete.hact_values['spot_checks']['completed']['total'])
def _prepare_test(self): currency = PublicsCurrencyFactory() expense_type = PublicsTravelExpenseTypeFactory() dsaregion = PublicsDSARegionFactory() airlines = PublicsAirlineCompanyFactory() data = { 'cost_assignments': [], 'deductions': [{ 'date': '2016-11-03', 'breakfast': True, 'lunch': True, 'dinner': False, 'accomodation': True }], 'itinerary': [{ 'origin': 'Berlin', 'destination': 'Budapest', 'departure_date': '2016-11-15T12:06:55.821490', 'arrival_date': '2016-11-16T12:06:55.821490', 'dsa_region': dsaregion.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [airlines.id] }, { 'origin': 'Budapest', 'destination': 'Berlin', 'departure_date': '2016-11-16T12:06:55.821490', 'arrival_date': '2016-11-17T12:06:55.821490', 'dsa_region': dsaregion.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [airlines.id] }], 'traveler': self.traveler.id, 'ta_required': True, 'supervisor': self.unicef_staff.id, 'expenses': [{ 'amount': '120', 'type': expense_type.id, 'currency': currency.id, 'document_currency': currency.id }] } response = self.forced_auth_req('post', reverse('t2f:travels:list:index'), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['cost_summary']['preserved_expenses'], None) travel_id = response_json['id'] response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'submit_for_approval' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['cost_summary']['preserved_expenses'], None) return travel_id, data
def test_multi_step_reach(self, permission_matrix_getter): permission_matrix_getter.return_value = {'travel': {}} travel_id, data = self._prepare_test() response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'approve' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) travel = Travel.objects.get(id=travel_id) self.assertEqual(travel.approved_cost_traveler, 0) self.assertEqual(travel.approved_cost_travel_agencies, 120) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'send_for_payment' }), data=response_json, user=self.traveler) response_json = json.loads(response.rendered_content) travel = Travel.objects.get(id=travel_id) self.assertEqual(travel.approved_cost_traveler, 0) self.assertEqual(travel.approved_cost_travel_agencies, 120) # Threshold not reached yet. Still send for payment data = response_json data['expenses'][0]['amount'] = '180' response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'send_for_payment' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['status'], Travel.SENT_FOR_PAYMENT) travel = Travel.objects.get(id=travel_id) self.assertEqual(travel.approved_cost_traveler, 0) self.assertEqual(travel.approved_cost_travel_agencies, 120) # Threshold reached. Send for approval currency = PublicsCurrencyFactory() # If vendor number is empty, considered as estimated travel cost # and should be included while calculating the threshold expense_type = PublicsTravelExpenseTypeFactory(vendor_number='') data = response_json data['expenses'].append({ 'amount': '41', 'type': expense_type.id, 'currency': currency.id, 'document_currency': currency.id }) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'send_for_payment' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['status'], Travel.SUBMITTED) travel = Travel.objects.get(id=travel_id) self.assertEqual(travel.approved_cost_traveler, 0) self.assertEqual(travel.approved_cost_travel_agencies, 120)
def test_almost_overlapping_trips(self): currency = PublicsCurrencyFactory() expense_type = PublicsTravelExpenseTypeFactory() dsa_rate = PublicsDSARateFactory( effective_from_date=datetime(2017, 4, 10, 16, 00, tzinfo=UTC)) dsa_region = dsa_rate.region data = { 'deductions': [], 'itinerary': [{ 'origin': 'Berlin', 'destination': 'Budapest', 'departure_date': '2017-04-14T17:06:55.821490', 'arrival_date': '2017-04-15T17:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [] }, { 'origin': 'Budapest', 'destination': 'Berlin', 'departure_date': '2017-05-20T12:06:55.821490', 'arrival_date': '2017-05-21T12:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [] }], 'activities': [], 'cost_assignments': [], 'clearances': { 'medical_clearance': 'requested', 'security_clearance': 'requested', 'security_course': 'requested' }, 'ta_required': True, 'international_travel': False, 'mode_of_travel': [ModeOfTravel.BOAT], 'traveler': self.traveler.id, 'supervisor': self.unicef_staff.id, 'start_date': '2017-04-14T16:05:00+00:00', 'end_date': '2017-05-22T15:02:13+00:00', 'currency': currency.id, 'purpose': 'Purpose', 'additional_note': 'Notes', 'expenses': [{ 'amount': '120', 'type': expense_type.id, 'currency': currency.id, 'document_currency': currency.id }] } response = self.forced_auth_req('post', reverse('t2f:travels:list:index'), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) with freeze_time(datetime(2017, 4, 14, 16, 00, tzinfo=UTC)): response = self.forced_auth_req( 'post', reverse('t2f:travels:details:state_change', kwargs={ 'travel_pk': response_json['id'], 'transition_name': 'submit_for_approval' }), data=response_json, user=self.traveler) # No error should appear, expected 200 response_json = json.loads(response.rendered_content) self.assertEqual(response.status_code, 200, response_json) response_json = json.loads(response.rendered_content) travel = Travel.objects.get(id=response_json['id']) travel.approve() travel.save() response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': response_json['id'], 'transition_name': 'send_for_payment' }), data=response_json, user=self.traveler) self.assertEqual(response.status_code, 200)
def test_state_machine_flow(self): currency = PublicsCurrencyFactory() expense_type = PublicsTravelExpenseTypeFactory() business_area = PublicsBusinessAreaFactory() dsa_region = PublicsDSARegionFactory() wbs = PublicsWBSFactory(business_area=business_area) grant = wbs.grants.first() fund = grant.funds.first() workspace = self.unicef_staff.profile.country workspace.business_area_code = business_area.code workspace.save() data = { 'cost_assignments': [{ 'wbs': wbs.id, 'grant': grant.id, 'fund': fund.id, 'share': 100 }], 'deductions': [{ 'date': '2016-11-03', 'breakfast': True, 'lunch': True, 'dinner': False, 'accomodation': True }], 'itinerary': [{ 'origin': 'Berlin', 'destination': 'Budapest', 'departure_date': '2017-04-14T17:06:55.821490', 'arrival_date': '2017-04-15T17:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [] }, { 'origin': 'Budapest', 'destination': 'Berlin', 'departure_date': '2017-05-20T12:06:55.821490', 'arrival_date': '2017-05-21T12:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [] }], 'traveler': self.traveler.id, 'ta_required': True, 'currency': currency.id, 'supervisor': self.unicef_staff.id, 'expenses': [{ 'amount': '120', 'type': expense_type.id, 'currency': currency.id, 'document_currency': currency.id }] } response = self.forced_auth_req('post', reverse('t2f:travels:list:index'), data=data, user=self.unicef_staff) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['cost_summary']['preserved_expenses'], None) travel_id = response_json['id'] response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'submit_for_approval' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) travel = Travel.objects.get(id=travel_id) self.assertIsNotNone(travel.submitted_at) self.assertIsNotNone(travel.first_submission_date) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'approve' }), data=response_json, user=self.traveler) response_json = json.loads(response.rendered_content) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'send_for_payment' }), data=response_json, user=self.traveler) response_json = json.loads(response.rendered_content) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_certified' }), data=response_json, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['non_field_errors'], [ 'Your TA has pending payments to be processed through ' 'VISION. Until payments are completed, you can not certify' ' your TA. Please check with your Finance focal point on ' 'how to proceed.' ]) travel = Travel.objects.get(id=travel_id) travel.invoices.all().update(status=Invoice.SUCCESS) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_certified' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['status'], Travel.CERTIFIED) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_completed' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['non_field_errors'], ['Field report has to be filled.']) self.assertEqual(travel.report_note, '') # None should be handled as empty string travel.report_note = None # This has to be set explicitly since serializer does not accept None response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_completed' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['non_field_errors'], ['Field report has to be filled.']) self.assertEqual(travel.report_note, None) data = response_json data['report'] = 'Something' response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_completed' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['status'], Travel.COMPLETED)
def test_state_machine_flow_invoice_disabled(self): currency = PublicsCurrencyFactory() expense_type = PublicsTravelExpenseTypeFactory() business_area = PublicsBusinessAreaFactory() dsa_region = PublicsDSARegionFactory() wbs = PublicsWBSFactory(business_area=business_area) grant = wbs.grants.first() fund = grant.funds.first() workspace = self.unicef_staff.profile.country workspace.business_area_code = business_area.code workspace.save() data = { 'cost_assignments': [{ 'wbs': wbs.id, 'grant': grant.id, 'fund': fund.id, 'share': 100 }], 'deductions': [{ 'date': '2016-11-03', 'breakfast': True, 'lunch': True, 'dinner': False, 'accomodation': True }], 'itinerary': [{ 'origin': 'Berlin', 'destination': 'Budapest', 'departure_date': '2017-04-14T17:06:55.821490', 'arrival_date': '2017-04-15T17:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [] }, { 'origin': 'Budapest', 'destination': 'Berlin', 'departure_date': '2017-05-20T12:06:55.821490', 'arrival_date': '2017-05-21T12:06:55.821490', 'dsa_region': dsa_region.id, 'overnight_travel': False, 'mode_of_travel': ModeOfTravel.RAIL, 'airlines': [] }], 'traveler': self.traveler.id, 'ta_required': True, 'currency': currency.id, 'supervisor': self.unicef_staff.id, 'expenses': [{ 'amount': '120', 'type': expense_type.id, 'currency': currency.id, 'document_currency': currency.id }] } response = self.forced_auth_req('post', reverse('t2f:travels:list:index'), data=data, user=self.unicef_staff) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['cost_summary']['preserved_expenses'], None) travel_id = response_json['id'] response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'submit_for_approval' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) travel = Travel.objects.get(id=travel_id) self.assertIsNotNone(travel.submitted_at) self.assertIsNotNone(travel.first_submission_date) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'approve' }), data=response_json, user=self.traveler) response_json = json.loads(response.rendered_content) # Go straight to sent for payment when invoicing is disabled. self.assertEqual(response_json['status'], Travel.SENT_FOR_PAYMENT) # No email has been sent regarding SENT_FOR_PAYMENT status when invoicing is disabled. subjects = [x.subject for x in mail.outbox] self.assertNotIn( 'Travel #{} sent for payment.'.format( response_json["reference_number"]), subjects) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_certified' }), data=response_json, user=self.traveler) response_json = json.loads(response.rendered_content) # No pending invoice check when invoicing is disabled. self.assertEqual(response_json['status'], Travel.CERTIFIED) response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_completed' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['non_field_errors'], ['Field report has to be filled.']) self.assertEqual(travel.report_note, '') # None should be handled as empty string travel.report_note = None # This has to be set explicitly since serializer does not accept None response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_completed' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['non_field_errors'], ['Field report has to be filled.']) self.assertEqual(travel.report_note, None) data = response_json data['report'] = 'Something' response = self.forced_auth_req('post', reverse( 't2f:travels:details:state_change', kwargs={ 'travel_pk': travel_id, 'transition_name': 'mark_as_completed' }), data=data, user=self.traveler) response_json = json.loads(response.rendered_content) self.assertEqual(response_json['status'], Travel.COMPLETED)
def setUpTestData(cls): cls.unicef_staff = UserFactory(is_staff=True) cls.currency_usd = PublicsCurrencyFactory(code='USD') cls.currency_huf = PublicsCurrencyFactory(name='Hungarian Forint', code='HUF') cls.user_et_1 = PublicsTravelExpenseTypeFactory( title='Train cost', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER ) cls.user_et_2 = PublicsTravelExpenseTypeFactory( title='Other expenses', vendor_number=TravelExpenseType.USER_VENDOR_NUMBER_PLACEHOLDER ) cls.ta_et = PublicsTravelExpenseTypeFactory(title='Travel agent') netherlands = PublicsCountryFactory(name='Netherlands', long_name='Netherlands') hungary = PublicsCountryFactory(name='Hungary', long_name='Hungary') denmark = PublicsCountryFactory(name='Denmark', long_name='Denmark') germany = PublicsCountryFactory(name='Germany', long_name='Germany') cls.amsterdam = PublicsDSARegionFactory( country=netherlands, area_name='Amsterdam', area_code='ds1' ) PublicsDSARateFactory( region=cls.amsterdam, dsa_amount_usd=100, dsa_amount_60plus_usd=60 ) cls.budapest = PublicsDSARegionFactory( country=hungary, area_name='Budapest', area_code='ds2' ) PublicsDSARateFactory( region=cls.budapest, dsa_amount_usd=200, dsa_amount_60plus_usd=120 ) cls.copenhagen = PublicsDSARegionFactory( country=denmark, area_name='Copenhagen', area_code='ds3' ) PublicsDSARateFactory( region=cls.copenhagen, dsa_amount_usd=300, dsa_amount_60plus_usd=180 ) cls.dusseldorf = PublicsDSARegionFactory( country=germany, area_name='Duesseldorf', area_code='ds4' ) PublicsDSARateFactory( region=cls.dusseldorf, dsa_amount_usd=400, dsa_amount_60plus_usd=240 ) # Delete default items created by factory cls.travel = TravelFactory(currency=cls.currency_huf) cls.travel.itinerary.all().delete() cls.travel.expenses.all().delete() cls.travel.deductions.all().delete()