def setUp(self): self.company1 = CompanyFactory( name='Test', description='Test description', ) self.company2 = CompanyFactory( name='Test2', description='Test description', ) self.user1 = UserFactory() self.user2 = UserFactory() self.user3 = UserFactory() self.user4 = UserFactory() self.user5 = UserFactory() self.permission1 = CompanyUserPermissionFactory(company=self.company1, user=self.user1) self.permission2 = CompanyUserPermissionFactory(company=self.company1, user=self.user2) self.permission3 = CompanyUserPermissionFactory(company=self.company1, user=self.user3) self.permission4 = CompanyUserPermissionFactory(company=self.company2, user=self.user4) self.permission5 = CompanyUserPermissionFactory(company=self.company2, user=self.user5) self.tax_payer1 = TaxPayerFactory(company=self.company1) self.tax_payer2 = TaxPayerFactory(company=self.company1) self.tax_payer3 = TaxPayerFactory(company=self.company2)
def setUp(self): self.company = CompanyFactory() self.taxpayer = TaxPayerFactory(company=self.company) self.taxpayer_without_eb_entities = TaxPayerArgentinaFactory() self.taxpayer_with_2_eb_entities = TaxPayerArgentinaFactory() self.taxpayer_eb_entity_1 = TaxPayerEBEntityFactory( taxpayer=self.taxpayer_with_2_eb_entities) self.taxpayer_eb_entity_2 = TaxPayerEBEntityFactory( taxpayer=self.taxpayer_with_2_eb_entities) self.eb_entity_1 = EBEntityFactory() self.eb_entity_2 = EBEntityFactory()
def test_tax_payer_entity_creation(self): taxpayer = TaxPayerFactory(company=self.company, business_name='Eventbrite', workday_id='12345') self.assertEqual(taxpayer.business_name, 'Eventbrite') self.assertEqual(taxpayer.workday_id, '12345') self.assertEqual('Eventbrite', str(taxpayer)) self.assertEqual(TaxPayer.objects.last(), taxpayer)
class TestTaxpayerModel(TestCase): def setUp(self): self.company = CompanyFactory() self.taxpayer = TaxPayerFactory(company=self.company) self.taxpayer_without_eb_entities = TaxPayerArgentinaFactory() self.taxpayer_with_2_eb_entities = TaxPayerArgentinaFactory() self.taxpayer_eb_entity_1 = TaxPayerEBEntityFactory( taxpayer=self.taxpayer_with_2_eb_entities) self.taxpayer_eb_entity_2 = TaxPayerEBEntityFactory( taxpayer=self.taxpayer_with_2_eb_entities) self.eb_entity_1 = EBEntityFactory() self.eb_entity_2 = EBEntityFactory() def test_tax_payer_entity_creation(self): taxpayer = TaxPayerFactory(company=self.company, business_name='Eventbrite', workday_id='12345') self.assertEqual(taxpayer.business_name, 'Eventbrite') self.assertEqual(taxpayer.workday_id, '12345') self.assertEqual('Eventbrite', str(taxpayer)) self.assertEqual(TaxPayer.objects.last(), taxpayer) def test_get_eb_entity(self): self.taxpayer_eb_entity_2 = TaxPayerEBEntityFactory( taxpayer=self.taxpayer_without_eb_entities) self.assertEqual(self.taxpayer_without_eb_entities.get_eb_entity, self.taxpayer_eb_entity_2.eb_entity.eb_name) def test_taxpayer_creation_first_time_should_have_PENDING_status(self): self.assertEqual(self.taxpayer.taxpayer_state, "PENDING") def test_create_child_of_tax_payer_should_also_create_taxpayer_father( self): taxpayer_arg = TaxPayerArgentinaFactory(business_name='Eventbrite') self.assertTrue(isinstance(taxpayer_arg, TaxPayer)) self.assertEqual(taxpayer_arg.business_name, 'Eventbrite') self.assertEqual(1, len(TaxPayer.objects.filter(pk=taxpayer_arg.id))) def test_get_taxpayer_child(self): taxpayer_arg = TaxPayerArgentinaFactory(company=self.company) taxpayer = TaxPayer.objects.get(pk=taxpayer_arg.id) self.assertEqual(taxpayer.get_taxpayer_child(), taxpayer_arg) def test_taxpayer_has_workday_id(self): self.assertTrue(self.taxpayer.has_workday_id()) def test_taxpayer_doesnt_have_workday_id(self): taxpayer = TaxPayerArgentinaFactory(workday_id="") self.assertFalse(taxpayer.has_workday_id()) def test_taxpayer_should_create_new_relation_if_eb_entity_doesnt_exists( self): self.taxpayer_without_eb_entities.create_if_not_exist_taxpayer_eb_entity( [self.eb_entity_1, self.eb_entity_2]) self.assertEqual(2, len(self.taxpayer_without_eb_entities.eb_entities)) def test_taxpayer_with_2_eb_entity_doesnt_add_this_2_eb_entity_again(self): self.taxpayer_without_eb_entities.create_if_not_exist_taxpayer_eb_entity( [self.eb_entity_1, self.eb_entity_2]) self.taxpayer_without_eb_entities.create_if_not_exist_taxpayer_eb_entity( [self.eb_entity_1, self.eb_entity_2]) self.assertEqual(2, len(self.taxpayer_without_eb_entities.eb_entities)) def test_new_eb_entities_should_be_current(self): self.taxpayer_without_eb_entities.create_if_not_exist_taxpayer_eb_entity( [self.eb_entity_1, self.eb_entity_2]) self.assertEqual(2, len(self.taxpayer_without_eb_entities.eb_entities)) def test_taxpayer_apply_set_unused_status_should_change_2_eb_entities( self): taxpayer_eb_entities = \ self.taxpayer_with_2_eb_entities.taxpayerebentity_set.all() self.taxpayer_with_2_eb_entities.apply_function_to_all_elems( taxpayer_eb_entities, TaxPayerEBEntity.set_unused_status, ) self.assertEqual( 2, len(TaxPayerEBEntity.objects.filter(status=UNUSED_STATUS))) def test_taxpayer_apply_set_unused_status_should_change_1_eb_entity(self): taxpayer_eb_entities = [ self.taxpayer_with_2_eb_entities.taxpayerebentity_set.last() ] self.taxpayer_with_2_eb_entities.apply_function_to_all_elems( taxpayer_eb_entities, TaxPayerEBEntity.set_unused_status, ) self.assertEqual( 1, len(TaxPayerEBEntity.objects.filter(status=UNUSED_STATUS))) def test_taxpayer_apply_set_current_status_should_change_2_eb_entities( self): taxpayer_eb_entities = self.taxpayer_with_2_eb_entities.taxpayerebentity_set.all( ) self.taxpayer_with_2_eb_entities.apply_function_to_all_elems( taxpayer_eb_entities, TaxPayerEBEntity.set_current_status, ) self.assertEqual( 2, len(TaxPayerEBEntity.objects.filter(status=CURRENT_STATUS))) def test_taxpayer_current_entities_should_be_the_selected_ones(self): eb_entity_selected_1 = EBEntityFactory() eb_entity_selected_2 = EBEntityFactory() eb_entities = [eb_entity_selected_1, eb_entity_selected_2] self.taxpayer_with_2_eb_entities.set_current_eb_entities(eb_entities) taxpayer_eb_entities = self.taxpayer_with_2_eb_entities.eb_entities self.assertEqual(2, len(taxpayer_eb_entities)) self.assertEqual(eb_entities, taxpayer_eb_entities) def test_old_taxpayer_eb_entities_should_be_unused_after_update(self): taxpayer = TaxPayerArgentinaFactory() taxpayer_eb_entity_old_1 = TaxPayerEBEntityFactory(taxpayer=taxpayer) taxpayer_eb_entity_old_2 = TaxPayerEBEntityFactory(taxpayer=taxpayer) eb_entity_selected_1 = EBEntityFactory() eb_entity_selected_2 = EBEntityFactory() eb_entities = [eb_entity_selected_1, eb_entity_selected_2] taxpayer.set_current_eb_entities(eb_entities) taxpayer_eb_entity_old_1 = TaxPayerEBEntity.objects.get( pk=taxpayer_eb_entity_old_1.id) taxpayer_eb_entity_old_2 = TaxPayerEBEntity.objects.get( pk=taxpayer_eb_entity_old_2.id) self.assertEqual(str(UNUSED_STATUS), taxpayer_eb_entity_old_1.status) self.assertEqual(str(UNUSED_STATUS), taxpayer_eb_entity_old_2.status) def test_create_taxpayer_usa(self): taxpayer_usa = TaxPayerUnitedStatedFactory() self.assertEqual(TaxPayerUnitedStates.objects.last(), taxpayer_usa)
def setUp(self): self.data = { 'taxpayer': TaxPayerFactory(), 'user': UserFactory(), 'message': 'please upload updated AFIP form', }
def test_company_tax_payer_relationship(self): company = CompanyFactory() taxpayer = TaxPayerFactory(company=company) self.assertEqual(taxpayer.company.name, company.name)
def test_bank_account_creation(self): taxpayer = TaxPayerFactory() bank = BankAccountUnitedStatesFactory(taxpayer=taxpayer) self.assertEqual(bank, BankAccountUnitedStates.objects.last()) self.assertEqual(bank.taxpayer, taxpayer)
def test_address(self): taxpayer = TaxPayerFactory() contact = ContactFactory(taxpayer=taxpayer) self.assertEqual(contact, ContactInformation.objects.last()) self.assertEqual(contact.taxpayer, taxpayer)
def test_address(self): taxpayer = TaxPayerFactory() address = AddressFactory(taxpayer=taxpayer) self.assertEqual(address, Address.objects.last()) self.assertEqual(address.taxpayer, taxpayer)