def test_createDuplicate(self):
"""Create a duplicate user"""
userSpec = self._createUserSpec("test_createDuplicate", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
yield self.failUnlessFailure(UserManager.create(userSpec), UserAlreadyExists)
def test_createDuplicate(self):
"""Create a duplicate user"""
userSpec = self._createUserSpec("test_createDuplicate", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
yield self.failUnlessFailure(UserManager.create(userSpec),
UserAlreadyExists)
def test_get(self):
"""Get a user"""
userSpec = self._createUserSpec("test_get", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
deferred2 = UserManager.get("test_get")
yield deferred2
userObj = deferred2.result
yield self.assertEquals(userObj.userSpec.username, "test_get")
def test_delete(self):
"""Delete a user"""
userSpec = self._createUserSpec("test_delete", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
userId = deferred.result
deferred2 = UserManager.delete(userSpec.username)
deferred2.addCallback(self._verifyDeleted, userId)
yield deferred2
def test_delete(self):
"""Delete a user"""
userSpec = self._createUserSpec("test_delete", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
userId = deferred.result
deferred2 = UserManager.delete(userSpec.username)
deferred2.addCallback(self._verifyDeleted, userId)
yield deferred2
def test_get(self):
"""Get a user"""
userSpec = self._createUserSpec("test_get", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
deferred2 = UserManager.get("test_get")
yield deferred2
userObj = deferred2.result
yield self.assertEquals(userObj.userSpec.username, "test_get")
def test_getDeleted(self):
"""Get a deleted user"""
userSpec = self._createUserSpec("test_getDeleted", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
userId = deferred.result
deferred2 = UserManager.delete(userSpec.username)
deferred2.addCallback(self._verifyDeleted, userId)
yield deferred2
yield self.failUnlessFailure(UserManager.get(userSpec.username), NoSuchUser)
def test_getDeleted(self):
"""Get a deleted user"""
userSpec = self._createUserSpec("test_getDeleted", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
yield deferred
userId = deferred.result
deferred2 = UserManager.delete(userSpec.username)
deferred2.addCallback(self._verifyDeleted, userId)
yield deferred2
yield self.failUnlessFailure(UserManager.get(userSpec.username),
NoSuchUser)
def startServer(port):
logging.basicConfig(level=eval("logging.%s" %
config.parameter("log", "level")))
log = logging.getLogger("main")
twistedLog.startLogging(sys.stderr)
# add slaves in the INI file if they're around and add-able
slaves = {}
for (key, val) in config.section("slave"):
slaveNo = key[:key.find(".")]
attr = key[key.find(".") + 1:]
try:
setattr(slaves[slaveNo], attr, val)
except KeyError:
slaves[slaveNo] = SlaveSpec()
setattr(slaves[slaveNo], attr, val)
deferreds = []
for slaveSpec in slaves.itervalues():
log.info("Connecting slave: %s" % slaveSpec)
deferreds.append(Orchestrator.registerSlave(slaveSpec))
deferredList = DeferredList(deferreds)
yield deferredList
result = deferredList.result
if result is not None:
for (success, slaveId) in result:
if success == False:
log.error("Error adding slave")
else:
log.info("Slave %s connected" % slaveId)
log.debug("Listening on port %s", port)
reactor.listenTCP(port, createRestApi())
# XXX
userSpec = UserSpec()
userSpec.username = "******"
userSpec.password = "******"
yield UserManager.create(userSpec)
userSpec2 = UserSpec()
userSpec2.username = "******"
userSpec2.password = "******"
yield UserManager.create(userSpec2)
def startServer(port):
logging.basicConfig(level=eval("logging.%s" % config.parameter("log", "level")))
log = logging.getLogger("main")
twistedLog.startLogging(sys.stderr)
# add slaves in the INI file if they're around and add-able
slaves = {}
for (key, val) in config.section("slave"):
slaveNo = key[:key.find(".")]
attr = key[key.find(".")+1:]
try:
setattr(slaves[slaveNo], attr, val)
except KeyError:
slaves[slaveNo] = SlaveSpec()
setattr(slaves[slaveNo], attr, val)
deferreds = []
for slaveSpec in slaves.itervalues():
log.info("Connecting slave: %s" % slaveSpec)
deferreds.append(Orchestrator.registerSlave(slaveSpec))
deferredList = DeferredList(deferreds)
yield deferredList
result = deferredList.result
if result is not None:
for (success, slaveId) in result:
if success == False:
log.error("Error adding slave")
else:
log.info("Slave %s connected" % slaveId)
log.debug("Listening on port %s", port)
reactor.listenTCP(port, createRestApi())
# XXX
userSpec = UserSpec()
userSpec.username = "******"
userSpec.password = "******"
yield UserManager.create(userSpec)
userSpec2 = UserSpec()
userSpec2.username = "******"
userSpec2.password = "******"
yield UserManager.create(userSpec2)
def test_deleteInvalid(self):
"""Delete an invalid user"""
return self.failUnlessFailure(UserManager.delete("aasdfasdfasdfsadfsfd"), NoSuchUser)
def test_createUser(self):
"""Create a basic, no frills user"""
userSpec = self._createUserSpec("test_createUser", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
return deferred
def test_deleteInvalid(self):
"""Delete an invalid user"""
return self.failUnlessFailure(
UserManager.delete("aasdfasdfasdfsadfsfd"), NoSuchUser)
def test_createUser(self):
"""Create a basic, no frills user"""
userSpec = self._createUserSpec("test_createUser", "foo")
deferred = UserManager.create(userSpec)
deferred.addCallback(self._verifyUser, userSpec)
return deferred
