Commit 507bd813 authored by Rémy Coutable's avatar Rémy Coutable

Reduce differences in spec/support/helpers/ldap_helpers.rb

Signed-off-by: default avatarRémy Coutable <remy@rymai.me>
parent fbfe1154
...@@ -4,6 +4,12 @@ module EE ...@@ -4,6 +4,12 @@ module EE
EE::Gitlab::Auth::LDAP::Sync::Proxy.new(provider, adapter) EE::Gitlab::Auth::LDAP::Sync::Proxy.new(provider, adapter)
end end
def fake_ldap_sync_proxy(provider)
fake_proxy = double(:proxy, adapter: ldap_adapter)
allow(::EE::Gitlab::Auth::LDAP::Sync::Proxy).to receive(:open).with(provider).and_yield(fake_proxy)
fake_proxy
end
# Stub an LDAP group search and provide the return entry. Specify `nil` for # Stub an LDAP group search and provide the return entry. Specify `nil` for
# `entry` to simulate when an LDAP group is not found # `entry` to simulate when an LDAP group is not found
# #
......
module LdapHelpers module LdapHelpers
include EE::LdapHelpers
def ldap_adapter(provider = 'ldapmain', ldap = double(:ldap)) def ldap_adapter(provider = 'ldapmain', ldap = double(:ldap))
::Gitlab::Auth::LDAP::Adapter.new(provider, ldap) ::Gitlab::Auth::LDAP::Adapter.new(provider, ldap)
end end
def fake_ldap_sync_proxy(provider)
fake_proxy = double(:proxy, adapter: ldap_adapter)
allow(::EE::Gitlab::Auth::LDAP::Sync::Proxy).to receive(:open).with(provider).and_yield(fake_proxy)
fake_proxy
end
def user_dn(uid) def user_dn(uid)
"uid=#{uid},ou=users,dc=example,dc=com" "uid=#{uid},ou=users,dc=example,dc=com"
end end
...@@ -76,3 +68,5 @@ module LdapHelpers ...@@ -76,3 +68,5 @@ module LdapHelpers
.to receive(:ldap_search).and_raise(Gitlab::Auth::LDAP::LDAPConnectionError) .to receive(:ldap_search).and_raise(Gitlab::Auth::LDAP::LDAPConnectionError)
end end
end end
LdapHelpers.include(EE::LdapHelpers)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment