示例#1
0
 def test_set_get_flags(self):
     manager = BlobManager('', self.uri, self.secret, self.secret, 'user')
     fd = BytesIO("flag me")
     yield manager._encrypt_and_upload('blob_id', fd)
     yield manager.set_flags('blob_id', [Flags.PROCESSING])
     flags = yield manager.get_flags('blob_id')
     self.assertEquals([Flags.PROCESSING], flags)
 def test_get_empty_flags(self):
     manager = BlobManager('', self.uri, self.secret,
                           self.secret, uuid4().hex)
     fd = BytesIO("flag me")
     yield manager._encrypt_and_upload('blob_id', fd)
     flags = yield manager.get_flags('blob_id')
     self.assertEquals([], flags)
示例#3
0
 def test_cant_set_invalid_flags(self):
     manager = BlobManager('', self.uri, self.secret, self.secret, 'user')
     fd = BytesIO("flag me")
     yield manager._encrypt_and_upload('blob_id', fd)
     with pytest.raises(InvalidFlagsError):
         yield manager.set_flags('blob_id', ['invalid'])
     flags = yield manager.get_flags('blob_id')
     self.assertEquals([], flags)
示例#4
0
 def test_cant_set_invalid_flags(self):
     manager = BlobManager('', self.uri, self.secret,
                           self.secret, uuid4().hex)
     fd = BytesIO("flag me")
     yield manager._encrypt_and_upload('blob_id', fd)
     with pytest.raises(InvalidFlagsError):
         yield manager.set_flags('blob_id', ['invalid'])
     flags = yield manager.get_flags('blob_id')
     self.assertEquals([], flags)
 def test_get_flags_raises_if_no_blob_found(self):
     manager = BlobManager('', self.uri, self.secret,
                           self.secret, uuid4().hex)
     with pytest.raises(SoledadError):
         yield manager.get_flags('missing_id')