module-postgresql/spec/postgresql_spec.rb

# puppet-postgresql
# For all details and documentation:
# http://github.com/inkling/puppet-postgresql
#
# Copyright 2012- Inkling Systems, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

require 'logger'
require 'vagrant'

# VM-based tests for postgresql::* defined resource types.
#
# The general structure is:
#  
#  - Roll back the VM
#  - Apply a minimal puppet config for the type and fail only on crashes
#  - Apply it again and fail if there were any new changes reported
#  - Check the state of the system

describe "postgresql" do

  def sudo_and_log(*args)
     @env.primary_vm.channel.sudo(args[0]) do |ch, data|
       @logger.debug(data)
     end
  end

  before(:all) do
    @logger = Logger.new(STDOUT)
    @logger.level = Logger::WARN # TODO: get from environment or rspec?

    vagrant_dir = File.dirname(__FILE__)
    @env = Vagrant::Environment::new(:cwd => vagrant_dir)

    # Sahara ignores :cwd so we have to chdir for now, see https://github.com/jedi4ever/sahara/issues/9
    Dir.chdir(vagrant_dir)

    # @env.cli("destroy") # Takes too long
    @env.cli("up")
    
    # We are not testing the "package" resource type, so pull stuff in in advance
    sudo_and_log('apt-get update')
    sudo_and_log('apt-get install --yes --download-only postgresql-8.4')
    @env.cli("sandbox", "on") 
  end
  
  after(:each) do
    @env.cli("sandbox", "rollback")
  end

  describe 'postgresql::initdb' do
    it "should idempotently create a working --pgdata directory so postgres can run" do
      @logger.info("starting")

      # A bare-minimum class to initdb the specified dir
      test_class = 'class {"postgresql_tests::test_initdb": dir => "/tmp/initdb", version => "8.4" }'
 
      # Run once to check for crashes
      sudo_and_log("puppet apply -e '#{test_class}'")

      # Run again to check for idempotence via --detailed-exitcodes
      sudo_and_log("puppet apply --detailed-exitcodes -e '#{test_class}'")
      
      # Run postgres on a different port so we don't have to deal w/ the one ubuntu starts up
      # NOTE: -n prevent prompting for password in case sudoers is wonky
      # NOTE: -l prevents hanging attach to stdout of pg_ctl
      # NOTE: -o "-p 57575" passes the option on to the underlying posgres command
      sudo_and_log('sudo -n -u postgres /usr/lib/postgresql/8.4/bin/pg_ctl start --pgdata /tmp/initdb -l /tmp/pgctl_stdout -o "-p 57575" && sleep 1')

      # Connect to it and list the databases
      sudo_and_log('sudo -n -u postgres /usr/lib/postgresql/8.4/bin/psql -p 57575 --list --tuples-only')
    end
  end

  describe 'postgresql::db' do
    it 'should idempotently create a db that we can connect to' do
      
      # A bare-minimum class to add a DB to postgres, which will be running due to ubuntu
      test_class = 'class {"postgresql_tests::test_db": db => "postgresql_test_db", version => "8.4" }'
  
      # Run once to check for crashes
      sudo_and_log("puppet apply -e '#{test_class}'")
  
      # Run again to check for idempotence
      sudo_and_log("puppet apply --detailed-exitcodes -e '#{test_class}'")

      # Check that the database name is present
      sudo_and_log('sudo -u postgres psql postgresql_test_db --command="select datname from pg_database limit 1"')
    end
  end

  describe 'postgresql::psql' do
    it 'should run some SQL when the unless query returns no rows' do
      test_class = 'class {"postgresql_tests::test_psql": command => "SELECT \'foo\'", unless => "SELECT 1 WHERE 1=2", version => "8.4"}'
      
      # Run once to get all packages set up
      sudo_and_log("puppet apply -e '#{test_class}'")
      
      # Check for exit code 2
      sudo_and_log("puppet apply --detailed-exitcodes -e '#{test_class}' ; [ $? == 2 ]")
    end
    
    it 'should not run SQL when the unless query returns rows' do
      test_class = 'class {"postgresql_tests::test_psql": command => "SELECT * FROM pg_datbase limit 1", unless => "SELECT 1 WHERE 1=1", version => "8.4"}'

      # Run once to get all packages set up
      sudo_and_log("puppet apply -e '#{test_class}'")

      # Check for exit code 0
      sudo_and_log("puppet apply --detailed-exitcodes -e '#{test_class}'")
    end
  end

  describe 'postgresql::user' do
    it 'should idempotently create a user who can log in' do
      test_class = 'class {"postgresql_tests::test_user": user => "postgresql_test_user", password => "postgresql_test_password", version => "8.4"}'
      
      # Run once to check for crashes
      sudo_and_log("puppet apply -e '#{test_class}'")
  
      # Run again to check for idempotence
      sudo_and_log("puppet apply --detailed-exitcodes -e '#{test_class}'")
  
      # Check that the user can log in
      sudo_and_log('sudo -u postgresql_test_user psql --command="select datname from pg_database limit 1" postgres')
    end
  end
  
  describe 'postgresql::grant' do
    it 'should grant access so a user can select from a table' do
      test_class = 'class {"postgresql_tests::test_grant_select": db => "postgres", table => "test_table", user => "psql_grant_tester", password => "psql_grant_pw", version => "8.4"}'
      
      # Run once to check for crashes
      sudo_and_log("puppet apply -e '#{test_class}'")
  
      # Run again to check for idempotence
      sudo_and_log("puppet apply --detailed-exitcodes -e '#{test_class}'")
  
      # Check that the user can select from the table in
      sudo_and_log('sudo -u psql_grant_tester psql --command="select * from test_table limit 1" postgres')
    end
  end
end