def tearDownClass(cls): GroupStagingTable.clear_records() UserStagingTable.clear_records() GroupDim.clear_records() UserDim.clear_records() UserGroupDim.clear_records() super(TestUserGroupDim, cls).tearDownClass()
def test_insert_and_update(self): DomainMembershipDim.commit(self.batch) # should create 4 domain membership columns self.assertEqual(DomainMembershipDim.objects.count(), 4) # 'u3' user should have 2 membership columns for each of the domain dim_id_of_user3 = UserDim.objects.filter(user_id='u3')[0].id self.assertEqual( DomainMembershipDim.objects.filter( user_dim_id=dim_id_of_user3).count(), 2) ## test removing a domain membership # clear and add new staging record to remove a membership of 2 UserStagingTable.clear_records() create_user_staging_record(domain=None, username='******', user_id='u3', doc_type='WebUser', batch_id=self.batch.id, domain_memberships=[ { 'domain': 'test1', 'is_admin': True }, ]) DomainMembershipDim.commit(self.batch) # should create 3 domain membership columns instead of 4 self.assertEqual(DomainMembershipDim.objects.count(), 3) # u3 user should have only 1 domain-membership dim_id_of_user3 = UserDim.objects.filter(user_id='u3')[0].id self.assertEqual( DomainMembershipDim.objects.filter( user_dim_id=dim_id_of_user3).count(), 1)
def tearDownClass(cls): for user in cls.user_records: user.delete() for domain in cls.domain_records: domain.delete() FormProcessorTestUtils.delete_all_sql_forms(cls.domain) DomainStagingTable.clear_records() DomainDim.clear_records() UserStagingTable.clear_records() UserDim.clear_records() FormStagingTable.clear_records() FormFact.clear_records()
def tearDownClass(cls): for user in cls.user_records: user.delete() for domain in cls.domain_records: domain.delete() FormProcessorTestUtils.delete_all_sql_forms(cls.domain) FormStagingTable.clear_records() FormFact.clear_records() DomainStagingTable.clear_records() DomainDim.clear_records() UserStagingTable.clear_records() UserDim.clear_records() super(FormFactIntegrationTest, cls).tearDownClass()
def tearDownClass(cls): for record in cls.records: record.delete() UserDim.clear_records() UserStagingTable.clear_records() super(TestUserDim, cls).tearDownClass()
def tearDownClass(cls): DomainMembershipDim.clear_records() UserDim.clear_records() UserStagingTable.clear_records() super(TestDomainMembershipDim, cls).tearDownClass()