def setUp(self):
super(UserLocMapTest, self).setUp()
self.user = CommCareUser.create(self.domain.name,
'commcareuser',
'password',
phone_numbers=['123123'],
user_data={},
first_name='test',
last_name='user')
self.ct_user = CommTrackUser.wrap(self.user.to_json())
self.loc = make_loc('secondloc')
self.sp = make_supply_point(self.domain.name, self.loc)
self.cache = LocationCache()
self.mapping = UserLocMapping(self.user.username, self.user.domain,
self.cache)
def test_location_lookup_caching(self):
user2 = CommCareUser.create(self.domain.name,
'commcareuser2',
'password',
phone_numbers=['123123'],
user_data={},
first_name='test',
last_name='user')
mapping2 = UserLocMapping(user2.username, user2.domain, self.cache)
self.mapping.to_add.add(self.loc.site_code)
mapping2.to_add.add(self.loc.site_code)
with patch('corehq.apps.commtrack.util.SupplyPointCase.get_by_location'
) as get_supply_point:
self.mapping.save()
mapping2.save()
self.assertEqual(get_supply_point.call_count, 1)
def setUp(self):
super(UserLocMapTest, self).setUp()
self.user = CommCareUser.create(self.domain.name,
'commcareuser',
'password',
phone_numbers=['123123'],
user_data={},
first_name='test',
last_name='user')
MULTIPLE_LOCATIONS_PER_USER.set(self.user.domain, True,
NAMESPACE_DOMAIN)
self.loc = make_loc('secondloc')
self.sp = make_supply_point(self.domain.name, self.loc)
self.cache = SiteCodeToSupplyPointCache(self.domain.name)
self.mapping = UserLocMapping(self.user.username, self.user.domain,
self.cache)
