1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
require 'spec_helper'
describe 'fqdn_rotate' do
it { is_expected.not_to eq(nil) }
it { is_expected.to run.with_params().and_raise_error(Puppet::ParseError, /wrong number of arguments/i) }
it { is_expected.to run.with_params(0).and_raise_error(Puppet::ParseError, /Requires either array or string to work with/) }
it { is_expected.to run.with_params({}).and_raise_error(Puppet::ParseError, /Requires either array or string to work with/) }
it { is_expected.to run.with_params('').and_return('') }
it { is_expected.to run.with_params('a').and_return('a') }
it { is_expected.to run.with_params([]).and_return([]) }
it { is_expected.to run.with_params(['a']).and_return(['a']) }
it "should rotate a string and the result should be the same size" do
expect(fqdn_rotate("asdf").size).to eq(4)
end
it "should rotate a string to give the same results for one host" do
val1 = fqdn_rotate("abcdefg", :host => 'one')
val2 = fqdn_rotate("abcdefg", :host => 'one')
expect(val1).to eq(val2)
end
it "allows extra arguments to control the random rotation on a single host" do
val1 = fqdn_rotate("abcdefg", :extra_identifier => [1, "different", "host"])
val2 = fqdn_rotate("abcdefg", :extra_identifier => [2, "different", "host"])
expect(val1).not_to eq(val2)
end
it "considers the same host and same extra arguments to have the same random rotation" do
val1 = fqdn_rotate("abcdefg", :extra_identifier => [1, "same", "host"])
val2 = fqdn_rotate("abcdefg", :extra_identifier => [1, "same", "host"])
expect(val1).to eq(val2)
end
it "should rotate a string to give different values on different hosts" do
val1 = fqdn_rotate("abcdefg", :host => 'one')
val2 = fqdn_rotate("abcdefg", :host => 'two')
expect(val1).not_to eq(val2)
end
it "should accept objects which extend String" do
result = fqdn_rotate(AlsoString.new('asdf'))
expect(result).to eq('dfas')
end
it "should use the Puppet::Util.deterministic_rand function" do
if Puppet::Util.respond_to?(:deterministic_rand)
Puppet::Util.expects(:deterministic_rand).with(44489829212339698569024999901561968770,4)
fqdn_rotate("asdf")
else
skip 'Puppet::Util#deterministic_rand not available'
end
end
it "should not leave the global seed in a deterministic state" do
fqdn_rotate("asdf")
rand1 = rand()
fqdn_rotate("asdf")
rand2 = rand()
expect(rand1).not_to eql(rand2)
end
def fqdn_rotate(value, args = {})
host = args[:host] || '127.0.0.1'
extra = args[:extra_identifier] || []
# workaround not being able to use let(:facts) because some tests need
# multiple different hostnames in one context
scope.stubs(:lookupvar).with("::fqdn").returns(host)
function_args = [value] + extra
scope.function_fqdn_rotate(function_args)
end
end
|
