aboutsummaryrefslogtreecommitdiff
path: root/tests/fixtures.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/fixtures.py')
-rw-r--r--tests/fixtures.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/tests/fixtures.py b/tests/fixtures.py
new file mode 100644
index 00000000..6aab645d
--- /dev/null
+++ b/tests/fixtures.py
@@ -0,0 +1,31 @@
+from pyaggr3g470r.models import db_create, db_empty, User, Article, Feed
+
+
+def populate_db(db):
+ role_admin, role_user = db_create(db)
+ user1, user2 = [User(nickname=name, email="%s@test.te" % name,
+ pwdhash=name, roles=[role_user], activation_key="")
+ for name in ["user1", "user2"]]
+ db.session.add(user1)
+ db.session.add(user2)
+ db.session.commit()
+
+ for user in (user1, user2):
+ for feed_name in ['feed1', 'feed2', 'feed3']:
+ feed = Feed(link=feed_name, user_id=user.id,
+ title="%r %r" % (user.nickname, feed_name))
+ db.session.add(feed)
+ db.session.commit()
+ for article in ['article1', 'article2', 'article3']:
+ entry = "%s %s %s" % (user.nickname, feed.title, article)
+ article = Article(entry_id=entry, link=article,
+ feed_id=feed.id, user_id=user.id,
+ title=entry, content=article)
+ db.session.add(article)
+ db.session.commit()
+
+ db.session.commit()
+
+
+def reset_db(db):
+ db_empty(db)
bgstack15