Split Testeaux into multiple files. Implement stub stresses and troubles as plugins.

Author: kvap

testeaux/Combineaux.pm

@@ -0,0 +1,31 @@
+package Combineaux;
+
+use Troubleaux;
+use Stresseaux;
+use Starteaux;
+
+sub combine
+{
+	my ($stresses, $troubles) = @_;
+
+	my $cluster = Starteaux->deploy('lxc');
+
+	foreach my $stress (@$stresses)
+	{
+		foreach my $trouble (@$troubles)
+		{
+			print("--- $stress vs. $trouble ---\n");
+			my $id = "$stress+$trouble";
+			Stresseaux::start($id, $stress, $cluster) || die "stress wouldn't start";
+			sleep(1); # FIXME: will this work?
+			Troubleaux::cause($cluster, $trouble);
+			sleep(1); # FIXME: will this work?
+			Stresseaux::stop($id) || die "stress wouldn't stop";
+			Troubleaux::fix($cluster);
+		}
+	}
+
+	$cluster->destroy();
+}
+
+1;

testeaux/Starteaux.pm

@@ -0,0 +1,47 @@
+package Starteaux;
+
+sub deploy
+{
+	my ($class, $driver, @args) = @_;
+	my $self = {};
+	print("deploy cluster using driver $driver\n");
+	# fixme: implement
+	return bless $self, 'Starteaux';
+}
+
+sub up
+{
+	my ($self, $id) = @_;
+	print("up node $id\n");
+	# FIXME: implement
+}
+
+sub down
+{
+	my ($self, $id) = @_;
+	print("down node $id\n");
+	# FIXME: implement
+}
+
+sub drop
+{
+	my ($self, $src, $dst, $ratio) = @_;
+	print("drop $ratio packets from $src to $dst\n");
+	# FIXME: implement
+}
+
+sub delay
+{
+	my ($self, $src, $dst, $msec) = @_;
+	print("delay packets from $src to $dst by $msec msec\n");
+	# FIXME: implement
+}
+
+sub destroy
+{
+	my ($self) = @_;
+	print("destroy cluster $cluster\n");
+	# FIXME: implement
+}
+
+1;

testeaux/Stresseaux.pm

@@ -4,27 +4,23 @@ my %running = ();
 
 sub start
 {
-	my ($id, $workload, $cluster) = @_;
-	print("start stress $id: workload $workload, cluster $cluster\n");
+	my ($id, $stressname, $cluster, @args) = @_;
 
 	if (exists $running{$id})
 	{
-		print("cannot start stress $id: that id has already been taken\n");
+		print("cannot start stress $stressname as $id: that id has already been taken\n");
 		return 0;
 	}
 
-	# FIXME: implement 'stress'/'workload' objects
-	$running{$id} = {hello => 'world'};
-
-	# FIXME: implement
+	require "stress/$stressname.pm";
+	$running{$id} = "stress::$stressname"->start($id, $cluster, @args);
 
 	return 1;
 }
 
 sub stop
 {
 	my $id = shift;
-	print("stop stress $id\n");
 
 	my $stress = delete $running{$id};
 
@@ -34,7 +30,7 @@ sub stop
 		return 0;
 	}
 
-	# FIXME: implement
+	$stress->stop();
 
 	return 1;
 }

testeaux/Testeaux.pm

@@ -0,0 +1,27 @@
+my @caused = ();
+
+package Troubleaux
+{
+	sub cause
+	{
+		my $cluster = shift;
+		my $troublename = shift;
+		my @args = @_;
+
+		require "trouble/$troublename.pm";
+		push @caused, "trouble::$troublename"->cause(@args);
+	}
+
+	sub fix
+	{
+		my ($cluster) = @_;
+
+		while (@caused)
+		{
+			my $trouble = pop @caused;
+			$trouble->fix();
+		}
+	}
+}
+
+1;

testeaux/Troubleaux.pm

@@ -1,14 +1,35 @@
 #!/usr/bin/perl
 
-use Testeaux;
-
-my @workloads = qw(
-	bank-transfers
-	pgbench-default
-);
-my @troubles = qw(
-	split-brain
-	time-shift
-);
-
-Combineaux::combine(\@workloads, \@troubles);
+use Combineaux;
+
+sub list_modules
+{
+	my $dir = shift;
+	my @modules = ();
+
+	opendir DIR, $dir or die "Cannot open directory: $!";
+
+	while (my $file = readdir(DIR)) {
+		# Use a regular expression to ignore files beginning with a period
+		next unless ($file =~ m/\.pm$/);
+		push @modules, substr($file, 0, -3);
+	}
+
+	closedir DIR;
+
+	return @modules;
+}
+
+#my @workloads = qw(
+#	banktransfer
+#	pgbench
+#);
+#my @troubles = qw(
+#	splitbrain
+#	timeshift
+#);
+
+my @stresses = list_modules('stress');
+my @troubles = list_modules('trouble');
+
+Combineaux::combine(\@stresses, \@troubles);

testeaux/eaux.pl

@@ -0,0 +1,17 @@
+package stress::banktransfer;
+
+sub start
+{
+	my ($class, @args) = @_;
+	my $self = {};
+	print("start $class with args " . join(',', @args) . "\n");
+	return bless $self, $class;
+}
+
+sub stop
+{
+	my $self = shift;
+	print("stop $self\n");
+}
+
+1;

testeaux/stress/banktransfer.pm

@@ -0,0 +1,17 @@
+package stress::pgbench;
+
+sub start
+{
+	my ($class, @args) = @_;
+	my $self = {};
+	print("start $class with args " . join(',', @args) . "\n");
+	return bless $self, $class;
+}
+
+sub stop
+{
+	my $self = shift;
+	print("stop $self\n");
+}
+
+1;

testeaux/stress/pgbench.pm

@@ -0,0 +1,17 @@
+package trouble::splitbrain;
+
+sub cause
+{
+	my ($class, @args) = @_;
+	my $self = {};
+	print("cause $class with args " . join(',', @args) . "\n");
+	return bless $self, $class;
+}
+
+sub fix
+{
+	my $self = shift;
+	print("fix $self\n");
+}
+
+1;

testeaux/trouble/splitbrain.pm

@@ -0,0 +1,17 @@
+package trouble::timeshift;
+
+sub cause
+{
+	my $class = shift;
+	my $self = {};
+	print("cause $class with args " . join(',', @args) . "\n");
+	return bless $self, $class;
+}
+
+sub fix
+{
+	my $self = shift;
+	print("fix $self\n");
+}
+
+1;