--- /dev/null
+#!perl -Tw
+
+use strict;
+use Test::More tests => 5;
+
+BEGIN {
+ use_ok( "HTML::TreeBuilder" );
+}
+
+my $html =<<'EOHTML';
+<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"
+"http://www.w3.org/TR/html4/loose.dtd">
+<html>
+<head>
+<meta http-equiv="Content-Type" content="text/html; charset=UTF-8" />
+</head>
+<body>
+blah blah
+</body>
+</html>
+EOHTML
+
+WITH_DECLARATION: { # Check default state
+ my $tree = HTML::TreeBuilder->new;
+ isa_ok( $tree, "HTML::TreeBuilder" );
+
+ $tree->parse( $html );
+ $tree->eof;
+
+ my @lines = split( "\n", $tree->as_HTML(undef, " ") );
+
+ like( $lines[0], qr/DOCTYPE/, "DOCTYPE is in the first line" );
+}
+
+
+WITHOUT_DECLARATION: {
+ my $tree = HTML::TreeBuilder->new;
+ isa_ok( $tree, "HTML::TreeBuilder" );
+
+ $tree->store_declarations(0);
+
+ $tree->parse( $html );
+ $tree->eof;
+
+ my @lines = split( "\n", $tree->as_HTML(undef, " ") );
+
+ unlike( $lines[0], qr/DOCTYPE/, "DOCTYPE is NOT in the first line" );
+}