python_works.rb

title 'Tests to confirm python2 works as expected'

plan_name = input('plan_name', value: 'python2')
plan_ident = "#{ENV['HAB_ORIGIN']}/#{plan_name}"
allow_list_stdout = input('allow_list_stdout')
allow_list_stderr = input('allow_list_stderr')
block_list = input('block_list')

control 'core-plans-python2' do
  impact 1.0
  title 'Ensure python2 works'
  desc '
  For python we check that the version is correctly returned e.g.

    $ python --version
    Python 2.7.12

  As is often customary, we also say hello to the world with python:

    $ python -c "print(\'hello world\')"
    hello world
  '
  python_pkg_ident = command("hab pkg path #{plan_ident}")
  describe python_pkg_ident do
    its('stdout') { should_not be_empty }
    its('exit_status') { should eq 0 }
  end
  python_pkg_ident = python_pkg_ident.stdout.strip

  describe command("#{python_pkg_ident}/bin/python --version") do
    its('stdout') { should be_empty }
    its('stderr') { should match /Python\s+#{python_pkg_ident.split('/')[5]}/ }
    its('exit_status') { should eq 0 }
  end

  describe command("#{python_pkg_ident}/bin/python -c \"print('hello world')\"") do
    its('stdout') { should_not be_empty }
    its('stdout') { should match /hello world/ }
    its('exit_status') { should eq 0 }
  end
end

control 'core-plans-binaries' do
  impact 1.0
  title 'Ensure binaries work'
  desc '
  To check the binaries that python installs, we first locate the path where they are installed.
  Using this path we can then check that the binaries exist, are not empty and are executable.
  Then we run a version check to ensure the binary works and is of the correct version.
  '

  hab_pkg_path = command("hab pkg path #{plan_ident}")
  describe hab_pkg_path do
    its('stdout') { should_not be_empty }
    its('exit_status') { should eq 0 }
  end
  hab_pkg_path = hab_pkg_path.stdout.strip

  # All binaries to test
  binaries_to_test = [allow_list_stdout, allow_list_stderr, block_list].reduce([], :concat)
  binaries_to_test.each do |binary|
    describe file("#{hab_pkg_path}/bin/#{binary}") do
      it { should exist }
      it { should be_executable }
    end
  end

  allow_list_stdout.each do |binary|
    describe command("#{hab_pkg_path}/bin/#{binary} --version") do
      its('stdout') { should_not be_empty }
      its('stdout') { should match /([0-9](\.)?){2,}/ }
      its('exit_status') { should eq 0 }
    end
  end
  
  allow_list_stderr.each do |binary|
    describe command("#{hab_pkg_path}/bin/#{binary} --version") do
      its('stdout') { should be_empty }
      its('stderr') { should match /([0-9](\.)?){2,}/ }
      its('stderr') { should_not be_empty }
      its('exit_status') { should eq 0 }
    end
  end
end