def test_by_id(self):
init_user_db()
self.session = user.DBSession()
u = user.User(u'bob', u'Bob Smith', u'*****@*****.**')
self.session.add(u)
u2 = user.User.by_id(u'bob')
self.assertEqual(u, u2)
destroy_user_db()
def test_construct_with_password(self, salt, hashpw):
salt.return_value = 'salty'
u = user.User('bob', 'Bob Smith', '*****@*****.**', 'mypassword')
self.assertEqual(u.userid, 'bob')
self.assertEqual(u.displayname, 'Bob Smith')
self.assertEqual(u.email, '*****@*****.**')
salt.assert_called_with(12)
hashpw.assert_called_with('mypassword', 'salty')
def test_add(self):
init_user_db()
u = user.User(u'bob', u'Bob Smith', u'*****@*****.**')
user.User.add(u)
session = user.DBSession()
u2 = session.query(user.User).get('bob')
self.assertEqual(u, u2)
destroy_user_db()
def test_delete(self):
init_user_db()
u = user.User(u'bob', u'Bob Smith', u'*****@*****.**')
user.User.add(u)
user.User.delete(u)
session = user.DBSession()
u2 = session.query(user.User).get(u'bob')
self.assertIsNone(u2)
destroy_user_db()
def test_generate(self):
init_user_db()
u = user.User.generate()
eu = user.User(u.userid, 'Temporary user', '*****@*****.**')
self.assertEqual(u, eu)
session = user.DBSession()
u2 = session.query(user.User).get(u.userid)
self.assertEqual(u, u2)
destroy_user_db()
def test_jobs(self):
init_user_db()
session = user.DBSession()
u = user.User(u'bob', u'Bob Smith', u'*****@*****.**')
session.add(u)
job_id = uuid.UUID('11111111-1111-1111-1111-111111111111')
j = user.JobMeta(job_id, u'bob')
session.add(j)
u2 = user.User.by_id(u'bob') # force commit
self.assertEqual(u2.jobs, [j])
destroy_user_db()
def setUp(self):
import tempfile
self.root_dir = tempfile.mkdtemp()
self.factory = JobFactory(root_dir=self.root_dir,
submit_url='http://localhost:9998/job',
)
# fill user db
transaction.begin()
engine = create_engine('sqlite:///:memory:')
mu.DBSession.configure(bind=engine)
mu.Base.metadata.create_all(engine) # @UndefinedVariable
mu.DBSession().add(mu.User(u'bob', u'Bob', u'*****@*****.**'))
jobid = uuid.UUID('3ad25048-26f6-11e1-851e-00012e260790')
mu.DBSession().add(mu.JobMeta(jobid, owner=u'bob'))
def setUp(self):
self.config = testing.setUp()
self.config.add_static_view('static', 'magmaweb:static')
self.request = testing.DummyRequest()
self.request.registry.settings = {
'extjsroot': 'extjsroot',
'jobfactory.root_dir': '/somedir',
'monitor_user': '******',
}
init_user_db()
self.session = user.DBSession()
u = user.User(u'me', u'My', u'myself')
self.session.add(u)
job_id = uuid.UUID('11111111-1111-1111-1111-111111111111')
j = user.JobMeta(job_id, u'me', description=u'My job')
self.session.add(j)
def test_repr(self):
u = user.User('bob', 'Bob Smith', '*****@*****.**', 'mypassword')
self.assertEqual(repr(u),
"<User('bob', 'Bob Smith', '*****@*****.**')>")
def test_validate_password_incorrect(self):
u = user.User('bob', 'Bob Smith', '*****@*****.**', 'mypassword')
self.assertFalse(u.validate_password('otherpassword'))
def test_validate_password_correct(self):
u = user.User('bob', 'Bob Smith', '*****@*****.**', 'mypassword')
self.assertTrue(u.validate_password('mypassword'))
def test_construct(self):
u = user.User('bob', 'Bob Smith', '*****@*****.**')
self.assertEqual(u.userid, 'bob')
self.assertEqual(u.displayname, 'Bob Smith')
self.assertEqual(u.email, '*****@*****.**')
def setUp(self):
init_user_db()
self.session = user.DBSession()
# job must be owned by a user
self.session.add(user.User(u'bob', u'Bob Smith', u'*****@*****.**'))