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