X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=test%2Fresources%2Flinux%2Fnode.py;h=02a2733137c2657e4712c2377ac236566a1ead69;hb=342187d6e7ce34d4fdf403952637659b93aa4bee;hp=829412ae69d23d0b58cf2852dce724ffd77f1957;hpb=62a8f5c90afe9033f532206c811cff8ea76b2c09;p=nepi.git diff --git a/test/resources/linux/node.py b/test/resources/linux/node.py index 829412ae..02a27331 100755 --- a/test/resources/linux/node.py +++ b/test/resources/linux/node.py @@ -4,9 +4,8 @@ # Copyright (C) 2013 INRIA # # This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU General Public License as published by -# the Free Software Foundation, either version 3 of the License, or -# (at your option) any later version. +# it under the terms of the GNU General Public License version 2 as +# published by the Free Software Foundation; # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -19,11 +18,12 @@ # Author: Alina Quereilhac -from nepi.resources.linux.node import LinuxNode -from nepi.util.sshfuncs import RUNNING, FINISHED +from nepi.resources.linux.node import LinuxNode, ExitCode +from nepi.util.sshfuncs import ProcStatus from test_utils import skipIfNotAlive, skipInteractive, create_node +import shutil import os import time import tempfile @@ -31,27 +31,13 @@ import unittest class LinuxNodeTestCase(unittest.TestCase): def setUp(self): - self.fedora_host = 'nepi2.pl.sophia.inria.fr' - self.fedora_user = 'inria_nepi' + self.fedora_host = "fedora.pl.sophia.inria.fr" + self.fedora_user = "inria_nepi" - self.ubuntu_host = 'roseval.pl.sophia.inria.fr' - self.ubuntu_user = 'alina' + self.ubuntu_host = "ubuntu.pl.sophia.inria.fr" + self.ubuntu_user = "inria_nepi" - self.target = 'nepi5.pl.sophia.inria.fr' - - @skipIfNotAlive - def t_xterm(self, host, user): - node, ec = create_node(host, user) - - node.install_packages('xterm') - - (out, err), proc = node.execute('xterm', forward_x11 = True) - - self.assertEquals(out, "") - - (out, err), proc = node.remove_packages('xterm') - - self.assertEquals(out, "") + self.target = "nepi.pl.sophia.inria.fr" @skipIfNotAlive def t_execute(self, host, user): @@ -69,19 +55,20 @@ class LinuxNodeTestCase(unittest.TestCase): def t_run(self, host, user): node, ec = create_node(host, user) + node.find_home() app_home = os.path.join(node.exp_home, "my-app") node.mkdir(app_home, clean = True) command = "ping %s" % self.target node.run(command, app_home) - pid, ppid = node.checkpid(app_home) + pid, ppid = node.getpid(app_home) status = node.status(pid, ppid) - self.assertTrue(status, RUNNING) + self.assertTrue(status, ProcStatus.RUNNING) node.kill(pid, ppid) status = node.status(pid, ppid) - self.assertTrue(status, FINISHED) + self.assertTrue(status, ProcStatus.FINISHED) (out, err), proc = node.check_output(app_home, "stdout") @@ -91,22 +78,163 @@ class LinuxNodeTestCase(unittest.TestCase): node.rmdir(app_home) + @skipIfNotAlive + def t_exitcode_ok(self, host, user): + command = "echo 'OK!'" + + node, ec = create_node(host, user) + + node.find_home() + app_home = os.path.join(node.exp_home, "my-app") + node.mkdir(app_home, clean = True) + + (out, err), proc = node.run_and_wait(command, app_home, + shfile = "cmd.sh", + pidfile = "pid", + ecodefile = "exitcode", + stdout = "stdout", + stderr = "stderr", + raise_on_error = True) + + # get the pid of the process + ecode = node.exitcode(app_home) + self.assertEquals(ecode, ExitCode.OK) + + @skipIfNotAlive + def t_exitcode_kill(self, host, user): + node, ec = create_node(host, user) + + node.find_home() + app_home = os.path.join(node.exp_home, "my-app") + node.mkdir(app_home, clean = True) + + # Upload command that will not finish + command = "ping localhost" + shfile = os.path.join(app_home, "cmd.sh") + (out, err), proc = node.upload_command(command, + shfile = shfile, + ecodefile = "exitcode") + + (out, err), proc = node.run(command, app_home, + pidfile = "pidfile", + stdout = "stdout", + stderr = "stderr") + + # Just wait to make sure the ping started + time.sleep(5) + + # The process is still running, so no retfile has been created yet + ecode = node.exitcode(app_home) + self.assertEquals(ecode, ExitCode.FILENOTFOUND) + + (out, err), proc = node.check_errors(app_home) + self.assertEquals(err, "") + + # Now kill the app + pid, ppid = node.getpid(app_home) + node.kill(pid, ppid) + + (out, err), proc = node.check_errors(app_home) + self.assertEquals(err, "") + + @skipIfNotAlive + def t_exitcode_error(self, host, user): + # Try to execute a command that doesn't exist + command = "unexistent-command" + + node, ec = create_node(host, user) + + node.find_home() + app_home = os.path.join(node.exp_home, "my-app") + node.mkdir(app_home, clean = True) + + (out, err), proc = node.run_and_wait(command, app_home, + shfile = "cmd.sh", + pidfile = "pid", + ecodefile = "exitcode", + stdout = "stdout", + stderr = "stderr", + raise_on_error = False) + + # get the pid of the process + ecode = node.exitcode(app_home) + + # bash erro 127 - command not found + self.assertEquals(ecode, 127) + + (out, err), proc = node.check_errors(app_home) + + self.assertTrue(err.find("cmd.sh: line 1: unexistent-command: command not found") > -1) + @skipIfNotAlive def t_install(self, host, user): node, ec = create_node(host, user) - (out, err), proc = node.install_packages('gcc') - self.assertEquals(out, "") + node.find_home() + (out, err), proc = node.mkdir(node.node_home, clean = True) + self.assertEquals(err, "") + + (out, err), proc = node.install_packages("gcc", node.node_home) + self.assertEquals(err, "") + + (out, err), proc = node.remove_packages("gcc", node.node_home) + self.assertEquals(err, "") + + (out, err), proc = node.rmdir(node.exp_home) + self.assertEquals(err, "") + + @skipIfNotAlive + def t_clean(self, host, user): + node, ec = create_node(host, user) + + node.find_home() + node.mkdir(node.lib_dir) + node.mkdir(node.node_home) + + command1 = " [ -d %s ] && echo 'Found'" % node.lib_dir + (out, err), proc = node.execute(command1) + + self.assertEquals(out.strip(), "Found") + + command2 = " [ -d %s ] && echo 'Found'" % node.node_home + (out, err), proc = node.execute(command2) + + self.assertEquals(out.strip(), "Found") + + node.clean_experiment() + + (out, err), proc = node.execute(command2) + + self.assertEquals(out.strip(), "") + + node.clean_home() + + (out, err), proc = node.execute(command1) + + self.assertEquals(out.strip(), "") + + @skipIfNotAlive + def t_xterm(self, host, user): + node, ec = create_node(host, user) - (out, err), proc = node.remove_packages('gcc') + node.find_home() + (out, err), proc = node.mkdir(node.node_home, clean = True) + self.assertEquals(err, "") - self.assertEquals(out, "") + node.install_packages("xterm", node.node_home) + self.assertEquals(err, "") + (out, err), proc = node.execute("xterm", forward_x11 = True) + self.assertEquals(err, "") + + (out, err), proc = node.remove_packages("xterm", node.node_home) + self.assertEquals(err, "") @skipIfNotAlive def t_compile(self, host, user): node, ec = create_node(host, user) + node.find_home() app_home = os.path.join(node.exp_home, "my-app") node.mkdir(app_home, clean = True) @@ -124,7 +252,7 @@ main (void) node.upload(prog, dst, text = True) # install gcc - node.install_packages('gcc') + node.install_packages('gcc', app_home) # compile the program using gcc command = "cd %s; gcc -Wall hello.c -o hello" % app_home @@ -143,12 +271,12 @@ main (void) # retrieve the output file src = os.path.join(app_home, "hello.out") - f = tempfile.NamedTemporaryFile(delete=False) + f = tempfile.NamedTemporaryFile(delete = False) dst = f.name node.download(src, dst) f.close() - node.remove_packages('gcc') + node.remove_packages("gcc", app_home) node.rmdir(app_home) f = open(dst, "r") @@ -157,6 +285,55 @@ main (void) self.assertEquals(out, "Hello, world!\n") + @skipIfNotAlive + def t_copy_files(self, host, user): + node, ec = create_node(host, user) + + node.find_home() + app_home = os.path.join(node.exp_home, "my-app") + node.mkdir(app_home, clean = True) + + # create some temp files and directories to copy + dirpath = tempfile.mkdtemp() + f = tempfile.NamedTemporaryFile(dir=dirpath, delete=False) + f.close() + + f1 = tempfile.NamedTemporaryFile(delete=False) + f1.close() + f1.name + + source = [dirpath, f1.name] + destdir = "test" + node.mkdir(destdir, clean = True) + dest = "%s@%s:test" % (user, host) + node.copy(source, dest) + + command = "ls %s" % destdir + + (out, err), proc = node.execute(command) + + os.remove(f1.name) + shutil.rmtree(dirpath) + + self.assertTrue(out.find(os.path.basename(dirpath)) > -1) + self.assertTrue(out.find(os.path.basename(f1.name)) > -1) + + f2 = tempfile.NamedTemporaryFile(delete=False) + f2.close() + f2.name + + node.mkdir(destdir, clean = True) + dest = "%s@%s:test" % (user, host) + node.copy(f2.name, dest) + + command = "ls %s" % destdir + + (out, err), proc = node.execute(command) + + os.remove(f2.name) + + self.assertTrue(out.find(os.path.basename(f2.name)) > -1) + def test_execute_fedora(self): self.t_execute(self.fedora_host, self.fedora_user) @@ -180,12 +357,41 @@ main (void) def test_compile_ubuntu(self): self.t_compile(self.ubuntu_host, self.ubuntu_user) - + + def test_exitcode_ok_fedora(self): + self.t_exitcode_ok(self.fedora_host, self.fedora_user) + + def test_exitcode_ok_ubuntu(self): + self.t_exitcode_ok(self.ubuntu_host, self.ubuntu_user) + + def test_exitcode_kill_fedora(self): + self.t_exitcode_kill(self.fedora_host, self.fedora_user) + + def test_exitcode_kill_ubuntu(self): + self.t_exitcode_kill(self.ubuntu_host, self.ubuntu_user) + + def test_exitcode_error_fedora(self): + self.t_exitcode_error(self.fedora_host, self.fedora_user) + + def test_exitcode_error_ubuntu(self): + self.t_exitcode_error(self.ubuntu_host, self.ubuntu_user) + + def test_clean_fedora(self): + self.t_clean(self.fedora_host, self.fedora_user) + + def test_clean_ubuntu(self): + self.t_clean(self.ubuntu_host, self.ubuntu_user) + @skipInteractive def test_xterm_ubuntu(self): """ Interactive test. Should not run automatically """ self.t_xterm(self.ubuntu_host, self.ubuntu_user) + def test_copy_files_fedora(self): + self.t_copy_files(self.fedora_host, self.fedora_user) + + def test_copy_files_ubuntu(self): + self.t_copy_files(self.ubuntu_host, self.ubuntu_user) if __name__ == '__main__': unittest.main()