Test fqdn_rand_base64() and fqdn_rand_string()

Author: elyscape

spec/acceptance/fqdn_rand_base64_spec.rb

@@ -0,0 +1,60 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper_acceptance'
+
+describe 'fqdn_rand_base64 function', :unless => unsupported_platforms.include?(fact('operatingsystem')) do
+  describe 'success' do
+    let(:facts_d) do
+      if fact('is_pe', '--puppet') == "true"
+        if fact('osfamily') =~ /windows/i
+          if fact('kernelmajversion').to_f < 6.0
+            'c:/documents and settings/all users/application data/puppetlabs/facter/facts.d'
+          else
+            'c:/programdata/puppetlabs/facter/facts.d'
+          end
+        else
+          '/etc/puppetlabs/facter/facts.d'
+        end
+      else
+        '/etc/facter/facts.d'
+      end
+    end
+    after :each do
+      shell("if [ -f '#{facts_d}/fqdn.txt' ] ; then rm '#{facts_d}/fqdn.txt' ; fi")
+    end
+    before :each do
+      #no need to create on windows, pe creates by default
+      if fact('osfamily') !~ /windows/i
+        shell("mkdir -p '#{facts_d}'")
+      end
+    end
+    it 'generates random base64 strings' do
+      shell("echo fqdn=fakehost.localdomain > '#{facts_d}/fqdn.txt'")
+      pp = <<-eos
+      $l = 10
+      $o = fqdn_rand_base64($l)
+      notice(inline_template('fqdn_rand_base64 is <%= @o.inspect %>'))
+      eos
+
+      apply_manifest(pp, :catch_failures => true) do |r|
+        expect(r.stdout).to match(/fqdn_rand_base64 is "8ySYp0dq0B"/)
+      end
+    end
+    it 'generates random base64 strings with custom seeds' do
+      shell("echo fqdn=fakehost.localdomain > '#{facts_d}/fqdn.txt'")
+      pp = <<-eos
+      $l = 10
+      $s = 'seed'
+      $o = fqdn_rand_base64($l, $s)
+      notice(inline_template('fqdn_rand_base64 is <%= @o.inspect %>'))
+      eos
+
+      apply_manifest(pp, :catch_failures => true) do |r|
+        expect(r.stdout).to match(/fqdn_rand_base64 is "6J2c4oMRUJ"/)
+      end
+    end
+  end
+  describe 'failure' do
+    it 'handles improper argument counts'
+    it 'handles non-numbers for length argument'
+  end
+end

spec/acceptance/fqdn_rand_string_spec.rb

@@ -0,0 +1,60 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper_acceptance'
+
+describe 'fqdn_rand_string function', :unless => unsupported_platforms.include?(fact('operatingsystem')) do
+  describe 'success' do
+    let(:facts_d) do
+      if fact('is_pe', '--puppet') == "true"
+        if fact('osfamily') =~ /windows/i
+          if fact('kernelmajversion').to_f < 6.0
+            'c:/documents and settings/all users/application data/puppetlabs/facter/facts.d'
+          else
+            'c:/programdata/puppetlabs/facter/facts.d'
+          end
+        else
+          '/etc/puppetlabs/facter/facts.d'
+        end
+      else
+        '/etc/facter/facts.d'
+      end
+    end
+    after :each do
+      shell("if [ -f '#{facts_d}/fqdn.txt' ] ; then rm '#{facts_d}/fqdn.txt' ; fi")
+    end
+    before :each do
+      #no need to create on windows, pe creates by default
+      if fact('osfamily') !~ /windows/i
+        shell("mkdir -p '#{facts_d}'")
+      end
+    end
+    it 'generates random alphanumeric strings' do
+      shell("echo fqdn=fakehost.localdomain > '#{facts_d}/fqdn.txt'")
+      pp = <<-eos
+      $l = 10
+      $o = fqdn_rand_string($l)
+      notice(inline_template('fqdn_rand_string is <%= @o.inspect %>'))
+      eos
+
+      apply_manifest(pp, :catch_failures => true) do |r|
+        expect(r.stdout).to match(/fqdn_rand_string is "7oDp0KOr1b"/)
+      end
+    end
+    it 'generates random alphanumeric strings with custom seeds' do
+      shell("echo fqdn=fakehost.localdomain > '#{facts_d}/fqdn.txt'")
+      pp = <<-eos
+      $l = 10
+      $s = 'seed'
+      $o = fqdn_rand_string($l, $s)
+      notice(inline_template('fqdn_rand_string is <%= @o.inspect %>'))
+      eos
+
+      apply_manifest(pp, :catch_failures => true) do |r|
+        expect(r.stdout).to match(/fqdn_rand_string is "3HS4mbuI3E"/)
+      end
+    end
+  end
+  describe 'failure' do
+    it 'handles improper argument counts'
+    it 'handles non-numbers for length argument'
+  end
+end

spec/functions/fqdn_rand_base64_spec.rb

@@ -0,0 +1,65 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper'
+
+describe "the fqdn_rand_base64 function" do
+  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+  it "should exist" do
+    expect(Puppet::Parser::Functions.function("fqdn_rand_base64")).to eq("function_fqdn_rand_base64")
+  end
+
+  it "should raise an ArgumentError if there is less than 1 argument" do
+    expect { fqdn_rand_base64() }.to( raise_error(ArgumentError, /wrong number of arguments/))
+  end
+
+  it "should raise an ArgumentError if argument 1 isn't a positive integer" do
+    expect { fqdn_rand_base64(0) }.to( raise_error(ArgumentError, /first argument must be a positive integer/))
+    expect { fqdn_rand_base64(-1) }.to( raise_error(ArgumentError, /first argument must be a positive integer/))
+    expect { fqdn_rand_base64(0.5) }.to( raise_error(ArgumentError, /first argument must be a positive integer/))
+  end
+
+  it "provides a random base64 string" do
+    base64 = %r{\A[a-zA-Z0-9+/]*\z}
+    expect(fqdn_rand_base64(100).match(base64)).not_to eq(nil)
+  end
+
+  it "provides a random string exactly as long as the given length" do
+    expect(fqdn_rand_base64(10).size).to eql(10)
+  end
+
+  it "provides the same 'random' value on subsequent calls for the same host" do
+    expect(fqdn_rand_base64(10)).to eql(fqdn_rand_base64(10))
+  end
+
+  it "considers the same host and same extra arguments to have the same random sequence" do
+    first_random = fqdn_rand_base64(10, :extra_identifier => [1, "same", "host"])
+    second_random = fqdn_rand_base64(10, :extra_identifier => [1, "same", "host"])
+
+    expect(first_random).to eql(second_random)
+  end
+
+  it "allows extra arguments to control the random value on a single host" do
+    first_random = fqdn_rand_base64(10, :extra_identifier => [1, "different", "host"])
+    second_different_random = fqdn_rand_base64(10, :extra_identifier => [2, "different", "host"])
+
+    expect(first_random).not_to eql(second_different_random)
+  end
+
+  it "should return different strings for different hosts" do
+    val1 = fqdn_rand_base64(10, :host => "first.host.com")
+    val2 = fqdn_rand_base64(10, :host => "second.host.com")
+
+    expect(val1).not_to eql(val2)
+  end
+
+  def fqdn_rand_base64(max, args = {})
+    host = args[:host] || '127.0.0.1'
+    extra = args[:extra_identifier] || []
+
+    scope = PuppetlabsSpec::PuppetInternals.scope
+    scope.stubs(:[]).with("::fqdn").returns(host)
+    scope.stubs(:lookupvar).with("::fqdn").returns(host)
+
+    scope.function_fqdn_rand_base64([max] + extra)
+  end
+end

spec/functions/fqdn_rand_string_spec.rb

@@ -0,0 +1,65 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper'
+
+describe "the fqdn_rand_string function" do
+  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+  it "should exist" do
+    expect(Puppet::Parser::Functions.function("fqdn_rand_string")).to eq("function_fqdn_rand_string")
+  end
+
+  it "should raise an ArgumentError if there is less than 1 argument" do
+    expect { fqdn_rand_string() }.to( raise_error(ArgumentError, /wrong number of arguments/))
+  end
+
+  it "should raise an ArgumentError if argument 1 isn't a positive integer" do
+    expect { fqdn_rand_string(0) }.to( raise_error(ArgumentError, /first argument must be a positive integer/))
+    expect { fqdn_rand_string(-1) }.to( raise_error(ArgumentError, /first argument must be a positive integer/))
+    expect { fqdn_rand_string(0.5) }.to( raise_error(ArgumentError, /first argument must be a positive integer/))
+  end
+
+  it "provides a random alphanumeric string" do
+    string = %r{\A[a-zA-Z0-9]*\z}
+    expect(fqdn_rand_string(100).match(string)).not_to eq(nil)
+  end
+
+  it "provides a random string exactly as long as the given length" do
+    expect(fqdn_rand_string(10).size).to eql(10)
+  end
+
+  it "provides the same 'random' value on subsequent calls for the same host" do
+    expect(fqdn_rand_string(10)).to eql(fqdn_rand_string(10))
+  end
+
+  it "considers the same host and same extra arguments to have the same random sequence" do
+    first_random = fqdn_rand_string(10, :extra_identifier => [1, "same", "host"])
+    second_random = fqdn_rand_string(10, :extra_identifier => [1, "same", "host"])
+
+    expect(first_random).to eql(second_random)
+  end
+
+  it "allows extra arguments to control the random value on a single host" do
+    first_random = fqdn_rand_string(10, :extra_identifier => [1, "different", "host"])
+    second_different_random = fqdn_rand_string(10, :extra_identifier => [2, "different", "host"])
+
+    expect(first_random).not_to eql(second_different_random)
+  end
+
+  it "should return different strings for different hosts" do
+    val1 = fqdn_rand_string(10, :host => "first.host.com")
+    val2 = fqdn_rand_string(10, :host => "second.host.com")
+
+    expect(val1).not_to eql(val2)
+  end
+
+  def fqdn_rand_string(max, args = {})
+    host = args[:host] || '127.0.0.1'
+    extra = args[:extra_identifier] || []
+
+    scope = PuppetlabsSpec::PuppetInternals.scope
+    scope.stubs(:[]).with("::fqdn").returns(host)
+    scope.stubs(:lookupvar).with("::fqdn").returns(host)
+
+    scope.function_fqdn_rand_string([max] + extra)
+  end
+end