4 files changed, 95 insertions, 0 deletions
diff --git a/spec/functional/assets/chocolatey_feed/test-A.1.0.nupkg b/spec/functional/assets/chocolatey_feed/test-A.1.0.nupkg
new file mode 100644
index 0000000000..a2d3e7bb4e
--- /dev/null
+++ b/spec/functional/assets/chocolatey_feed/test-A.1.0.nupkg
diff --git a/spec/functional/assets/chocolatey_feed/test-A.2.0.nupkg b/spec/functional/assets/chocolatey_feed/test-A.2.0.nupkg
new file mode 100644
index 0000000000..250084d3ce
--- /dev/null
+++ b/spec/functional/assets/chocolatey_feed/test-A.2.0.nupkg
diff --git a/spec/functional/assets/chocolatey_feed/test-B.1.0.nupkg b/spec/functional/assets/chocolatey_feed/test-B.1.0.nupkg
new file mode 100644
index 0000000000..f275c78dc6
--- /dev/null
+++ b/spec/functional/assets/chocolatey_feed/test-B.1.0.nupkg
diff --git a/spec/functional/resource/chocolatey_package_spec.rb b/spec/functional/resource/chocolatey_package_spec.rb
new file mode 100644
index 0000000000..c180f8a938
--- /dev/null
+++ b/spec/functional/resource/chocolatey_package_spec.rb
@@ -0,0 +1,95 @@
+#
+# Author:: Matt Wrock (<matt@mattwrock.com>)
+# Copyright:: Copyright (c) 2016 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 'spec_helper'
+require 'chef/mixin/powershell_out'
+
+describe Chef::Resource::ChocolateyPackage, :windows_only do
+  include Chef::Mixin::PowershellOut
+
+  before(:all) do
+    powershell_out("iex ((new-object net.webclient).DownloadString('https://chocolatey.org/install.ps1'))")
+  end
+
+  let(:package_name) { 'test-A' }
+  let(:package_list) { proc { powershell_out("choco list -lo -r #{Array(package_name).join(' ')}").stdout.chomp } }
+  let(:package_source) { File.join(CHEF_SPEC_ASSETS, 'chocolatey_feed') }
+  
+  subject do
+    new_resource = Chef::Resource::ChocolateyPackage.new('test choco package', run_context)
+    new_resource.package_name package_name
+    new_resource.source package_source if package_source
+    new_resource
+  end
+
+  context 'installing a package' do
+    after { Chef::Resource::ChocolateyPackage.new(package_name, run_context).run_action(:remove) }
+
+    it 'installs the latest version' do
+      subject.run_action(:install)
+      expect(package_list.call).to eq("#{package_name}|2.0")
+    end
+
+    it 'does not install if already installed' do
+      subject.run_action(:install)
+      subject.run_action(:install)
+      expect(subject).not_to be_updated_by_last_action
+    end
+
+    it 'installs version given' do
+      subject.version '1.0'
+      subject.run_action(:install)
+      expect(package_list.call).to eq("#{package_name}|1.0")
+    end
+
+    it 'installs new version if one is already installed' do
+      subject.version '1.0'
+      subject.run_action(:install)
+      expect(package_list.call).to eq("#{package_name}|1.0")
+
+      subject.version '2.0'
+      subject.run_action(:install)
+      expect(package_list.call).to eq("#{package_name}|2.0")
+    end
+
+    context 'installing multiple packages' do
+      let(:package_name) { [ 'test-A', 'test-B' ] }
+
+      it 'installs both packages' do
+        subject.run_action(:install)
+        expect(package_list.call).to eq("test-A|2.0\r\ntest-B|1.0")
+      end
+    end
+
+    context 'installing a bogus package' do
+      let(:package_name) { 'blah' }
+      let(:package_source) { nil }
+
+      it 'raises if package is not found' do
+        expect{ subject.run_action(:install) }.to raise_error Chef::Exceptions::Package
+      end
+    end
+  end
+
+  context 'removing a package' do
+    it 'removes an installed package' do
+      subject.run_action(:install)
+      Chef::Resource::ChocolateyPackage.new(package_name, run_context).run_action(:remove)
+      expect(package_list.call).to eq('')
+    end
+  end
+end
+\ No newline at end of file