Can pass template at database creation

Author: mcanevet

manifests/server/database.pp

@@ -3,6 +3,7 @@
   $dbname     = $title,
   $owner      = $postgresql::server::user,
   $tablespace = undef,
+  $template   = 'template0',
   $encoding   = $postgresql::server::encoding,
   $locale     = $postgresql::server::locale,
   $istemplate = false
@@ -43,7 +44,7 @@
     default => "--tablespace='${tablespace}' ",
   }
 
-  $createdb_command = "${createdb_path} --owner='${owner}' --template=template0 ${encoding_option}${locale_option}${tablespace_option} '${dbname}'"
+  $createdb_command = "${createdb_path} --owner='${owner}' --template=${template} ${encoding_option}${locale_option}${tablespace_option} '${dbname}'"
 
   postgresql_psql { "Check for existence of db '${dbname}'":
     command => 'SELECT 1',

manifests/server/db.pp

@@ -7,12 +7,14 @@
   $locale     = $postgresql::server::locale,
   $grant      = 'ALL',
   $tablespace = undef,
+  $template   = 'template0',
   $istemplate = false,
   $owner      = undef
 ) {
   postgresql::server::database { $name:
     encoding   => $encoding,
     tablespace => $tablespace,
+    template   => $template,
     locale     => $locale,
     istemplate => $istemplate,
     owner      => $owner,

spec/system/server/db_spec.rb

@@ -140,4 +140,35 @@ class { 'postgresql::server': }
       end
     end
   end
+
+  it 'should take a template parameter' do
+    begin
+      pp = <<-EOS.unindent
+        $db = 'postgresql_test_db'
+        class { 'postgresql::server': }
+
+        postgresql::server::db { $db:
+          user       => $db,
+          template   => 'template1',
+          password   => postgresql_password($db, $db),
+        }
+      EOS
+
+      puppet_apply(pp) do |r|
+        r.exit_code.should_not == 1
+        r.refresh
+        r.exit_code.should == 0
+      end
+
+      psql('--command="select datname from pg_database" postgresql_test_db') do |r|
+        r.stdout.should =~ /postgresql_test_db/
+        r.stderr.should be_empty
+        r.exit_code.should == 0
+      end
+    ensure
+      psql('--command="drop database postgresql_test_db" postgres') do |r|
+        r.exit_code.should == 0
+      end
+    end
+  end
 end