def test_get_restore_user_not_found(self):
with self.assertRaises(Exception):
get_restore_user(
self.domain,
self.web_user,
'{}@wrong-domain'.format(self.commcare_user.raw_username, self.domain)
)
def test_get_restore_user_not_found(self):
with self.assertRaises(Exception):
get_restore_user(
self.domain,
self.web_user,
'{}@wrong-domain'.format(self.commcare_user.raw_username, self.domain)
)
def test_get_restore_user_as_user_for_commcare_user(self):
user = get_restore_user(
self.domain,
self.commcare_user,
self.other_commcare_user
)
self.assertEquals(user.user_id, self.other_commcare_user._id)
def test_get_restore_user_as_user_for_commcare_user(self):
user = get_restore_user(
self.domain,
self.commcare_user,
self.other_commcare_user.username
)
self.assertEquals(user.user_id, self.other_commcare_user._id)
def test_get_restore_user_not_found(self):
self.assertIsNone(
get_restore_user(
self.domain,
self.web_user,
'{}@wrong-domain'.format(self.commcare_user.raw_username, self.domain)
)
)
def test_get_restore_user_as_super_user(self):
self.assertIsNotNone(
get_restore_user(
self.domain,
self.web_user,
self.super_user.username,
)
)
def test_get_restore_user_as_user(self):
self.assertIsNotNone(
get_restore_user(
self.domain,
self.web_user,
'{}@{}'.format(self.commcare_user.raw_username, self.domain)
)
)
def test_get_restore_user_as_super_user(self):
self.assertIsInstance(
get_restore_user(
self.domain,
self.web_user,
self.super_user.username,
),
OTARestoreWebUser,
)
def test_get_restore_user_as_user(self):
self.assertIsInstance(
get_restore_user(
self.domain,
self.web_user,
self.commcare_user.username
),
OTARestoreCommCareUser,
)
def setUp(self):
super(CaseClaimTests, self).setUp()
self.domain = create_domain(DOMAIN)
self.user = CommCareUser.create(DOMAIN, USERNAME, PASSWORD, None, None)
self.restore_user = get_restore_user(DOMAIN, self.user, None)
self.host_case_id = uuid4().hex
self.host_case_name = 'Dmitri Bashkirov'
self.host_case_type = 'person'
self.create_case()
def test_get_restore_user_as_super_user(self):
self.assertIsInstance(
get_restore_user(
self.domain,
self.web_user,
self.super_user
),
OTARestoreWebUser,
)
def test_get_restore_user_as_web_user(self):
self.assertIsInstance(
get_restore_user(
self.domain,
self.web_user,
self.web_user.username,
),
OTARestoreWebUser,
)
def test_get_restore_user_as_user(self):
self.assertIsInstance(
get_restore_user(
self.domain,
self.web_user,
self.commcare_user
),
OTARestoreCommCareUser,
)
def explode_cases(domain, user_id, factor, task=None):
if task:
DownloadBase.set_progress(explode_case_task, 0, 0)
explosion_id = str(uuid.uuid4())
couch_user = CommCareUser.get_by_user_id(user_id, domain)
restore_user = get_restore_user(domain, couch_user, None)
device = MockDevice(restore_user.project, restore_user, {"overwrite_cache": True, "case_sync": LIVEQUERY})
sync_result = device.restore()
cases = {}
new_case_ids = {}
for case_id, case in sync_result.cases.items():
if case.case_type != USERCASE_TYPE:
cases[case_id] = case
new_case_ids[case_id] = [str(uuid.uuid4()) for _ in range(factor - 1)]
def test_get_restore_user_web_user(self):
self.assertIsNotNone(get_restore_user(self.domain, self.web_user, None))
def test_get_restore_user_web_user(self):
self.assertIsInstance(
get_restore_user(self.domain, self.web_user, None),
OTARestoreWebUser)
def test_get_restore_user_commcare_user(self):
user = get_restore_user(self.domain, self.commcare_user, None)
self.assertIsInstance(user, OTARestoreCommCareUser)
self.assertEqual(user.request_user_id, self.commcare_user.user_id)
def test_get_restore_user_web_user(self):
self.assertIsInstance(get_restore_user(self.domain, self.web_user, None), OTARestoreWebUser)
def test_get_restore_user_commcare_user(self):
self.assertIsInstance(get_restore_user(self.domain, self.commcare_user, None), OTARestoreCommCareUser)
def test_get_restore_user_commcare_user(self):
self.assertIsNotNone(get_restore_user(self.domain, self.commcare_user, None))
