def test_user_types(self):
UserDimLoader().commit(self.batch)
self.assertEqual(UserDim.objects.count(), 5)
self.assertEqual(
UserDim.objects.filter(user_type=SYSTEM_USER_TYPE).first().user_id,
SYSTEM_USER_ID,
)
self.assertEqual(
UserDim.objects.filter(user_type=DEMO_USER_TYPE).first().user_id,
DEMO_USER_ID,
)
self.assertEqual(
UserDim.objects.filter(
user_type=COMMCARE_SUPPLY_USER_TYPE).first().user_id,
COMMTRACK_USERNAME,
)
self.assertEqual(
UserDim.objects.filter(user_type=MOBILE_USER_TYPE).first().user_id,
'greengoblin',
)
self.assertEqual(
UserDim.objects.filter(user_type=WEB_USER_TYPE).first().user_id,
'beeboobop',
)
def tearDownClass(cls):
cls.user.delete()
UserStagingLoader().clear_records()
UserDimLoader().clear_records()
CommitRecord.objects.all().delete()
Batch.objects.all().delete()
super(TestUserLoad, cls).tearDownClass()
def test_basic_user_group_insert(self):
UserDimLoader().commit(self.batch)
self.assertEqual(UserDim.objects.count(), 3)
# Setup group records to have multiple users
dogs = create_group_staging_record(
self.domain,
'dogs',
user_ids=[self.blue_dog.user_id, self.black_dog.user_id],
batch_id=self.batch.id)
create_group_staging_record(self.domain,
'cats',
user_ids=[self.yellow_cat.user_id],
batch_id=self.batch.id)
GroupDimLoader().commit(self.batch)
self.assertEqual(GroupDim.objects.count(), 2)
UserGroupDimLoader().commit(self.batch)
self.assertEqual(UserGroupDim.objects.count(), 3)
dog_relations = UserGroupDim.objects.filter(
group_dim=GroupDim.objects.get(group_id=dogs.group_id))
self.assertEqual(
dog_relations.count(),
2,
)
self.assertEqual(
set(dog_relations.values_list('user_dim_id', flat=True)),
set(
UserDim.objects.filter(user_id__in=[
self.blue_dog.user_id, self.black_dog.user_id
]).values_list('id', flat=True)),
)
def tearDownClass(cls):
GroupStagingLoader().clear_records()
UserStagingLoader().clear_records()
GroupDimLoader().clear_records()
UserDimLoader().clear_records()
UserGroupDimLoader().clear_records()
super(TestUserGroupDim, cls).tearDownClass()
def bootstrap_user_staging(cls):
create_user_staging_record(
domain='test1',
user_id='u1',
username='******',
doc_type='CommCareUser',
batch_id=cls.batch.id,
)
create_user_staging_record(
domain='test1',
user_id='u2',
username='******',
doc_type='CommCareUser',
batch_id=cls.batch.id,
)
create_user_staging_record(domain=None,
username='******',
user_id='u3',
doc_type='WebUser',
batch_id=cls.batch.id,
domain_memberships=[
{
'domain': 'test1',
'is_admin': True
},
{
'domain': 'test2',
'is_admin': False
},
])
UserDimLoader().commit(cls.batch)
def tearDownClass(cls):
for record in cls.records:
record.delete()
UserDimLoader().clear_records()
UserStagingLoader().clear_records()
super(TestUserDim, cls).tearDownClass()
def tearDownClass(cls):
DomainMembershipDimLoader().clear_records()
UserDimLoader().clear_records()
UserStagingLoader().clear_records()
super(TestDomainMembershipDim, cls).tearDownClass()