X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=3rdparty%2Fmodules%2Fstdlib%2Fspec%2Ffunctions%2Floadyaml_spec.rb;h=d71fd32bccc7f8831e848cf788fa018b9d67d5a5;hb=59812388abf4638d6065ffda934d1866af902302;hp=cdc3d6f5150bde6aaf894ada21bc1566413f3d2b;hpb=ad88f67c13ae0f1a08936dad643f1e3509ab5f40;p=mirror%2Fdsa-puppet.git

diff --git a/3rdparty/modules/stdlib/spec/functions/loadyaml_spec.rb b/3rdparty/modules/stdlib/spec/functions/loadyaml_spec.rb
old mode 100755
new mode 100644
index cdc3d6f51..d71fd32bc
--- a/3rdparty/modules/stdlib/spec/functions/loadyaml_spec.rb
+++ b/3rdparty/modules/stdlib/spec/functions/loadyaml_spec.rb
@@ -1,25 +1,101 @@
-#! /usr/bin/env ruby -S rspec
 require 'spec_helper'
 
-describe "the loadyaml function" do
-  include PuppetlabsSpec::Files
+describe 'loadyaml' do
+  it { is_expected.not_to eq(nil) }
+  it { is_expected.to run.with_params.and_raise_error(ArgumentError, %r{wrong number of arguments}i) }
 
-  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+  context 'when a non-existing file is specified' do
+    let(:filename) { '/tmp/doesnotexist' }
 
-  it "should exist" do
-    expect(Puppet::Parser::Functions.function("loadyaml")).to eq("function_loadyaml")
+    it "'default' => 'value'" do
+      expect(File).to receive(:exists?).with(filename).and_return(false).once
+      expect(YAML).to receive(:load_file).never
+      is_expected.to run.with_params(filename, 'default' => 'value').and_return('default' => 'value')
+    end
   end
 
-  it "should raise a ParseError if there is less than 1 arguments" do
-    expect { scope.function_loadyaml([]) }.to raise_error(Puppet::ParseError)
+  context 'when an existing file is specified' do
+    let(:filename) { '/tmp/doesexist' }
+    let(:data) { { 'key' => 'value', 'ķęŷ' => 'νậŀųề', 'キー' => '値' } }
+
+    it "returns 'key' => 'value', 'ķęŷ' => 'νậŀųề', 'キー' => '値'" do
+      expect(File).to receive(:exists?).with(filename).and_return(true).once
+      expect(YAML).to receive(:load_file).with(filename).and_return(data).once
+      is_expected.to run.with_params(filename).and_return(data)
+    end
   end
 
-  it "should convert YAML file to a data structure" do
-    yaml_file = tmpfilename ('yamlfile')
-    File.open(yaml_file, 'w') do |fh|
-      fh.write("---\n aaa: 1\n bbb: 2\n ccc: 3\n ddd: 4\n")
+  context 'when the file could not be parsed' do
+    let(:filename) { '/tmp/doesexist' }
+
+    it 'filename /tmp/doesexist' do
+      expect(File).to receive(:exists?).with(filename).and_return(true).once
+      allow(YAML).to receive(:load_file).with(filename).once.and_raise(StandardError, 'Something terrible have happened!')
+      is_expected.to run.with_params(filename, 'default' => 'value').and_return('default' => 'value')
     end
-    result = scope.function_loadyaml([yaml_file])
-    expect(result).to eq({"aaa" => 1, "bbb" => 2, "ccc" => 3, "ddd" => 4 })
+  end
+
+  context 'when an existing URL is specified' do
+    let(:filename) { 'https://example.local/myhash.yaml' }
+    let(:basic_auth) { { :http_basic_authentication => ['', ''] } }
+    let(:yaml) { 'Dummy YAML' }
+    let(:data) { { 'key' => 'value', 'ķęŷ' => 'νậŀųề', 'キー' => '値' } }
+
+    it {
+      expect(OpenURI).to receive(:open_uri).with(filename, basic_auth).and_return(yaml)
+      expect(YAML).to receive(:safe_load).with(yaml).and_return(data).once
+      is_expected.to run.with_params(filename).and_return(data)
+    }
+  end
+
+  context 'when an existing URL (with username and password) is specified' do
+    let(:filename) { 'https://user1:pass1@example.local/myhash.yaml' }
+    let(:url_no_auth) { 'https://example.local/myhash.yaml' }
+    let(:basic_auth) { { :http_basic_authentication => ['user1', 'pass1'] } }
+    let(:yaml) { 'Dummy YAML' }
+    let(:data) { { 'key' => 'value', 'ķęŷ' => 'νậŀųề', 'キー' => '値' } }
+
+    it {
+      expect(OpenURI).to receive(:open_uri).with(url_no_auth, basic_auth).and_return(yaml)
+      expect(YAML).to receive(:safe_load).with(yaml).and_return(data).once
+      is_expected.to run.with_params(filename).and_return(data)
+    }
+  end
+
+  context 'when an existing URL (with username) is specified' do
+    let(:filename) { 'https://user1@example.local/myhash.yaml' }
+    let(:url_no_auth) { 'https://example.local/myhash.yaml' }
+    let(:basic_auth) { { :http_basic_authentication => ['user1', ''] } }
+    let(:yaml) { 'Dummy YAML' }
+    let(:data) { { 'key' => 'value', 'ķęŷ' => 'νậŀųề', 'キー' => '値' } }
+
+    it {
+      expect(OpenURI).to receive(:open_uri).with(url_no_auth, basic_auth).and_return(yaml)
+      expect(YAML).to receive(:safe_load).with(yaml).and_return(data).once
+      is_expected.to run.with_params(filename).and_return(data)
+    }
+  end
+
+  context 'when an existing URL could not be parsed, with default specified' do
+    let(:filename) { 'https://example.local/myhash.yaml' }
+    let(:basic_auth) { { :http_basic_authentication => ['', ''] } }
+    let(:yaml) { 'Dummy YAML' }
+
+    it {
+      expect(OpenURI).to receive(:open_uri).with(filename, basic_auth).and_return(yaml)
+      expect(YAML).to receive(:safe_load).with(yaml).and_raise StandardError, 'Cannot parse data'
+      is_expected.to run.with_params(filename, 'default' => 'value').and_return('default' => 'value')
+    }
+  end
+
+  context 'when a URL does not exist, with default specified' do
+    let(:filename) { 'https://example.local/myhash.yaml' }
+    let(:basic_auth) { { :http_basic_authentication => ['', ''] } }
+    let(:yaml) { 'Dummy YAML' }
+
+    it {
+      expect(OpenURI).to receive(:open_uri).with(filename, basic_auth).and_raise OpenURI::HTTPError, '404 File not Found'
+      is_expected.to run.with_params(filename, 'default' => 'value').and_return('default' => 'value')
+    }
   end
 end