def test_prepare_user_does_not_exist(self):
user_name = helpers.random_string()
# Sanity check; make sure the user doesn't exist yet.
self.assertEqual(
0, len(models.OqUser.objects.filter(user_name=user_name))
)
engine2.prepare_user(user_name)
# Now the user should exist.
self.assertEqual(
1, len(models.OqUser.objects.filter(user_name=user_name))
)
def setUp(self):
owner = engine2.prepare_user('openquake')
self.site_model_inp = models.Input(
owner=owner, digest='fake', path='fake',
input_type='site_model', size=0
)
self.site_model_inp.save()
def test_prepare_user_exists(self):
user_name = helpers.random_string()
existing_user = models.OqUser(
user_name=user_name, full_name=user_name, organization_id=1
)
existing_user.save()
user = engine2.prepare_user(user_name)
self.assertEqual(existing_user.id, user.id)
