wutta/wuttjamaican
3
0
Fork 0
Code Activity

feat: add model for Person; tie to User

Browse source
This commit is contained in:
Lance Edgar 2024-07-14 15:47:39 -05:00
parent 60d3fcd13b
commit 43ca404837
6 changed files with 164 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/db/model/test_auth.py
View file

@ -5,6 +5,7 @@ from unittest import TestCase
try:
import sqlalchemy as sa
from wuttjamaican.db.model import auth as model
from wuttjamaican.db.model.base import Person
except ImportError:
pass
else:
@ -29,8 +30,16 @@ else:
class TestUser(TestCase):
def test_basic(self):
def test_str(self):
user = model.User()
self.assertEqual(str(user), "")
user.username = 'barney'
self.assertEqual(str(user), "barney")
def test_str_with_person(self):
user = model.User()
self.assertEqual(str(user), "")
person = Person(full_name="Barney Rubble")
user.person = person
self.assertEqual(str(user), "Barney Rubble")

17
tests/db/model/test_base.py
View file

@ -5,6 +5,7 @@ from unittest import TestCase
try:
import sqlalchemy as sa
from wuttjamaican.db.model import base as model
from wuttjamaican.db.model.auth import User
except ImportError:
pass
else:
@ -32,3 +33,19 @@ else:
self.assertEqual(str(setting), "")
setting.name = 'foo'
self.assertEqual(str(setting), "foo")
class TestPerson(TestCase):
def test_basic(self):
person = model.Person()
self.assertEqual(str(person), "")
person.full_name = "Barney Rubble"
self.assertEqual(str(person), "Barney Rubble")
def test_users(self):
person = model.Person()
self.assertIsNone(person.user)
user = User()
person.users.append(user)
self.assertIs(person.user, user)