def setUp(self):
self.output_file = tempfile.mkstemp(prefix="final_screenshot",
suffix=".png")[1]
os.unlink(self.output_file)
self.tmpdir = None
self.scripts = mockprocess.MockProc()
self.oldstdout = sys.stdout
def test_get_pdc_filelist(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
self.scripts.append("dsmc",
returncode=0,
script="""#!/bin/bash
cat tests/resources/dsmc_output/dsmc_pdc_filelist.txt
""")
with self.scripts:
archive_path = "/data/mm-xart002/runfolders/johanhe_test_0809_001-AG2UJ_archive"
filelist = helper.get_pdc_filelist(
archive_path,
"e374bd6b-ab36-4f41-94d3-f4eaea9f30d4",
dsmc_log_dir="")
with open("tests/resources/dsmc_output/dsmc_pdc_converted_filelist.txt"
) as f:
nr_of_files = 0
for line in f:
size, name = line.split()
self.assertEqual(int(filelist[name]), int(size))
nr_of_files += 1
self.assertEqual(len(filelist.keys()), nr_of_files)
def test_get_pdc_descr_failing_proc(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
self.scripts.append("dsmc", returncode=10)
with self.scripts:
archive_path = "/foo"
descr = helper.get_pdc_descr(archive_path, dsmc_log_dir="")
def test_get_pdc_filelist_failing_proc(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
self.scripts.append("dsmc", returncode=13)
with self.scripts:
archive_path = "foo"
filelist = helper.get_pdc_filelist(archive_path, "foo-bar", dsmc_log_dir="")
def test_get_pdc_filelist_no_result(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
self.scripts.append("dsmc", returncode=0, script="""#!/bin/bash
echo uggla
""")
with self.scripts:
archive_path = "foo"
filelist = helper.get_pdc_filelist(archive_path, "foo-bar", dsmc_log_dir="")
def test_get_pdc_descr_no_results(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
self.scripts.append("dsmc", returncode=0, script="""#!/bin/bash
echo apa
""")
with self.scripts:
archive_path = "foobar"
descr = helper.get_pdc_descr(archive_path, dsmc_log_dir="")
def test_get_pdc_descr(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
# FIXME: this dsmc-output can not be removed, or the test will fail
self.scripts.append("dsmc", returncode=0,
script="""#!/bin/bash
cat tests/resources/dsmc_output/dsmc_descr.txt
""")
with self.scripts:
archive_path = "/data/mm-xart002/runfolders/johanhe_test_0809_001-AG2UJ_archive"
descr = helper.get_pdc_descr(archive_path, dsmc_log_dir="")
self.assertEqual(descr, "e374bd6b-ab36-4f41-94d3-f4eaea9f30d4")
def test_get_pdc_descr(self):
self.scripts = mockprocess.MockProc()
helper = ReuploadHelper()
self.scripts.append("dsmc",
returncode=0,
script="""#!/bin/bash
cat tests/resources/dsmc_output/dsmc_descr.txt
""")
with self.scripts:
descr = helper.get_pdc_descr(
"/data/mm-xart002/runfolders/johanhe_test_150821_M00485_0220_000000000-AG2UJ_archive"
)
self.assertEqual(descr, "e374bd6b-ab36-4f41-94d3-f4eaea9f30d4")
def setUp(self):
self.scripts_dir = mkdtemp()
self.output_filename = mkdtemp() + "/output.txt"
# Re-create the file for each run
open(self.output_filename, 'w+').close()
self.scripts = mockprocess.MockProc(self.scripts_dir)