use strictures 1;
use Test::More;
use HTML::Zoom::Parser::BuiltIn;
use HTML::Zoom::Producer::BuiltIn;
use HTML::Zoom::SelectorParser;
use HTML::Zoom::FilterBuilder;
use HTML::Zoom::FilterStream;
my $tmpl = <
Bob
Builder
END
sub src_stream { HTML::Zoom::Parser::BuiltIn->new->html_to_stream($tmpl); }
sub html_sink { HTML::Zoom::Producer::BuiltIn->new->html_from_stream($_[0]) }
my $fb = HTML::Zoom::FilterBuilder->new;
my $sp = HTML::Zoom::SelectorParser->new;
sub filter {
my ($stream, $sel, $cb) = @_;
return HTML::Zoom::FilterStream->new({
stream => $stream,
match => $sp->parse_selector($sel),
filter => do { local $_ = $fb; $cb->($fb) }
});
}
sub run_for (&;$) {
my $cb = shift;
(html_sink
(filter
src_stream,
(shift or '.main'),
$cb
)
)
}
my ($expect, @ev);
($expect = $tmpl) =~ s/class="main"/class="foo"/;
is(
run_for { $_->set_attribute( 'class' => 'foo' ) },
$expect,
'set attribute on existing attribute'
);
($expect = $tmpl) =~ s/name="cow" class="main"/name="bar" class="foo"/;
is(
run_for { $_->set_attr({ 'class' => 'foo', 'name' => 'bar'}) },
$expect,
'set attributes using hashref form (shorthand)'
);
($expect = $tmpl) =~ s/class="main"/class="main" foo="bar"/;
is(
run_for { $_->set_attribute( 'foo' => 'bar' ) },
$expect,
'set attribute on non existing attribute'
);
($expect = $tmpl) =~ s/class="main"/class="main foo"/;
is(
run_for { $_->add_to_attribute( 'class' => 'foo' ) },
$expect,
'add attribute on existing attribute'
);
($expect = $tmpl) =~ s/class="main"/class="main foo"/;
is(
run_for { $_->add_class('foo') },
$expect,
'add attribute on existing attribute (shorthand)'
);
($expect = $tmpl) =~ s/class="main"/class="main" id="foo"/;
is(
run_for { $_->set_id('foo') },
$expect,
'set_id (shorthand)'
);
($expect = $tmpl) =~ s/class="main"/class="main" foo="bar"/;
is(
run_for { $_->add_to_attribute( 'foo' => 'bar' ) },
$expect,
'add attribute on non existing attribute'
);
($expect = $tmpl) =~ s/ class="main"//;
is(
run_for { $_->remove_attribute({ name => 'class' }) },
$expect,
'remove attribute on existing attribute'
);
is(
run_for { $_->remove_attribute({ name => 'foo' }) },
$tmpl,
'remove attribute on non existing attribute'
);
($expect = $tmpl) =~ s/class="main"/class=""/;
is(
run_for { $_->remove_from_attribute({ class => 'main' }) },
$expect,
'remove name from attribute'
);
is(
run_for { $_->remove_from_attribute({ madeup => 'main' }) },
$tmpl,
'remove name from non existing attribute (ignored)'
);
($expect = $tmpl) =~ s/class="main"/class=""/;
is(
run_for { $_->remove_class( 'main' ) },
$expect,
'remove_class'
);
($expect = $tmpl) =~ s/ class="main"//;
is(
run_for {
$_->transform_attribute({
name => 'class',
code => sub {
my $a = shift;
return if $a eq 'main';
return $a;
},
})
},
$expect,
'transform_attribute deletes the attr if code returns undef',
);
($expect = $tmpl) =~ s/ class="main"/ class="moan"/;
is(
run_for {
$_->transform_attribute({
name => 'class',
code => sub {
( my $b = shift ) =~ s/main/moan/;
$b
},
})
},
$expect,
'transform_attribute transforms something',
);
($expect = $tmpl) =~ s/ class="main"/ class="main" noggin="zonk"/;
is(
run_for {
$_->transform_attribute({
name => 'noggin',
code => sub { 'zonk' },
})
},
$expect,
'transform_attribute adds attribute if not there before',
);
is(
run_for {
$_->transform_attribute({
name => 'noggin',
code => sub { },
})
},
$tmpl,
'transform_attribute on nonexistent att does not add it if code returns undef',
);
($expect = $tmpl) =~ s/(?=