diff options
Diffstat (limited to 't/CP_Testing.pm')
| -rw-r--r-- | t/CP_Testing.pm | 38 |
1 files changed, 34 insertions, 4 deletions
diff --git a/t/CP_Testing.pm b/t/CP_Testing.pm index 1bf8c4522..af1fa2f62 100644 --- a/t/CP_Testing.pm +++ b/t/CP_Testing.pm @@ -218,8 +218,8 @@ sub create_fake_pg_table { ## Dangerous: do not try this at home! my $self = shift; - my $name = shift; - my $dbh = $self->{dbh}; + my $name = shift || die; + my $dbh = $self->{dbh} || die; my $dbuser = $self->{testuser} || die; { local $dbh->{Warn}; @@ -235,8 +235,8 @@ sub create_fake_pg_table { sub remove_fake_pg_table { my $self = shift; - my $name = shift; - my $dbh = $self->{dbh}; + my $name = shift || die; + my $dbh = $self->{dbh} || die; my $dbuser = $self->{testuser} || die; { local $dbh->{Warn}; @@ -259,4 +259,34 @@ sub table_exists { } ## end of table_exists +sub fake_version { + + my $self = shift; + my $version = shift || '9.9'; + my $dbh = $self->{dbh} || die; + my $dbuser = $self->{testuser} || die; + + $dbh->do(qq{ +CREATE OR REPLACE FUNCTION public.version() +RETURNS TEXT +LANGUAGE SQL +AS \$\$ +SELECT 'PostgreSQL $version on fakefunction for check_postgres.pl testing'::text; +\$\$ +}); + $dbh->do("ALTER USER $dbuser SET search_path = public, pg_catalog"); + $dbh->commit(); + +} ## end of fake version + +sub reset_path { + + my $self = shift; + my $dbh = $self->{dbh} || die; + my $dbuser = $self->{testuser} || die; + $dbh->do("ALTER USER $dbuser SET search_path = public"); + $dbh->commit(); + +} ## end of reset_path + 1; |