newcontext-oss/kitchen-terraform

# frozen_string_literal: true

# Copyright 2016-2021 Copado NCS LLC
#
# 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 "kitchen"
require "kitchen/terraform/transport/connection"
require "kitchen/transport/exec"
require "mixlib/shellout"

::RSpec.describe ::Kitchen::Terraform::Transport::Connection do
  subject do
    described_class.new options
  end

  let :client do
    instance_double ::String
  end

  let :command_timeout do
    instance_double ::Integer
  end

  let :environment_variable_key do
    instance_double ::String
  end

  let :environment_variable_value do
    instance_double ::String
  end

  let :instance_name do
    instance_double ::String
  end

  let :kitchen_root do
    # kitchen_root must support implicit string conversion
    "test-kitchen-root"
  end

  let :options do
    {
      client: client,
      command_timeout: command_timeout,
      environment: { environment_variable_key => environment_variable_value },
      instance_name: instance_name,
      kitchen_root: kitchen_root,
      root_module_directory: root_module_directory,
    }
  end

  let :root_module_directory do
    instance_double ::String
  end

  describe "#execute" do
    let :command do
      instance_double ::String
    end

    let :shell_out do
      instance_double ::Mixlib::ShellOut
    end

    let :stdout do
      instance_double ::String
    end

    specify "should invoke the ShellOut superclass implementation with the client and options configured" do
      allow(subject).to receive(:run_from_file_command).with("#{client} #{command}").and_return "#{client} #{command}"
      allow(subject).to receive :close
      allow(::Mixlib::ShellOut).to receive(:new).with(
        "#{client} #{command}",
        including({
          cwd: root_module_directory,
          environment: {
            "LC_ALL" => nil,
            environment_variable_key => environment_variable_value,
            "TF_IN_AUTOMATION" => "true",
          },
          timeout: command_timeout,
        })
      ).and_return shell_out
      allow(shell_out).to receive :run_command
      allow(shell_out).to receive :execution_time
      allow(shell_out).to receive :error!
      allow(shell_out).to receive(:stdout).and_return stdout

      expect(subject.execute(command)).to be stdout
    end
  end
end