#! /usr/bin/env perl
##**************************************************************
##
## Copyright (C) 1990-2007, Condor Team, Computer Sciences Department,
## University of Wisconsin-Madison, WI.
## 
## Licensed under the Apache License, Version 2.0 (the "License"); you
## may not use this file except in compliance with the License.  You may
## obtain a copy of the License at
## 
##    http://www.apache.org/licenses/LICENSE-2.0
## 
## Unless required by applicable law or agreed to in writing, software
## distributed under the License is distributed on an "AS IS" BASIS,
## WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
## See the License for the specific language governing permissions and
## limitations under the License.
##
##**************************************************************

use CondorTest;
use CondorUtils;
use Check::SimpleJob;

my $testname = "job_core_perhold_sched";

#testreq: personal
my $config = <<CONDOR_TESTREQ_CONFIG;
    DAEMON_LIST = MASTER,SCHEDD,COLLECTOR,NEGOTIATOR,STARTD
	PERIODIC_EXPR_interval = 30
    NUM_CPUS = 2
    NEGOTIATOR_INTERVAL = 5
CONDOR_TESTREQ_CONFIG
#endtestreq

my $variation = "perholdfalse";

# truly const variables in perl
sub IDLE{1};
sub HELD{5};
sub RUNNING{2};

my %info;
my $cluster;


$executed = sub
{
	%info = @_;
	$cluster = $info{"cluster"};

	if($variation eq "perholdfalse") {
		CondorTest::debug("Good. for periodic_hold cluster $cluster must run first\n",1);
	} else {
		CondorTest::debug("Periodic Hold should see an execute, followed by a hold and then we remove the job\n",1);
	}
};

$success = sub
{
	my %info = @_;
	my $cluster = $info{"cluster"};

	CondorTest::debug("Good, job should complete trivially\n",1);
};

my $killedchosen = 0;

$abnormal = sub {
	my %info = @_;

	die "Want to see only execute, hold and abort events for periodic hold test\n";
};

$aborted = sub {
	my %info = @_;
	my $done;
	CondorTest::debug("Abort event expected from periodic remove after hold event seen\n",1);
};

$held = sub {
	my %info = @_;
	my $cluster = $info{"cluster"};

	CondorTest::debug("Held event expected, removing job.....\n",1);
	my @adarray;
	my $status = 1;
	my $cmd = "condor_rm $cluster";
	$status = CondorTest::runCondorTool($cmd,\@adarray,2,{emit_output=>0});
	if(!$status)
	{
		CondorTest::debug("Test failure due to Condor Tool Failure<$cmd>\n",1);
		exit(1)
	}
};

my $on_evictedwithoutcheckpoint = sub {
	CondorTest::debug("Evicted Without Checkpoint from removing jobs.\n",1);
};


my $falsepolicy = '
periodic_hold = (time() - QDate) < 0
';

my $truepolicy = '
periodic_hold = JobStatus == 2
';

if(is_windows()) {
	SimpleJob::RunCheck(
		runthis=>"sleep.exe",
		universe=>"scheduler",
	    append_submit_commands=>"$falsepolicy",
	    duration=>"20s",
	    on_execute=>$executed,
	    on_success=>$success,
	);

	$variation = "perholdtrue";

	SimpleJob::RunCheck(
		runthis=>"sleep.exe",
		universe=>"scheduler",
	    duration=>"40s",
	    append_submit_commands=>$truepolicy,
		on_evictedwithoutcheckpoint=>$on_evictedwithoutcheckpoint,
	    on_exitedabnormal=>$abnormal,
	    on_execute=>$executed,
	    on_hold=>$held,
	    on_abort=>$aborted
	);
} else {
	SimpleJob::RunCheck(
		universe=>"scheduler",
	    append_submit_commands=>"$falsepolicy",
	    duration=>20,
	    on_execute=>$executed,
	    on_success=>$success,
	);

	$variation = "perholdtrue";

	SimpleJob::RunCheck(
		universe=>"scheduler",
	    duration=>40,
	    append_submit_commands=>$truepolicy,
		on_evictedwithoutcheckpoint=>$on_evictedwithoutcheckpoint,
	    on_exitedabnormal=>$abnormal,
	    on_execute=>$executed,
	    on_hold=>$held,
	    on_abort=>$aborted
	);
}

CondorTest::EndTest();
exit(0);
