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)) ) engine.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 = engine.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 = engine.prepare_user(user_name) self.assertEqual(existing_user.id, user.id)