def setUp(self):
self.data = utils.to_bytes('Héllo!😀')
self.dirname = tempfile.mkdtemp()
logger.debug('created directory: %s', self.dirname)
with tempfile.NamedTemporaryFile(dir=self.dirname, delete=False) as tf:
logger.debug('created file: %s', tf.name)
tf.write(self.data)
self.filename = tf.name
def setUp(self):
self.data = utils.to_bytes('Héllo!😀')
self.dirname = tempfile.mkdtemp()
logger.debug('created directory: %s', self.dirname)
with tempfile.NamedTemporaryFile(dir=self.dirname, delete=False) as tf:
logger.debug('created file: %s', tf.name)
tf.write(self.data)
self.filename = tf.name
def setUp(self):
self.s3 = boto3.resource('s3')
self.data = utils.to_bytes('Héllo!😀')
self.bucket_name = 'somebucket'
self.other_bucket_name = 'otherbucket'
self.bucket = self.s3.Bucket(self.bucket_name)
self.assertIsInstance(self.bucket.create(), dict)
with tempfile.NamedTemporaryFile(delete=True) as tf:
self.key = tf.name[1:]
self.filename = 's3://{}/{}'.format(self.bucket_name, self.key)
self.object = self.bucket.Object(self.key)
self.object.put(Body=self.data)
def setUp(self):
self.s3 = boto3.resource('s3')
self.data = utils.to_bytes('Héllo!😀')
self.bucket_name = 'somebucket'
self.other_bucket_name = 'otherbucket'
self.bucket = self.s3.Bucket(self.bucket_name)
self.assertIsInstance(self.bucket.create(), dict)
with tempfile.NamedTemporaryFile(delete=True) as tf:
self.key = tf.name[1:]
self.filename = 's3://{}/{}'.format(self.bucket_name, self.key)
self.object = self.bucket.Object(self.key)
self.object.put(Body=self.data)
def _create_and_verify_good_jws(self, claims, keypairs=None):
keypairs = keypairs or self.keypairs
jws = jwts.make_jws(claims, keypairs)
jws.should.be.a(str)
verifications = [
jwts.verify_jws(jws, keypairs),
jwts.verify_jws(jws),
jwts.verify_jws(utils.to_bytes(jws)),
]
for verification in verifications:
verification.should.be.truthy
for claim in claims:
verification.should.have.key(claim).equal_to(claims[claim])
def _create_and_verify_good_jws(self, claims, keypairs=None):
keypairs = keypairs or self.keypairs
jws = jwts.make_jws(claims, keypairs)
self.assertIsInstance(jws, str)
verifications = [
jwts.verify_jws(jws, keypairs),
jwts.verify_jws(jws),
jwts.verify_jws(utils.to_bytes(jws)),
]
for verification in verifications:
self.assertTrue(verification)
for claim in claims:
self.assertIn(claim, verification)
self.assertEqual(verification.get(claim), claims[claim])
def _create_and_verify_good_jws(self, claims, keypairs=None):
keypairs = keypairs or self.keypairs
jws = jwts.make_jws(claims, keypairs)
self.assertIsInstance(jws, str)
verifications = [
jwts.verify_jws(jws, keypairs),
jwts.verify_jws(jws),
jwts.verify_jws(utils.to_bytes(jws)),
]
for verification in verifications:
self.assertTrue(verification)
for claim in claims:
self.assertIn(claim, verification)
self.assertEqual(verification.get(claim), claims[claim])
def test_remove_bytes_jws_signature(self):
jws = utils.to_bytes(jwts.make_jws({'a': 1}, self.keypairs))
jws = jwts.remove_jws_signatures(jws, self.keypairs[0].identity)
verified_msg = jwts.verify_jws(jws, self.keypairs[1:])
self.assertIsInstance(verified_msg, dict)
def test_to_bytes(self):
self.assertEqual(self.bytes_thing, utils.to_bytes(self.str_thing))
self.assertEqual(self.bytes_thing, utils.to_bytes(self.bytes_thing))
def test_read_file(self):
with plain_file_handler.read_file(self.filename, True) as data:
self.assertEqual(data, self.data)
with plain_file_handler.read_file(self.filename, False) as data:
self.assertEqual(utils.to_bytes(data), self.data)
def test_decrypt_bytes(self):
data = utils.to_bytes(self.data)
enc = service.encrypt_attr_value(data, self.key)
decrypted = service.decrypt_attr_value(enc, self.key)
self.assertEqual(decrypted, data)
def test_read_file(self):
with s3_handler.read_file(self.filename, True) as data:
self.assertEqual(data, self.data)
with s3_handler.read_file(self.filename, False) as data:
self.assertEqual(utils.to_bytes(data), self.data)
def test_decrypt_bytes(self):
data = utils.to_bytes(self.data)
enc = service.encrypt_attr_value(data, self.key)
decrypted = service.decrypt_attr_value(enc, self.key)
self.assertEqual(decrypted, data)
def test_to_bytes(self):
self.assertEqual(self.bytes_thing, utils.to_bytes(self.str_thing))
self.assertEqual(self.bytes_thing, utils.to_bytes(self.bytes_thing))
def test_decrypt_bytes(self):
data = utils.to_bytes(self.data)
enc = service.encrypt_attr_value(data, self.key)
service.decrypt_attr_value(enc, self.key).should.equal(data)
