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'*****@*****.**'))