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)