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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
#! /usr/bin/env ruby
require 'spec_helper'
require 'puppet/indirector/ldap'
describe Puppet::Indirector::Ldap do
before do
@indirection = stub 'indirection', :name => :testing
Puppet::Indirector::Indirection.stubs(:instance).returns(@indirection)
module Testing; end
@ldap_class = class Testing::MyLdap < Puppet::Indirector::Ldap
self
end
@connection = mock 'ldap'
@searcher = @ldap_class.new
end
describe "when searching ldap" do
before do
# Stub everything, and we can selectively replace with an expect as
# we need to for testing.
@searcher.stubs(:connection).returns(@connection)
@searcher.stubs(:search_filter).returns(:filter)
@searcher.stubs(:search_base).returns(:base)
@searcher.stubs(:process)
@request = stub 'request', :key => "yay"
end
it "should call the ldapsearch method with the search filter" do
@searcher.expects(:search_filter).with("yay").returns("yay's filter")
@searcher.expects(:ldapsearch).with("yay's filter")
@searcher.find @request
end
it "should fail if no block is passed to the ldapsearch method" do
expect { @searcher.ldapsearch("blah") }.to raise_error(ArgumentError)
end
it "should use the results of the ldapbase method as the ldap search base" do
@searcher.stubs(:search_base).returns("mybase")
@connection.expects(:search).with do |*args|
expect(args[0]).to eq("mybase")
true
end
@searcher.find @request
end
it "should default to the value of the :search_base setting as the result of the ldapbase method" do
Puppet[:ldapbase] = "myldapbase"
searcher = @ldap_class.new
expect(searcher.search_base).to eq("myldapbase")
end
it "should use the results of the :search_attributes method as the list of attributes to return" do
@searcher.stubs(:search_attributes).returns(:myattrs)
@connection.expects(:search).with do |*args|
expect(args[3]).to eq(:myattrs)
true
end
@searcher.find @request
end
it "should use depth 2 when searching" do
@connection.expects(:search).with do |*args|
expect(args[1]).to eq(2)
true
end
@searcher.find @request
end
it "should call process() on the first found entry" do
@connection.expects(:search).yields("myresult")
@searcher.expects(:process).with("myresult")
@searcher.find @request
end
it "should reconnect and retry the search if there is a failure" do
run = false
@connection.stubs(:search).with do |*args|
if run
true
else
run = true
raise "failed"
end
end.yields("myresult")
@searcher.expects(:process).with("myresult")
@searcher.find @request
end
it "should not reconnect on failure more than once" do
count = 0
@connection.stubs(:search).with do |*args|
count += 1
raise ArgumentError, "yay"
end
expect { @searcher.find(@request) }.to raise_error(Puppet::Error)
expect(count).to eq(2)
end
it "should return true if an entry is found" do
@connection.expects(:search).yields("result")
expect(@searcher.ldapsearch("whatever") { |r| }).to be_truthy
end
end
describe "when connecting to ldap", :if => Puppet.features.ldap? do
it "should create and start a Util::Ldap::Connection instance" do
conn = double 'connection', :connection => "myconn", :start => nil
Puppet::Util::Ldap::Connection.expects(:instance).returns conn
expect(@searcher.connection).to eq("myconn")
end
it "should only create the ldap connection when asked for it the first time" do
conn = double 'connection', :connection => "myconn", :start => nil
Puppet::Util::Ldap::Connection.expects(:instance).returns conn
@searcher.connection
end
it "should cache the connection" do
conn = double 'connection', :connection => "myconn", :start => nil
Puppet::Util::Ldap::Connection.expects(:instance).returns conn
expect(@searcher.connection).to equal(@searcher.connection)
end
end
describe "when reconnecting to ldap", :if => (Puppet.features.root? and Facter.value("hostname") == "culain") do
it "should reconnect to ldap when connections are lost"
end
end
|
