# Unit test for Task::WithSubtask

package TaskWithSubtaskTest;

use Test::More;

use Task::WithSubtask;

use Task::Testing;

sub start {

print "### Starting TaskWithSubtaskTest\n";

testBasicUsage();

testStop();

testSubtaskFailure();

testMutexChanges();

}

sub testBasicUsage {

print "Testing basic usage...\n";

my $task = new TaskWithSubtaskTest::TestTask();

$task->activate();

$task->iterate();

is($task->getStatus(), Task::RUNNING, "Task is running.");

ok(defined $task->getSubtask(), "Task has subtask.");

$task->iterate();

is($task->getStatus(), Task::DONE, "Task is done.");

}

sub testStop {

print "Testing stop()...\n";

my $task = new TaskWithSubtaskTest::TestTask();

$task->activate();

$task->iterate();

$task->stop();

is($task->getStatus(), Task::STOPPED);

$task = new TaskWithSubtaskTest::TestTask(autostop => 0);

$task->activate();

$task->iterate();

$task->stop();

is($task->getStatus(), Task::RUNNING);

$task->iterate();

is($task->getStatus(), Task::DONE);

}

sub testSubtaskFailure {

print "Testing subtask failure...\n";

my $task = new TaskWithSubtaskTest::TestTask(autofail => 0);

$task->activate();

$task->iterate();

is($task->getStatus(), Task::RUNNING, "Task is still running.");

$task->getSubtask()->markFailed();

$task->iterate();

is($task->getStatus(), Task::DONE, "Task is done.");

ok(!defined $task->getError());

my $task = new TaskWithSubtaskTest::TestTask();

$task->activate();

$task->iterate();

$task->getSubtask()->markFailed();

is($task->getStatus(), Task::RUNNING, "Task is still running.");

$task->iterate();

is($task->getStatus(), Task::DONE, "Task is done.");

ok(defined $task->getError());

}

sub testMutexChanges {

print "Testing dynamic mutex changes...\n";

my $task = new Task::Testing(name => 'A', mutexes => ['3'], manageMutexes => 1);

my $subtask = new Task::Testing(name => 'B', mutexes => ['1', '2']);

$task->activate();

is_deeply($task->getMutexes(), ['3']);

$task->setSubtask($subtask);

is_deeply($task->getMutexes(), ['1', '2']);

$task->iterate();

is_deeply($task->getMutexes(), ['1', '2']);

$subtask->markDone();

$task->iterate();

is_deeply($task->getMutexes(), ['3']);

$subtask = new Task::Testing(name => 'B', mutexes => []);

$task->setSubtask($subtask);

is_deeply($task->getMutexes(), []);

$subtask->setMutexes('hello', 'world');

is_deeply($task->getMutexes(), ['hello', 'world']);

$subtask->markDone();

$task->iterate();

is_deeply($task->getMutexes(), ['3']);

}

package TaskWithSubtaskTest::TestTask;

# A test task which has exactly one subtask. It works as follows:

# - The first iteration will switch context to the subtask.

# - The second iteration will run the subtask, which in turn only

# runs one iteration. The task is then complete.

use base qw(Task::WithSubtask);

sub activate {

my ($self) = @_;

$self->SUPER::activate();

$self->{state} = 'Normal';

}

sub iterate {

my ($self) = @_;

return if (!$self->SUPER::iterate());

if ($self->{state} eq 'Normal') {

$self->setSubtask(new TaskWithSubtaskTest::TestSubtask);

$self->{state} = 'Subtask Done';

} else {

$self->setDone();

}

}

package TaskWithSubtaskTest::TestSubtask;

use base qw(Task);

sub iterate {

my ($self) = @_;

$self->SUPER::iterate();

if ($self->{failed}) {

$self->setError(0, "foo");

} else {

$self->setDone();

}

}

sub markFailed {

my ($self) = @_;

$self->{failed} = 1;

}

1;