From 804d6fc6b9268577dbfd85f169f1a3f58e5cd6bc Mon Sep 17 00:00:00 2001 From: François Schmidts Date: Fri, 24 Apr 2015 19:05:14 +0200 Subject: adding base unittests --- pyaggr3g470r/models/__init__.py | 3 ++- pyaggr3g470r/models/article.py | 10 +++++----- 2 files changed, 7 insertions(+), 6 deletions(-) (limited to 'pyaggr3g470r/models') diff --git a/pyaggr3g470r/models/__init__.py b/pyaggr3g470r/models/__init__.py index 42903f4e..ba52b0de 100644 --- a/pyaggr3g470r/models/__init__.py +++ b/pyaggr3g470r/models/__init__.py @@ -98,4 +98,5 @@ def db_create(db): user1.roles.extend([role_admin, role_user]) db.session.add(user1) - db.session.commit() \ No newline at end of file + db.session.commit() + return role_admin, role_user diff --git a/pyaggr3g470r/models/article.py b/pyaggr3g470r/models/article.py index 58cd0384..3151071a 100644 --- a/pyaggr3g470r/models/article.py +++ b/pyaggr3g470r/models/article.py @@ -66,14 +66,14 @@ class Article(db.Model): .order_by(asc("Article.date")).first() def __repr__(self): - return json.dumps({ - "title": self.title, - "link": self.link, - "content": self.content - }) + return json.dumps({"title": self.title, + "link": self.link, + "content": self.content + }) def dump(self): return {"id": self.id, + "user_id": self.user_id, "title": self.title, "link": self.link, "content": self.content, -- cgit From b6603ce4607982590d7d55c0d583809192d84cad Mon Sep 17 00:00:00 2001 From: François Schmidts Date: Fri, 15 May 2015 00:21:40 +0200 Subject: testing special controller methods --- pyaggr3g470r/models/article.py | 1 + 1 file changed, 1 insertion(+) (limited to 'pyaggr3g470r/models') diff --git a/pyaggr3g470r/models/article.py b/pyaggr3g470r/models/article.py index 3151071a..16f15b9e 100644 --- a/pyaggr3g470r/models/article.py +++ b/pyaggr3g470r/models/article.py @@ -74,6 +74,7 @@ class Article(db.Model): def dump(self): return {"id": self.id, "user_id": self.user_id, + "entry_id": self.entry_id, "title": self.title, "link": self.link, "content": self.content, -- cgit