def setUpClass(cls): super(TestUserDim, cls).setUpClass() cls.records = [ create_user_staging_record( cls.domain, user_id=SYSTEM_USER_ID, username='******', ), create_user_staging_record( cls.domain, user_id=DEMO_USER_ID, username='******', ), create_user_staging_record( cls.domain, user_id=COMMTRACK_USERNAME, username='******', ), create_user_staging_record(cls.domain, user_id='beeboobop', username='******', doc_type='WebUser'), create_user_staging_record( cls.domain, user_id='greengoblin', username='******', ), ] cls.batch = get_default_batch()
def setUpClass(cls): super(TestUserGroupDim, cls).setUpClass() cls.blue_dog = create_user_staging_record(cls.domain, username='******') cls.black_dog = create_user_staging_record(cls.domain, username='******') cls.yellow_cat = create_user_staging_record(cls.domain, username='******') cls.batch = get_default_batch()
def test_stage_records(self): batch = get_default_batch() self.assertEqual(self.staging_table_cls.objects.count(), 0) self.staging_table_cls.commit(batch) self.assertEqual(self.staging_table_cls.objects.count(), len(self.records)) self.staging_table_cls.commit(batch) self.assertEqual(self.staging_table_cls.objects.count(), len(self.records))
def test_stage_records_bulk(self): batch = get_default_batch() # 1 Query for clearing records # 1 Query for inserting recorrds with self.assertNumQueries(2, using=self.using): GroupStagingTable.commit(batch) # 1 Query for clearing records # 2 Queries for inserting recorrds with self.assertNumQueries(3, using=self.using): with patch('corehq.warehouse.utils.DJANGO_MAX_BATCH_SIZE', 2): GroupStagingTable.commit(batch) self.assertEqual(GroupStagingTable.objects.count(), 3)
def test_loading_form_fact(self): batch = get_default_batch() DomainStagingTable.commit(batch) self.assertEqual(DomainStagingTable.objects.count(), len(self.domain_records)) DomainDim.commit(batch) self.assertEqual(DomainDim.objects.count(), len(self.domain_records)) UserStagingTable.commit(batch) self.assertEqual(UserStagingTable.objects.count(), len(self.user_records)) UserDim.commit(batch) self.assertEqual(UserDim.objects.count(), len(self.user_records)) FormStagingTable.commit(batch) self.assertEqual(FormStagingTable.objects.count(), len(self.form_records)) FormFact.commit(batch) self.assertEqual(FormFact.objects.count(), len(self.form_records))
def setUpClass(cls): super(TestAppDim, cls).setUpClass() cls.batch = get_default_batch()
def setUpClass(cls): super(TestLocationDim, cls).setUpClass() cls.batch = get_default_batch()