1 files changed, 93 insertions, 0 deletions
diff --git a/spec/unit/resource/rhsm_subscription_spec.rb b/spec/unit/resource/rhsm_subscription_spec.rb
new file mode 100644
index 0000000000..0160624f39
--- /dev/null
+++ b/spec/unit/resource/rhsm_subscription_spec.rb
@@ -0,0 +1,93 @@
+#
+# Copyright:: Copyright 2018, 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"
+
+describe Chef::Resource::RhsmSubscription do
+  let(:resource) { Chef::Resource::RhsmSubscription.new("foo") }
+  let(:provider) { resource.provider_for_action(:attach) }
+
+  it "has a resource name of :rhsm_subscription" do
+    expect(resource.resource_name).to eql(:rhsm_subscription)
+  end
+
+  it "has a default action of attach" do
+    expect(resource.action).to eql([:attach])
+  end
+
+  it "the pool_id property is the name property" do
+    expect(resource.pool_id).to eql("foo")
+  end
+
+  describe "#subscription_attached?" do
+    let(:cmd)    { double("cmd") }
+    let(:output) { "Pool ID:    pool123" }
+
+    before do
+      allow(Mixlib::ShellOut).to receive(:new).and_return(cmd)
+      allow(cmd).to receive(:run_command)
+      allow(cmd).to receive(:stdout).and_return(output)
+    end
+
+    context "when the pool provided matches the output" do
+      it "returns true" do
+        expect(provider.subscription_attached?("pool123")).to eq(true)
+      end
+    end
+
+    context "when the pool provided does not match the output" do
+      it "returns false" do
+        expect(provider.subscription_attached?("differentpool")).to eq(false)
+      end
+    end
+  end
+
+  describe "#serials_by_pool" do
+    let(:cmd) { double("cmd") }
+    let(:output) do
+      <<~EOL
+        Key1:       value1
+        Pool ID:    pool1
+        Serial:     serial1
+        Key2:       value2
+
+        Key1:       value1
+        Pool ID:    pool2
+        Serial:     serial2
+        Key2:       value2
+EOL
+    end
+
+    it "parses the output correctly" do
+      allow(Mixlib::ShellOut).to receive(:new).and_return(cmd)
+      allow(cmd).to receive(:run_command)
+      allow(cmd).to receive(:stdout).and_return(output)
+
+      expect(provider.serials_by_pool["pool1"]).to eq("serial1")
+      expect(provider.serials_by_pool["pool2"]).to eq("serial2")
+    end
+  end
+
+  describe "#pool_serial" do
+    let(:serials) { { "pool1" => "serial1", "pool2" => "serial2" } }
+
+    it "returns the serial for a given pool" do
+      allow(provider).to receive(:serials_by_pool).and_return(serials)
+      expect(provider.pool_serial("pool1")).to eq("serial1")
+    end
+  end
+end