File Coverage

UnitTestSetup.pm

Criterion	Covered	Total	%
statement	76	79	96.2
branch	8	12	66.6
condition	5	9	55.5
subroutine	15	16	93.7
pod	0	7	0.0
total	104	123	84.5

package UnitTestSetup;

use strict;
use warnings;

use parent qw( Exporter );

use Data::Dumper;
use Date::Format;
use File::Path;
use File::Temp qw( tempdir );
use Test::More;

our @EXPORT_OK
  = qw( init_test format_time TRUE FALSE FIVE_MINUTES ISO_8601_FORMAT);

our %EXPORT_TAGS = ( all => [@EXPORT_OK], );

use constant {
  ISO_8601_FORMAT => '%Y-%m-%dT%H:%M:%SZ',
  TRUE            => 1,
  FALSE           => 0,
  FIVE_MINUTES    => 5 * 60,
};

caller or __PACKAGE__->main();

########################################################################
sub format_time {
########################################################################
  my ($time) = @_;

  return time2str( ISO_8601_FORMAT, time + $time, 'GMT' );
}

########################################################################
sub read_config {
########################################################################
  my ( $fh, $test ) = @_;

  $fh = $fh || *DATA;

  my %configs;

  my $config_name;

  while ( my $line = <$fh> ) {
    chomp $line;
    if ( $line =~ /^--- (.*) ---$/ ) {
      $config_name = $1;
      $configs{$1} = [];
      next;
    }

    push @{ $configs{$config_name} }, $line;
  }

  close $fh;

  return $configs{$test} ? $configs{$test} : $configs{'01-credentials.t'};
}

########################################################################
sub create_credentials_file {
########################################################################
  my ( $home, $credentials, $vars ) = @_;

  # poor mans templating...
  foreach ( @{$credentials} ) {
    next if !/(@[^@]+@)/;

    my $tmpl_var = $1;

    my $var = $tmpl_var;
    $var =~ s/@//g;

    my $val = $vars->{$var};

    s/$tmpl_var/$val/g;
  }

  mkdir "$home/.aws";

  open( my $fh, '>', "$home/.aws/credentials" )
    or BAIL_OUT('could not create temporary credentials file');

  print {$fh} join "\n", @{$credentials};

  close $fh;

  return "$home/.aws/credentials";
}

########################################################################
sub create_config_file {
########################################################################
  my ($home) = @_;

  if ( !-d "$home/.aws" ) {
    mkdir "$home/.aws";
  }

  open my $fh, '>', "$home/.aws/config"
    or BAIL_OUT('could not create temporary config file');

  print {$fh} join "\n", qw{[default] region=us-east-2};

  close $fh;

  return "$home/.aws/config";
}

########################################################################
sub create_home_dir {
########################################################################
  my ($cleanup) = @_;

  my $home = tempdir( 'amz-credentials-XXXXX', CLEANUP => $cleanup );

  $ENV{HOME} = $home;

  return $home;
}

########################################################################
sub init_test {
########################################################################
  my (%args) = @_;

  $ENV{'AWS_PROFILE'} = $args{'profile'} // 'default';

  my $credentials = read_config( *DATA, $args{'test'} // '01-credentials.t' );

  my $home = create_home_dir( $args{'cleanup'} // 1 );

  create_credentials_file( $home, $credentials, $args{'vars'} );
  create_config_file($home);

  return $home;
}

sub main {
  return print {*STDERR} Dumper [
    init_test(
      cleanup => 0,
      test    => '01-credentials.t',
      vars    => { process => 'foo' },
    )
  ];
}

1;

__DATA__