def test_web_user_flag(self):
self.assertTrue(WebUser().is_web_user())
self.assertTrue(CouchUser.wrap(WebUser().to_json()).is_web_user())
self.assertFalse(CommCareUser().is_web_user())
self.assertFalse(CouchUser.wrap(CommCareUser().to_json()).is_web_user())
with self.assertRaises(NotImplementedError):
CouchUser().is_web_user()
def test_commcare_user_flag(self):
self.assertFalse(WebUser().is_commcare_user())
self.assertFalse(CouchUser.wrap(WebUser().to_json()).is_commcare_user())
self.assertTrue(CommCareUser().is_commcare_user())
self.assertTrue(CouchUser.wrap(CommCareUser().to_json()).is_commcare_user())
with self.assertRaises(NotImplementedError):
CouchUser().is_commcare_user()
def _get_username(self, user_id):
username = self.report.usernames.get(user_id)
if not username:
mc = cache.caches['default']
cache_key = "%s.%s" % (CouchUser.__class__.__name__, user_id)
try:
if mc.has_key(cache_key):
user_dict = json.loads(mc.get(cache_key))
else:
user_obj = CouchUser.get_by_user_id(user_id) if user_id else None
if user_obj:
user_dict = user_obj.to_json()
else:
user_dict = {}
cache_payload = json.dumps(user_dict)
mc.set(cache_key, cache_payload)
if user_dict == {}:
return None
else:
user_obj = CouchUser.wrap(user_dict)
username = user_obj.username
except Exception:
return None
return username
def _get_username(self, user_id):
username = self.report.usernames.get(user_id)
if not username:
mc = cache.get_cache('default')
cache_key = "%s.%s" % (CouchUser.__class__.__name__, user_id)
try:
if mc.has_key(cache_key):
user_dict = simplejson.loads(mc.get(cache_key))
else:
user_obj = CouchUser.get_by_user_id(user_id) if user_id else None
if user_obj:
user_dict = user_obj.to_json()
else:
user_dict = {}
cache_payload = simplejson.dumps(user_dict)
mc.set(cache_key, cache_payload)
if user_dict == {}:
return None
else:
user_obj = CouchUser.wrap(user_dict)
username = user_obj.username
except Exception:
return None
return username
