def test_local_upload(self): """Tests whether uploading to a local file works.""" dst = tempfile.NamedTemporaryFile() dstUri = "file://"+dst.name stager = DataStager(self.source_path, dstUri) stager.perform(asynchronous=False) tmpdata = dst.read() self.assertEqual(self.data, tmpdata)
def test_local_retrieve(self):
"""Tests whether retrieving a local file works."""
dst_fd, dst_path = tempfile.mkstemp(text=True)
stager = DataStager("file://" + self.source_path, dst_path)
stager.perform(asynchronous=False)
tmpdata = open(dst_path).read()
os.unlink(dst_path)
os.close(dst_fd)
self.assertEqual(self.data, tmpdata)
def __retrieveFile(self, uri, dst):
from xbe.util.staging import DataStager
ds = DataStager(uri, dst)
return ds.perform()
def test_remote_retrieve(self): remoteSource = "http://www.heise.de/index.html" dst = tempfile.NamedTemporaryFile() stager = DataStager(remoteSource, dst.name) stager.perform(asynchronous=False)
