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 _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 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) service.decrypt_attr_value(enc, self.key).should.equal(data)