Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
More deeply test supplies, with both schedulers
  • Loading branch information
@lizmat
lizmat committed Apr 16, 2014
1 parent 0f6ab3d commit 640c38f
Showing 1 changed file with 116 additions and 99 deletions.
215 changes: 116 additions & 99 deletions S17-concurrency/supply.t
View file
@@ -1,118 +1,135 @@
use v6;
use Test;

plan 16;
plan 34;

{
my $p = Supply.new;

my @vals;
my $saw_done;
my $tap = $p.tap( -> $val { @vals.push($val) },
done => { $saw_done = True });

$p.more(1);
is ~@vals, "1", "Tap got initial value";
nok $saw_done, "No done yet";

$p.more(2);
$p.more(3);
$p.done;
is ~@vals, "1 2 3", "Tap saw all values";
ok $saw_done, "Saw done";
}
for (ThreadPoolScheduler, CurrentThreadScheduler) {
$*SCHEDULER = .new;
isa_ok $*SCHEDULER, $_, "***** scheduling with {$_.gist}";

{
my $p = Supply.new;

my @tap1_vals;
my @tap2_vals;
my $tap1 = $p.tap(-> $val { @tap1_vals.push($val) });
{
my $p = Supply.new;

$p.more(1);
is ~@tap1_vals, "1", "First tap got initial value";

my $tap2 = $p.tap(-> $val { @tap2_vals.push($val) });
$p.more(2);
is ~@tap1_vals, "1 2", "First tap has both values";
is ~@tap2_vals, "2", "Second tap missed first value";

$tap1.close;
$p.more(3);
is ~@tap1_vals, "1 2", "First tap closed, missed third value";
is ~@tap2_vals, "2 3", "Second tap gets third value";
}
my @vals;
my $saw_done;
my $tap = $p.tap( -> $val { @vals.push($val) },
done => { $saw_done = True });

{
my $p = Supply.for(1..10, :scheduler(CurrentThreadScheduler));

my @a1;
my $tap1 = $p.tap( -> $val { @a1.push($val) },
done => { @a1.push("end") });
is ~@a1, "1 2 3 4 5 6 7 8 9 10 end", "Synchronous publish worked";

my @a2;
my $tap2 = $p.tap( -> $val { @a2.push($val) },
done => { @a2.push("end") });
is ~@a2, "1 2 3 4 5 6 7 8 9 10 end", "Second tap also gets all values";
}
$p.more(1);
is ~@vals, "1", "Tap got initial value";
nok $saw_done, "No done yet";

#?rakudo.jvm skip "hangs"
{
my $p = Supply.for(2..6);
my @a;
for $p.list {
@a.push($_);
$p.more(2);
$p.more(3);
$p.done;
is ~@vals, "1 2 3", "Tap saw all values";
ok $saw_done, "Saw done";
}
is ~@a, "2 3 4 5 6", "Supply.for and .list work";
}

{
my $p = Supply.new;

my @tap1_vals;
my @tap2_vals;
my $tap1 = $p.tap(-> $val { @tap1_vals.push($val) });

$p.more(1);
is ~@tap1_vals, "1", "First tap got initial value";

my $tap2 = $p.tap(-> $val { @tap2_vals.push($val) });
$p.more(2);
is ~@tap1_vals, "1 2", "First tap has both values";
is ~@tap2_vals, "2", "Second tap missed first value";

$tap1.close;
$p.more(3);
is ~@tap1_vals, "1 2", "First tap closed, missed third value";
is ~@tap2_vals, "2 3", "Second tap gets third value";
}

{
my $p = Supply.for(1..10);

my @a1;
my $done1 = False;
my $tap1 = $p.tap( -> $val { @a1.push($val) },
done => { @a1.push("end"); $done1 = True });
sleep .1 until $done1;
is ~@a1, "1 2 3 4 5 6 7 8 9 10 end", "Synchronous publish worked";

my @a2;
my $done2 = False;
my $tap2 = $p.tap( -> $val { @a2.push($val) },
done => { @a2.push("end"); $done2 = True });
sleep .1 until $done2;
is ~@a2, "1 2 3 4 5 6 7 8 9 10 end", "Second tap also gets all values";
}

#?rakudo.jvm skip "hangs"
{
my $p1 = Supply.new;
my $p2 = Supply.new;

my @res;
my $tap = $p1.zip($p2, &infix:<~>).tap({ @res.push($_) });

$p1.more(1);
$p1.more(2);
$p2.more('a');
$p2.more('b');
$p2.more('c');
$p1.done();
$p2.done();

is @res.join(','), '1a,2b', 'zipping taps works';
my $p = Supply.for(2..6);
my @a;
for $p.list {
@a.push($_);
}
is ~@a, "2 3 4 5 6", "Supply.for and .list work";
}

{
my $p1 = Supply.new;
my $p2 = Supply.new;

my @res;
my $tap = $p1.zip($p2, &infix:<~>).tap({ @res.push($_) });

$p1.more(1);
$p1.more(2);
$p2.more('a');
$p2.more('b');
$p2.more('c');
$p1.done();
$p2.done();

is @res.join(','), '1a,2b', 'zipping taps works';
}

#?rakudo.moar skip "hangs"
{
my $p1 = Supply.new;
my $p2 = Supply.new;

my @res;
my $tap = $p1.merge($p2).tap({ @res.push($_) });

$p1.more(1);
$p1.more(2);
$p2.more('a');
$p1.more(3);
$p1.done();
$p2.more('b');
my $done = False;
my $p1 = Supply.new;
my $p2 = Supply.new;

my @res;
my $tap = $p1.merge($p2).tap({ @res.push($_) }, :done( {$done = True}));

$p1.more(1);
$p1.more(2);
$p2.more('a');
$p1.more(3);
$p2.more('b');
$p2.done();

#?rakudo.moar todo "merging apparently does *not* work"
is @res.join(','), '1,2,a,3,b', "merging taps works";
sleep .1 until $done;
is @res.join(','), '1,2,a,3,b', "merging taps works";
}

{
my $p1 = Supply.for(1..10, :scheduler(CurrentThreadScheduler));
my @res;
$p1.grep(* > 5).tap({ @res.push($_) });
is ~@res, '6 7 8 9 10', "grepping taps works";
}
{
my $p1 = Supply.for(1..10);
my @res;
my $done;
$p1.grep(* > 5).tap({ @res.push($_) }, :done( {$done = True} ));

{
my $p1 = Supply.for(1..5, :scheduler(CurrentThreadScheduler));
my @res;
$p1.map(2 * *).tap({ @res.push($_) });
is ~@res, '2 4 6 8 10', "mapping taps works";
sleep .1 until $done;
is ~@res, '6 7 8 9 10', "grepping taps works";
}

{
my $p1 = Supply.for(1..5);
my @res;
my $done;
$p1.map(2 * *).tap({ @res.push($_) }, :done( {$done = True} ));

sleep .1 until $done;
is ~@res, '2 4 6 8 10', "mapping taps works";
}
}

0 comments on commit 640c38f

Please sign in to comment.