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 create_user_staging_record(domain,
user_id=None,
username=None,
doc_type=None,
base_doc=None):
record = UserStagingTable(
user_id=user_id or uuid.uuid().hex,
username=username or 'user-staging',
doc_type=doc_type or 'CommCareUser',
base_doc=base_doc or 'CouchUser',
domain=domain,
is_active=True,
is_staff=False,
is_superuser=False,
date_joined=datetime.utcnow(),
)
record.save()
return record
def test_loading_form_fact(self):
batch = self.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 create_user_staging_record(
domain,
user_id=None,
username=None,
doc_type=None,
base_doc=None,
batch_id=None,
domain_memberships=None):
record = UserStagingTable(
user_id=user_id or uuid.uuid4().hex,
username=username or 'user-staging',
doc_type=doc_type or 'CommCareUser',
base_doc=base_doc or 'CouchUser',
domain=domain,
is_active=True,
is_staff=False,
is_superuser=False,
date_joined=datetime.utcnow(),
batch_id=batch_id,
domain_memberships=domain_memberships,
)
record.save()
return record
def test_loading_form_fact(self):
start = datetime.utcnow() - timedelta(days=3)
end = datetime.utcnow() + timedelta(days=3)
DomainStagingTable.commit(start, end)
self.assertEqual(DomainStagingTable.objects.count(),
len(self.domain_records))
DomainDim.commit(start, end)
self.assertEqual(DomainDim.objects.count(), len(self.domain_records))
UserStagingTable.commit(start, end)
self.assertEqual(UserStagingTable.objects.count(),
len(self.user_records))
UserDim.commit(start, end)
self.assertEqual(UserDim.objects.count(), len(self.user_records))
FormStagingTable.commit(start, end)
self.assertEqual(FormStagingTable.objects.count(),
len(self.form_records))
FormFact.commit(start, end)
self.assertEqual(FormFact.objects.count(), len(self.form_records))
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()
