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()
