#
# Author:: Jay Mundrawala (<jdm@chef.io>)
#
# Copyright:: Copyright (c) 2014 Chef Software, Inc.
# License:: Apache License, Version 2.0
#
# 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 'chef'
require 'spec_helper'

describe Chef::Provider::DscResource do
  let (:events) { Chef::EventDispatch::Dispatcher.new }
  let (:run_context) { Chef::RunContext.new(node, {}, events) }
  let (:resource) { Chef::Resource::DscResource.new("dscresource", run_context) }
  let (:provider) do
    Chef::Provider::DscResource.new(resource, run_context)
  end

  context 'when Powershell does not support Invoke-DscResource' do
    let (:node) {
      node = Chef::Node.new
      node.automatic[:languages][:powershell][:version] = '4.0'
      node
    }
    it 'raises a ProviderNotFound exception' do
      expect(provider).not_to receive(:meta_configuration)
      expect{provider.run_action(:run)}.to raise_error(
              Chef::Exceptions::ProviderNotFound, /5\.0\.10018\.0/)
    end
  end

  context 'when Powershell supports Invoke-DscResource' do

    context 'when RefreshMode is not set to Disabled' do
      context 'and the WMF 5 is a preview release' do
        let (:node) {
          node = Chef::Node.new
          node.automatic[:languages][:powershell][:version] = '5.0.10018.0'
          node
        }
        it 'raises an exception' do
          expect(provider).to receive(:dsc_refresh_mode_disabled?).and_return(false)
          expect { provider.run_action(:run) }.to raise_error(
            Chef::Exceptions::ProviderNotFound, /Disabled/)
        end
      end
      context 'and the WMF is 5 RTM or newer' do
        let (:node) {
          node = Chef::Node.new
          node.automatic[:languages][:powershell][:version] = '5.0.10586.0'
          node
        }
        it 'does not raises an exception' do
          expect(provider).to receive(:test_resource)
          expect(provider).to receive(:set_resource)
          expect(provider).to receive(:reboot_if_required)
          expect { provider.run_action(:run) }.to_not raise_error
        end
      end
    end
  end
  
  context 'when the LCM supports Invoke-DscResource' do
    let (:node) {
      node = Chef::Node.new
      node.automatic[:languages][:powershell][:version] = '5.0.10018.0'
      node
    }

    it 'does not update the resource if it is up to date' do
      expect(provider).to receive(:dsc_refresh_mode_disabled?).and_return(true)
      expect(provider).to receive(:test_resource).and_return(true)
      provider.run_action(:run)
      expect(resource).not_to be_updated
    end

    it 'converges the resource if it is not up to date' do
      expect(provider).to receive(:dsc_refresh_mode_disabled?).and_return(true)
      expect(provider).to receive(:test_resource).and_return(false)
      expect(provider).to receive(:set_resource)
      provider.run_action(:run)
      expect(resource).to be_updated
    end
    
    it 'flags the resource as reboot required when required' do
      expect(provider).to receive(:dsc_refresh_mode_disabled?).and_return(true)
      expect(provider).to receive(:test_resource).and_return(false)
      expect(provider).to receive(:invoke_resource).
        and_return(double(:stdout => '', :return_value =>nil))
      expect(provider).to receive(:return_dsc_resource_result).and_return(true)
      expect(provider).to receive(:create_reboot_resource)
      provider.run_action(:run)
    end
    
    it 'does not flag the resource as reboot required when not required' do
      expect(provider).to receive(:dsc_refresh_mode_disabled?).and_return(true)
      expect(provider).to receive(:test_resource).and_return(false)
      expect(provider).to receive(:invoke_resource).
        and_return(double(:stdout => '', :return_value =>nil))
      expect(provider).to receive(:return_dsc_resource_result).and_return(false)
      expect(provider).to_not receive(:create_reboot_resource)
      provider.run_action(:run)
    end
  end
end