def test_save(self):
"""Setup the test cases"""
SecureString.create(MY_STRING, NAME).save()
with self.assertRaises(Exception):
SecureString.create(name=NAME).save()
with self.assertRaises(Exception):
SecureString(encrypted_string='something', name=NAME).save()
def test_save(self):
"""Setup the test cases"""
SecureString.create(MY_STRING, NAME).save()
with self.assertRaises(Exception):
SecureString.create(unencrypted="", name=NAME).save()
with self.assertRaises(Exception):
SecureString(encrypted_string='something', name=NAME).save()
def save(self, *args, **kwargs):
"""Override the default save behavior to also include the secure secret"""
password = kwargs.pop('password', '')
super(ThermoFisherCloudAccount, self).save(*args, **kwargs)
if password:
secret = SecureString.create(password, self.ampliseq_secret_name)
secret.save()
def obj_create(self, bundle, request=None, **kwargs):
"""Custom create method to override the default post"""
try:
params = bundle.data
# sanity check on the input parameters
if 'encrypted_string' in params:
del params['encrypted_string']
if 'unencrypted' not in params:
return HttpApplicationError("API creation of SecureString requires \'unencrypted\' parameter")
if 'name' not in params:
return HttpApplicationError("API creation of SecureString requires \'name\' parameter")
# create and save the new object
try:
secured = SecureString.objects.get(name=params['name'])
secured.encrypt(params['unencrypted'])
secured.save()
except SecureString.DoesNotExist:
secured = SecureString.create(params['unencrypted'], params['name'])
secured.save()
return bundle
except Exception as exc:
logger.debug(str(exc))
return HttpApplicationError("Error creating secure string: " + str(exc))
def obj_create(self, bundle, request=None, **kwargs):
"""Custom create method to override the default post"""
try:
params = bundle.data
# sanity check on the input parameters
if 'encrypted_string' in params:
del params['encrypted_string']
if 'unencrypted' not in params:
return HttpApplicationError(
"API creation of SecureString requires \'unencrypted\' parameter"
)
if 'name' not in params:
return HttpApplicationError(
"API creation of SecureString requires \'name\' parameter")
# create and save the new object
try:
secured = SecureString.objects.get(name=params['name'])
secured.encrypt(params['unencrypted'])
secured.save()
except SecureString.DoesNotExist:
secured = SecureString.create(params['unencrypted'],
params['name'])
secured.save()
return bundle
except Exception as exc:
logger.debug(str(exc))
return HttpApplicationError("Error creating secure string: " +
str(exc))
def test_create(self):
"""test the objects create method"""
self.assertIsNotNone(SecureString.create(unencrypted=MY_STRING, name=NAME))
with self.assertRaises(Exception):
SecureString.objects.create(name=NAME)
with self.assertRaises(TypeError):
SecureString.objects.create(password=MY_STRING, name=NAME)
def test_create(self):
"""test the objects create method"""
self.assertIsNotNone(SecureString.create(unencrypted=MY_STRING, name=NAME))
with self.assertRaises(Exception):
SecureString.objects.create(name=NAME)
with self.assertRaises(TypeError):
SecureString.objects.create(password=MY_STRING, name=NAME)
def test_created(self):
SecureString.create(MY_STRING, NAME).save()
self.assertIsNotNone(SecureString.objects.get(name=NAME).created)
def test_decrypt(self):
"""tests the get method"""
SecureString.create(MY_STRING, NAME).save()
sp = SecureString.objects.get(name=NAME)
self.assertEqual(sp.decrypted, MY_STRING)
def test_get(self):
"""tests the get method"""
SecureString.create(MY_STRING, NAME).save()
self.assertIsNotNone(SecureString.objects.get(name=NAME))
def test_init(self):
"""make sure an init does not work"""
with self.assertRaises(Exception):
SecureString(unencrypted=MY_STRING, name=NAME)
def test_created(self):
SecureString.create(MY_STRING, NAME).save()
self.assertIsNotNone(SecureString.objects.get(name=NAME).created)
def test_decrypt(self):
"""tests the get method"""
SecureString.create(MY_STRING, NAME).save()
sp = SecureString.objects.get(name=NAME)
self.assertEqual(sp.decrypted, MY_STRING)
def test_get(self):
"""tests the get method"""
SecureString.create(MY_STRING, NAME).save()
self.assertIsNotNone(SecureString.objects.get(name=NAME))
