xcat-core/xCAT-test/autotest/testcase/genesis/genesistest.pl

#!/usr/bin/env perl
# IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html

use strict;
use warnings;
use Getopt::Long;
use Data::Dumper;
use Term::ANSIColor;
use Time::Local;
use File::Basename;
use File::Path;
use File::Copy;
BEGIN
{
    $::XCATROOT = $ENV{'XCATROOT'} ? $ENV{'XCATROOT'} : -d '/opt/xcat' ? '/opt/xcat' : '/usr';
}
use lib "$::XCATROOT/lib/perl";
my $rungenesiscmd = 0;
my $rungenesisimg = 0;
my $shellmode = 0;
my $cmdtest = "/tmp/cmdtest";
my $timesleep = 0;
my $noderange ;
my $clearenv = 0;
my $arch ;
my $imgip ;
my $runimgtest = "/tmp/imgtest";
my $testresult = "/tmp/testresult";
my $genesisdir = "/opt/xcat/share/xcat/netboot/genesis";
my $genesisfiledir ;

if (
    !GetOptions("h|?"  => \$::HELP,
                "d"=>\$rungenesiscmd,
                "g"=>\$rungenesisimg,
                "t"=>\$timesleep,
                "c"=>\$clearenv,
                "n=s"=>\$::NODE,
                "i=s"=>\$::IMGIP,
                "r=s"=>\$::ARCH)
)
{
    &usage;
    exit 1;
}
sub usage
{
    print "Usage:run for genesis cases.\n";
    print "  genesistest.pl [-?|-h]\n";
    print "  genesistest.pl [-d] [-n node] [-r arch]  Test runcmd for genesis \n";
    print "  genesistest.pl [-g] [-n ndoe] [-i imgip] Test runimg for genesis\n";
    print "  genesistest.pl [-t] [-n node] Sleep for genesis test\n";
    print "  genesistest.pl [-c] [-n node][-r arch] Clear environment for genesis test\n";
    print "\n";
    return;
}
sub runcmd
{
    my ($cmd) = @_;
    my $rc = 0;
    $::RUNCMD_RC = 0;
    my $outref = [];
    @$outref = `$cmd 2>&1`;
    if ($?)
    {
        $rc = $? ;
        $rc = $rc >> 8;
        $::RUNCMD_RC = $rc;
    }
    chomp(@$outref);
    return @$outref;

}
sub rungenesiscmd
{
    open(TESTCMD, ">$cmdtest")
    or die "Can't open testscripts for writing: $!";
    print TESTCMD join("\n", "#!/bin/bash"), "\n";
    print TESTCMD join("\n", "#This is test for genesis scripts"), "\n";
    print TESTCMD join("\n", "echo \"testcmd\" >> $testresult"), "\n";
    close(TESTCMD);
    if ($arch =~ /ppc64/)
     {
         $arch = "ppc64";
     }
     $genesisfiledir="$genesisdir/$arch/fs/bin";
     copy("$cmdtest" ,"$genesisfiledir");
     chmod 0755, "$genesisfiledir/cmdtest";
     `mknb $arch`;
     print "mknb $arch\n";
}
sub rungenesisimg
{
    mkdir("/install/my_image");
    open(TESTIMG, ">$runimgtest")
    or die "Can't open testscripts for writing: $!";
    print TESTIMG join("\n", "#!/bin/bash"), "\n";
    print TESTIMG join("\n", "#This is test for genesis scripts"), "\n";
    print TESTIMG join("\n", "echo \"testimg\" >> $testresult"), "\n";
    close(TESTIMG);
    copy("$runimgtest" ,"/install/my_image/runme.sh" ) or die "Copy failed: $!";
    chmod 0755,"/install/my_image/runme.sh"; 
    `cd /install/my_image ;tar -zcvf my_image.tgz  .`;
    `nodeset $noderange "runimage=http://$imgip/install/my_image/my_image.tgz",shell`;
}
sub timesleep
{
    my @output = runcmd("ping $noderange -c 10");
    my $value = 0;
    print "output is $value ,@output\n";
    if ($::RUNCMD_RC){
        foreach $value (1 .. 60) {
        @output = runcmd("ping $noderange -c 10");
        last  if ($::RUNCMD_RC == 0);
        }
     }
    my @output1 = runcmd("xdsh $noderange date");
    if ($::RUNCMD_RC){
        foreach $value (1 .. 60) {
        @output1 = runcmd("xdsh $noderange -t 1 date");
        print "sleep $value\n";
        last if ($::RUNCMD_RC == 0);
        }
    }
    if ($::RUNCMD_RC == 0){
        print "test ok\n";
     }
}
sub clearenv
{
   if (-f "/tmp/imgtest"){
       unlink("/install/my_image/runme.sh");
       unlink("/install/my_image/my_image.tgz");
       unlink("$runimgtest");
       rmdir("/install/my_image");
       print "img del ok\n";
   }
   if (-f "/tmp/cmdtest"){
      if ($arch =~ /ppc64/)
      {
         $arch = "ppc64";
      }
      
      $genesisfiledir="$genesisdir/$arch/fs/bin";
      my $genesisfile = "$genesisfiledir/cmdtest";
      print "genesis file is $genesisfile\n";
      unlink("$genesisfile");
      unlink("$cmdtest");
     `mknb $arch`;
     print "mknb $arch\n";
   }
   `nodeset $noderange boot`; 
}
if ($::NODE)
{
    $noderange = $::NODE; 
}
if($::ARCH)
{
   $arch = $::ARCH;
}
if($::IMGIP)
{
   $imgip = $::IMGIP;
}
if ($::HELP) {
usage;
}
if ($rungenesiscmd)
{
    &rungenesiscmd;
}
if ($timesleep)
{
    &timesleep;
}
if($rungenesisimg)
   {
   &rungenesisimg;
   }
if($clearenv)
{
   &clearenv;
}