Rakefile

#!/usr/bin/env rake

require 'rake/testtask'
require 'rubocop/rake_task'
require_relative 'test/integration/configuration/gcp_config'

# Rubocop
desc 'Run Rubocop lint checks'
task :rubocop do
  RuboCop::RakeTask.new
end

# lint the project
desc 'Run robocop linter'
task lint: [:rubocop]

# run tests
task default: [:lint, 'test:check']

namespace :test do
  # Specify the directory for the integration tests
  integration_dir = "test/integration"

  # Specify the terraform plan name
  plan_name = "gcp-kube.plan"

  # Specify the file_name for terraform variables to be stored
  variable_file_name = "gcp-kube.tfvars"

  # The below file allows to inject parameters as profile attributes to inspec
  profile_attributes = "gcp-kube-attributes.yaml"

  # run inspec check to verify that the profile is properly configured
  task :check do
    dir = File.join(File.dirname(__FILE__))
    sh("bundle exec inspec check #{dir}")
    # run inspec check on the sample profile to ensure all resources are loaded okay
    sh("bundle exec inspec check .")
  end

  task :init_workspace do
    # Initialize terraform workspace
    cmd = format("cd %s/build/ && terraform init", integration_dir)
    sh(cmd)
  end

  task :plan_integration_tests do

    puts "----> Generating terraform and inspec variable files"
    GcpConfig.store_json(variable_file_name)
    GcpConfig.store_yaml(profile_attributes)

    puts "----> Setup"
    # Create the plan that can be applied to GCP
    cmd = format("cd %s/build/ && terraform plan  -var-file=%s -out %s", integration_dir, variable_file_name, plan_name)
    #   puts cmd
    sh(cmd)

  end

  task :setup_integration_tests do
    # Apply the plan on GCP
    cmd = format("cd %s/build/ && terraform apply %s", integration_dir, plan_name)
    sh(cmd)
  end

  task :run_integration_tests do
    puts "----> Run"
    # Since the default behaviour is to skip tests, the below absorbs an inspec "101 run okay + skipped only" exit code as successful
    cmd = format("bundle exec inspec exec . --attrs %s/build/%s -t gcp://; rc=$?; if [ $rc -eq 0 ] || [ $rc -eq 101 ]; then exit 0; else exit 1; fi", integration_dir, profile_attributes)
    sh(cmd)
  end

  # task :get_package_creds do
  #   GcpConfig.update_from_environment
  #   puts "----> Getting credentials for service account (note you must have the Service Account Key Admin role)"
  #   cmd = format("gcloud iam service-accounts keys create ~/key%s.json --iam-account %s@%s.iam.gserviceaccount.com",GcpConfig.config[:hab_container_service_account_name],GcpConfig.config[:hab_container_service_account_name],GcpConfig.config[:gcp_project_id])
  #   sh(cmd)
  #   puts "Now run: docker login -u _json_key --password-stdin https://gcr.io < ~/key#{GcpConfig.config[:hab_container_service_account_name]}.json"
  # end

  task :cleanup_integration_tests do
    puts "----> Cleanup"
    cmd = format("cd %s/build/ && terraform destroy -force -var-file=%s || true", integration_dir, variable_file_name)
    sh(cmd)
  end

  task :setup_cluster do
    GcpConfig.update_from_environment
    puts "----> Getting credentials for kube cluster"
    cmd = format("gcloud container clusters get-credentials %s --zone %s --project %s",GcpConfig.config[:gcp_kube_cluster_name],GcpConfig.config[:gcp_kube_cluster_zone],GcpConfig.config[:gcp_project_id])
    sh(cmd)
    puts "----> Ensure current user can administer cluster"
    cmd = "kubectl create clusterrolebinding cluster-admin-binding --clusterrole cluster-admin --user $(gcloud config get-value account)"
    sh(cmd)
    puts "----> Setting up Helm ..."
    setup_helm = File.join(File.dirname(__FILE__),'scripts','setup_helm.sh')
    sh(setup_helm)
    puts "----> Setting up Habitat ..."
    setup_hab = File.join(File.dirname(__FILE__),'scripts','setup_habitat.sh')
    sh(setup_hab)
    puts "----> Setting up Service Catalog and GCP Broker ..."
    setup_osb = File.join(File.dirname(__FILE__),'scripts','setup_service_catalog.sh')
    sh(setup_osb)
    puts '----> Cluster automated setup finished. Run the following commands interactively to complete setup:'
    puts 'sc add-gcp-broker'
    puts 'export GCP_PROJECT_ID=$(gcloud config get-value project)'
    puts 'export GCP_PROJECT_NUMBER=$(gcloud projects describe $GCP_PROJECT_ID --format=\'value(projectNumber)\')'
    puts 'gcloud projects add-iam-policy-binding ${GCP_PROJECT_ID} --member serviceAccount:${GCP_PROJECT_NUMBER}@cloudservices.gserviceaccount.com --role=roles/owner'
  end

  desc "Setup Cluster Without Cleanup"
  task :integration do
    Rake::Task["test:init_workspace"].execute
    if File.exists?(File.join(integration_dir,"build",variable_file_name))
      Rake::Task["test:cleanup_integration_tests"].execute
    end
    Rake::Task["test:plan_integration_tests"].execute
    Rake::Task["test:setup_integration_tests"].execute
    Rake::Task["test:run_integration_tests"].execute
    Rake::Task["test:setup_cluster"].execute
    #Rake::Task["test:cleanup_integration_tests"].execute
  end
end

# Automatically generate a changelog for this project. Only loaded if
# the necessary gem is installed.
# use `rake changelog to=1.2.0`
begin
  v = ENV['to']
  require 'github_changelog_generator/task'
  GitHubChangelogGenerator::RakeTask.new :changelog do |config|
    config.future_release = v
  end
rescue LoadError
  puts '>>>>> GitHub Changelog Generator not loaded, omitting tasks'
end