def test_set_uploader_fail(self): u = Uploader() u.init({'test': 'test'}) with pytest.raises(UploaderError): u.set_uploader([]) with pytest.raises(UploaderPluginError): u.set_uploader('bad plugin name')
def test_upload(self, httpbin): class MyUploader(BaseUploader): def init(self, **kwargs): self.file_list = ['test'] def upload_file(self, filename): with open('testfile.txt', 'wb') as f: f.write('this is some text') with open('testfile.txt', 'rb') as f: data = f.read() files = {'file': data} r = requests.post(httpbin.url + '/post', files=files) assert r.json()['files']['file'] == data u = Uploader() u.init({'test': 'test'}) u.uploader = MyUploader() u.uploader.init(test='test') u.upload()
def test_fail_no_uploader_set_fail(self, httpbin): with pytest.raises(UploaderError): u = Uploader() u.init({}) u.upload()
def test_set_uploader(self): u = Uploader() u.init({'OBJECT_BUCKET': 'vdpro'}) u.set_uploader('s3') assert u.uploader is not None