def createServer(self, *responses, **kwds):
"""
Create a server for each test.
"""
protocol = kwds.get("protocol", proxybase.ProxyBase)
clock = Clock()
clock = kwds.get('clock', clock)
server = protocol()
clientTestDriver = testutil.LDAPClientTestDriver(*responses)
def simulateConnectToServer():
d = defer.Deferred()
def onConnect():
clientTestDriver.connectionMade()
d.callback(clientTestDriver)
clock.callLater(0, onConnect)
return d
clientConnector = kwds.get('clientConnector', simulateConnectToServer)
server.clientConnector = clientConnector
server.clientTestDriver = clientTestDriver
server.transport = proto_helpers.StringTransport()
server.reactor = clock
server.connectionMade()
return server
def createMergedServer(self, *responses):
"""
Create an MergedLDAP server for testing. Initialize with
len(responses) clients.
:param responses: The responses to initialize the `LDAPClientTestDrives`.
:type responses: args of lists of lists
:return a deferred, fires when server finished connecting
"""
def createClient(factory):
factory.doStart()
proto = factory.buildProtocol(addr=None)
proto.connectionMade()
clients = []
for r in responses:
clients.append(testutil.LDAPClientTestDriver(*r))
conf = config.LDAPConfig(serviceLocationOverrides={"": createClient})
server = MergedLDAPServer([conf for _ in clients],
[False for _ in clients])
self.clients = clients * 1
server.protocol = lambda: clients.pop()
server.transport = proto_helpers.StringTransport()
server.connectionMade()
d = server._whenConnected(lambda: server)
return d
def testPrimaryGroupId(self):
"""Test that primaryGroupID field is updated based on gidNumber."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(client=client,
dn='cn=foo,dc=example,dc=com',
attributes={
'objectClass':
['sambaAccount', 'other'],
})
d = o.addAutofiller(sambaAccount.Autofill_samba())
def cb(dummy):
client.assertNothingSent()
o['gidNumber'] = ['1000']
self.failUnless('primaryGroupID' in o)
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 1000 + 1001)])
o['gidNumber'] = ['1001']
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 1001 + 1001)])
o['gidNumber'] = ['1002']
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 1002 + 1001)])
o['gidNumber'] = ['2000']
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 2000 + 1001)])
o['gidNumber'] = ['3000']
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 3000 + 1001)])
o['gidNumber'] = ['0']
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 0 + 1001)])
o['gidNumber'] = ['16000']
self.failUnlessEqual(o['primaryGroupID'], [str(2 * 16000 + 1001)])
d.addCallback(cb)
return d
def testDefaultSetting(self):
"""Test that fields get their default values."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(client=client,
dn='cn=foo,dc=example,dc=com',
attributes={
'objectClass':
['sambaAccount', 'other'],
})
d = o.addAutofiller(sambaAccount.Autofill_samba())
def cb(dummy):
client.assertNothingSent()
self.failUnless('acctFlags' in o)
self.failUnlessEqual(o['acctFlags'], ['[UX ]'])
self.failUnless('pwdLastSet' in o)
self.failUnlessEqual(o['pwdLastSet'], ['0'])
self.failUnless('logonTime' in o)
self.failUnlessEqual(o['logonTime'], ['0'])
self.failUnless('logoffTime' in o)
self.failUnlessEqual(o['logoffTime'], ['0'])
self.failUnless('pwdCanChange' in o)
self.failUnlessEqual(o['pwdCanChange'], ['0'])
self.failUnless('pwdMustChange' in o)
self.failUnlessEqual(o['pwdMustChange'], ['0'])
d.addCallback(cb)
return d
def testPrimaryGroupId(self):
"""Test that primaryGroupID field is updated based on gidNumber."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(
client=client,
dn="cn=foo,dc=example,dc=com",
attributes={
"objectClass": ["sambaAccount", "other"],
},
)
d = o.addAutofiller(sambaAccount.Autofill_samba())
def cb(dummy):
client.assertNothingSent()
o["gidNumber"] = ["1000"]
self.failUnless("primaryGroupID" in o)
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 1000 + 1001)])
o["gidNumber"] = ["1001"]
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 1001 + 1001)])
o["gidNumber"] = ["1002"]
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 1002 + 1001)])
o["gidNumber"] = ["2000"]
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 2000 + 1001)])
o["gidNumber"] = ["3000"]
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 3000 + 1001)])
o["gidNumber"] = ["0"]
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 0 + 1001)])
o["gidNumber"] = ["16000"]
self.failUnlessEqual(o["primaryGroupID"], [str(2 * 16000 + 1001)])
d.addCallback(cb)
return d
def testSambaPrimaryGroupSID_notUpdatedWhenFixed(self):
"""Test that sambaPrimaryGroupSID field is updated based on gidNumber."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(
client=client,
dn="cn=foo,dc=example,dc=com",
attributes={
"objectClass": ["sambaSamAccount", "other"],
},
)
d = o.addAutofiller(
sambaSamAccount.Autofill_samba(domainSID="foo", fixedPrimaryGroupSID=4242)
)
def cb(dummy):
client.assertNothingSent()
self.failUnless("sambaPrimaryGroupSID" in o)
self.failUnlessEqual(o["sambaPrimaryGroupSID"], ["foo-4242"])
o["gidNumber"] = ["1000"]
self.failUnless("sambaPrimaryGroupSID" in o)
self.failUnlessEqual(o["sambaPrimaryGroupSID"], ["foo-4242"])
d.addCallback(cb)
return d
def testDefaultSetting(self):
"""Test that fields get their default values."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(
client=client,
dn="cn=foo,dc=example,dc=com",
attributes={
"objectClass": ["sambaAccount", "other"],
},
)
d = o.addAutofiller(sambaAccount.Autofill_samba())
def cb(dummy):
client.assertNothingSent()
self.failUnless("acctFlags" in o)
self.failUnlessEqual(o["acctFlags"], ["[UX ]"])
self.failUnless("pwdLastSet" in o)
self.failUnlessEqual(o["pwdLastSet"], ["0"])
self.failUnless("logonTime" in o)
self.failUnlessEqual(o["logonTime"], ["0"])
self.failUnless("logoffTime" in o)
self.failUnlessEqual(o["logoffTime"], ["0"])
self.failUnless("pwdCanChange" in o)
self.failUnlessEqual(o["pwdCanChange"], ["0"])
self.failUnless("pwdMustChange" in o)
self.failUnlessEqual(o["pwdMustChange"], ["0"])
d.addCallback(cb)
return d
def testMustHaveObjectClass(self):
"""Test that Autofill_samba fails unless object is a sambaSamAccount."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(client=client,
dn='cn=foo,dc=example,dc=com',
attributes={
'objectClass':
['something', 'other'],
})
autoFiller = sambaSamAccount.Autofill_samba(domainSID='foo')
d = o.addAutofiller(autoFiller)
def eb(val):
client.assertNothingSent()
val.trap(sambaSamAccount.ObjectMissingObjectClassException)
d.addCallbacks(testutil.mustRaise, eb)
return d
def testDefaultSetting_fixedPrimaryGroupSID(self):
"""Test that fields get their default values."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(
client=client,
dn="cn=foo,dc=example,dc=com",
attributes={
"objectClass": ["sambaSamAccount", "other"],
},
)
d = o.addAutofiller(
sambaSamAccount.Autofill_samba(
domainSID="foo", fixedPrimaryGroupSID=4131312
)
)
def cb(dummy):
client.assertNothingSent()
self.failUnlessEqual(
set(o.keys()),
{
"objectClass",
"sambaAcctFlags",
"sambaLogoffTime",
"sambaLogonTime",
"sambaPwdCanChange",
"sambaPwdLastSet",
"sambaPwdMustChange",
"sambaPrimaryGroupSID",
},
)
self.failUnlessEqual(o["sambaPrimaryGroupSID"], ["foo-4131312"])
self.failUnlessEqual(o["sambaAcctFlags"], ["[UX ]"])
self.failUnlessEqual(o["sambaPwdLastSet"], ["1"])
self.failUnlessEqual(o["sambaLogonTime"], ["0"])
self.failUnlessEqual(o["sambaLogoffTime"], ["0"])
self.failUnlessEqual(o["sambaPwdCanChange"], ["0"])
self.failUnlessEqual(o["sambaPwdMustChange"], ["0"])
d.addCallback(cb)
return d
def testMustHaveObjectClass(self):
"""Test that Autofill_samba fails unless object is a sambaAccount."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(
client=client,
dn="cn=foo,dc=example,dc=com",
attributes={
"objectClass": ["something", "other"],
},
)
autoFiller = sambaAccount.Autofill_samba()
d = o.addAutofiller(autoFiller)
def eb(val):
client.assertNothingSent()
val.trap(sambaAccount.ObjectMissingObjectClassException)
d.addCallbacks(testutil.mustRaise, eb)
return d
def testSambaSID_preExisting(self):
"""Test that sambaSID field is updated based on uidNumber."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(client=client,
dn='cn=foo,dc=example,dc=com',
attributes={
'objectClass':
['sambaSamAccount', 'other'],
'uidNumber': ['1000'],
})
d = o.addAutofiller(sambaSamAccount.Autofill_samba(domainSID='foo'))
def cb(dummy):
client.assertNothingSent()
self.failUnless('sambaSID' in o)
self.failUnlessEqual(o['sambaSID'], ['foo-%s' % (2 * 1000 + 1000)])
d.addCallback(cb)
return d
def testSambaSID_preExisting(self):
"""Test that sambaSID field is updated based on uidNumber."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(
client=client,
dn="cn=foo,dc=example,dc=com",
attributes={
"objectClass": ["sambaSamAccount", "other"],
"uidNumber": ["1000"],
},
)
d = o.addAutofiller(sambaSamAccount.Autofill_samba(domainSID="foo"))
def cb(dummy):
client.assertNothingSent()
self.failUnless("sambaSID" in o)
self.failUnlessEqual(o["sambaSID"], ["foo-%s" % (2 * 1000 + 1000)])
d.addCallback(cb)
return d
def testDefaultSetting_fixedPrimaryGroupSID(self):
"""Test that fields get their default values."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(client=client,
dn='cn=foo,dc=example,dc=com',
attributes={
'objectClass':
['sambaSamAccount', 'other'],
})
d = o.addAutofiller(
sambaSamAccount.Autofill_samba(domainSID='foo',
fixedPrimaryGroupSID=4131312))
def cb(dummy):
client.assertNothingSent()
self.failUnlessEqual(
sets.Set(o.keys()),
sets.Set([
'objectClass',
'sambaAcctFlags',
'sambaLogoffTime',
'sambaLogonTime',
'sambaPwdCanChange',
'sambaPwdLastSet',
'sambaPwdMustChange',
'sambaPrimaryGroupSID',
]))
self.failUnlessEqual(o['sambaPrimaryGroupSID'], ['foo-4131312'])
self.failUnlessEqual(o['sambaAcctFlags'], ['[UX ]'])
self.failUnlessEqual(o['sambaPwdLastSet'], ['1'])
self.failUnlessEqual(o['sambaLogonTime'], ['0'])
self.failUnlessEqual(o['sambaLogoffTime'], ['0'])
self.failUnlessEqual(o['sambaPwdCanChange'], ['0'])
self.failUnlessEqual(o['sambaPwdMustChange'], ['0'])
d.addCallback(cb)
return d
def testSambaPrimaryGroupSID_notUpdatedWhenFixed(self):
"""Test that sambaPrimaryGroupSID field is updated based on gidNumber."""
client = testutil.LDAPClientTestDriver()
o = ldapsyntax.LDAPEntryWithAutoFill(client=client,
dn='cn=foo,dc=example,dc=com',
attributes={
'objectClass':
['sambaSamAccount', 'other'],
})
d = o.addAutofiller(
sambaSamAccount.Autofill_samba(domainSID='foo',
fixedPrimaryGroupSID=4242))
def cb(dummy):
client.assertNothingSent()
self.failUnless('sambaPrimaryGroupSID' in o)
self.failUnlessEqual(o['sambaPrimaryGroupSID'], ['foo-4242'])
o['gidNumber'] = ['1000']
self.failUnless('sambaPrimaryGroupSID' in o)
self.failUnlessEqual(o['sambaPrimaryGroupSID'], ['foo-4242'])
d.addCallback(cb)
return d
