def setUp(self): self.wrapper = Wrapper(bash_location='/bin/sh') self.script = TestableScript(script=dedent("""\ #!/bin/sh #PBS -q debug #PBS -l nodes=1:ppn=1 echo "hello world" echo "this is to stderr" >&2 """), status=Status.success, stdout='hello world', stderr='this is to stderr') self.script_fd = NamedTemporaryFile() self.script.write(self.script_fd) self.script_fd.seek(0) self.script_path = self.script_fd.name
class TestWrapper(TestCase): def setUp(self): self.wrapper = Wrapper(bash_location='/bin/sh') self.script = TestableScript( script=dedent("""\ #!/bin/sh #PBS -q debug #PBS -l nodes=1:ppn=1 echo "hello world" echo "this is to stderr" >&2 """), status=Status.success, stdout='hello world', stderr='this is to stderr') self.script_fd = NamedTemporaryFile() self.script.write(self.script_fd) self.script_fd.seek(0) self.script_path = self.script_fd.name def tearDown(self): self.script_fd.close() @property def wrapped(self): return self.wrapper.wrap(self.script_path) def test_wrap_works(self): "Wrapper should work" wrapped = self.wrapped self.assertIsNotNone(wrapped) def test_names(self): w = self.wrapped name = os.path.basename(self.script_path) self.assertEquals(w.entrypoint.name, name) self.assertEquals(w.wrapped.name, 'wrapped-{}'.format(name)) def test_run(self): w = Worker() r = w(self.wrapped) self.assertEquals(r.status, self.script.status) self.assertEquals(r.stdout, self.script.stdout) self.assertEquals(r.stderr, self.script.stderr)
class TestWrapper(TestCase): def setUp(self): self.wrapper = Wrapper(bash_location='/bin/sh') self.script = TestableScript(script=dedent("""\ #!/bin/sh #PBS -q debug #PBS -l nodes=1:ppn=1 echo "hello world" echo "this is to stderr" >&2 """), status=Status.success, stdout='hello world', stderr='this is to stderr') self.script_fd = NamedTemporaryFile() self.script.write(self.script_fd) self.script_fd.seek(0) self.script_path = self.script_fd.name def tearDown(self): self.script_fd.close() @property def wrapped(self): return self.wrapper.wrap(self.script_path) def test_wrap_works(self): "Wrapper should work" wrapped = self.wrapped self.assertIsNotNone(wrapped) def test_names(self): w = self.wrapped name = os.path.basename(self.script_path) self.assertEquals(w.entrypoint.name, name) self.assertEquals(w.wrapped.name, 'wrapped-{}'.format(name)) def test_run(self): w = Worker() r = w(self.wrapped) self.assertEquals(r.status, self.script.status) self.assertEquals(r.stdout, self.script.stdout) self.assertEquals(r.stderr, self.script.stderr)
def setUp(self): self.wrapper = Wrapper(bash_location='/bin/sh') self.script = TestableScript( script=dedent("""\ #!/bin/sh #PBS -q debug #PBS -l nodes=1:ppn=1 echo "hello world" echo "this is to stderr" >&2 """), status=Status.success, stdout='hello world', stderr='this is to stderr') self.script_fd = NamedTemporaryFile() self.script.write(self.script_fd) self.script_fd.seek(0) self.script_path = self.script_fd.name