3
0
Fork 0

fix: add make_person() method for people, auth handlers

This commit is contained in:
Lance Edgar 2025-01-27 15:48:28 -06:00
parent 06524dc4ed
commit 57ff7ec1d3
4 changed files with 75 additions and 1 deletions

View file

@ -17,7 +17,7 @@ else:
def setUp(self):
self.config = WuttaConfig()
self.app = self.config.get_app()
self.handler = mod.AuthHandler(self.config)
self.handler = self.make_handler()
self.engine = sa.create_engine('sqlite://')
self.app.model.Base.metadata.create_all(bind=self.engine)
@ -30,6 +30,9 @@ else:
def make_session(self):
return self.app.make_session(bind=self.engine)
def make_handler(self):
return mod.AuthHandler(self.config)
def test_authenticate_user(self):
model = self.app.model
barney = model.User(username='barney')
@ -133,6 +136,24 @@ else:
user = self.handler.get_user(myperson)
self.assertIs(user, myuser)
def test_make_person(self):
model = self.app.model
handler = self.handler
person = handler.make_person()
self.assertIsInstance(person, model.Person)
self.assertIsNone(person.first_name)
self.assertIsNone(person.last_name)
self.assertIsNone(person.full_name)
self.assertNotIn(person, self.session)
person = handler.make_person(first_name='Barney', last_name='Rubble')
self.assertIsInstance(person, model.Person)
self.assertEqual(person.first_name, 'Barney')
self.assertEqual(person.last_name, 'Rubble')
self.assertEqual(person.full_name, 'Barney Rubble')
self.assertNotIn(person, self.session)
def test_make_user(self):
model = self.app.model

View file

@ -36,3 +36,21 @@ else:
self.session.commit()
person = handler.get_person(myuser)
self.assertIs(person, myperson)
def test_make_person(self):
model = self.app.model
handler = self.make_handler()
person = handler.make_person()
self.assertIsInstance(person, model.Person)
self.assertIsNone(person.first_name)
self.assertIsNone(person.last_name)
self.assertIsNone(person.full_name)
self.assertNotIn(person, self.session)
person = handler.make_person(first_name='Barney', last_name='Rubble')
self.assertIsInstance(person, model.Person)
self.assertEqual(person.first_name, 'Barney')
self.assertEqual(person.last_name, 'Rubble')
self.assertEqual(person.full_name, 'Barney Rubble')
self.assertNotIn(person, self.session)