|
@@ -2,20 +2,21 @@ import lc.config as c
|
|
|
import lc.model as m
|
|
|
import tempfile
|
|
|
|
|
|
+
|
|
|
class TestDB:
|
|
|
def setup_method(self, _):
|
|
|
- c.DB.init(':memory:')
|
|
|
+ c.DB.init(":memory:")
|
|
|
c.DB.create_tables(m.MODELS)
|
|
|
|
|
|
def test_create_user(self):
|
|
|
- name = 'gdritter'
|
|
|
+ name = "gdritter"
|
|
|
u = m.User.create(name=name)
|
|
|
|
|
|
# it should be the only thing in the db
|
|
|
all_users = m.User.select()
|
|
|
assert len(all_users) == 1
|
|
|
- assert(all_users[0].id == u.id)
|
|
|
- assert(all_users[0].name == name)
|
|
|
+ assert all_users[0].id == u.id
|
|
|
+ assert all_users[0].name == name
|
|
|
|
|
|
# we should be able to find it with the given name, too
|
|
|
named_user = m.User.get(m.User.name == name)
|
|
@@ -23,8 +24,8 @@ class TestDB:
|
|
|
assert named_user.name == name
|
|
|
|
|
|
def test_find_tag(self):
|
|
|
- tag_name = 'food'
|
|
|
- u = m.User.create(name='gdritter')
|
|
|
+ tag_name = "food"
|
|
|
+ u = m.User.create(name="gdritter")
|
|
|
t = m.Tag.find_tag(u, tag_name)
|
|
|
|
|
|
# we should be able to find the tag with the given name
|
|
@@ -36,8 +37,8 @@ class TestDB:
|
|
|
assert t.id == t2.id
|
|
|
|
|
|
def test_find_hierarchy(self):
|
|
|
- u = m.User.create(name='gdritter')
|
|
|
- t = m.Tag.find_tag(u, 'food/bread/rye')
|
|
|
+ u = m.User.create(name="gdritter")
|
|
|
+ t = m.Tag.find_tag(u, "food/bread/rye")
|
|
|
|
|
|
# this should have created three DB rows: for 'food', for
|
|
|
# 'food/bread', and for 'food/bread/rye':
|
|
@@ -45,12 +46,12 @@ class TestDB:
|
|
|
|
|
|
# searching for a prefix of the tag should yield the same
|
|
|
# parent tag
|
|
|
- assert t.parent.id == m.Tag.get(name='food/bread').id
|
|
|
- assert t.parent.parent.id == m.Tag.get(name='food').id
|
|
|
+ assert t.parent.id == m.Tag.get(name="food/bread").id
|
|
|
+ assert t.parent.parent.id == m.Tag.get(name="food").id
|
|
|
|
|
|
# creating a new hierarchical tag with a shared prefix should
|
|
|
# only create the new child tag
|
|
|
- t2 = m.Tag.find_tag(u, 'food/bread/baguette')
|
|
|
+ t2 = m.Tag.find_tag(u, "food/bread/baguette")
|
|
|
print([t.name for t in m.Tag.select()])
|
|
|
|
|
|
assert len(m.Tag.select()) == 4
|
|
@@ -60,9 +61,8 @@ class TestDB:
|
|
|
|
|
|
# trying to get a hierarchical tag should result in the same
|
|
|
# one already entered
|
|
|
- assert t.id == m.Tag.get(name='food/bread/rye').id
|
|
|
- assert t2.id == m.Tag.get(name='food/bread/baguette').id
|
|
|
-
|
|
|
+ assert t.id == m.Tag.get(name="food/bread/rye").id
|
|
|
+ assert t2.id == m.Tag.get(name="food/bread/baguette").id
|
|
|
|
|
|
def teardown_method(self, _):
|
|
|
c.DB.close()
|