Added test stub for User favorites packages

2 files changed, 50 insertions, 1 deletions

diff --git a/grumpy/testsuite/__init__.py b/grumpy/testsuite/__init__.py
index e9762db..92290d0 100644
--- a/grumpy/testsuite/__init__.py
+++ b/grumpy/testsuite/__init__.py
@@ -27,8 +27,9 @@ class GrumpyTestCase(unittest.TestCase):
         self.db.drop_all()
 
 def suite():
-    from . import pkgmodel, usermodel
+    from . import favorites, pkgmodel, usermodel
     suite = unittest.TestSuite()
+    suite.addTest(favorites.suite())
     suite.addTest(pkgmodel.suite())
     suite.addTest(usermodel.suite())
     return suite
diff --git a/grumpy/testsuite/favorites.py b/grumpy/testsuite/favorites.py
new file mode 100644
index 0000000..e919c11
--- /dev/null
+++ b/grumpy/testsuite/favorites.py
@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+"""
+    grumpy.testsuite.favorites
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    Unittests for "favorite" user packages unittests.
+
+    :copyright: (c) by 2010 Priit Laes.
+    :license: BSD, see LICENSE for details.
+"""
+from . import GrumpyTestCase
+
+from grumpy.models import Ebuild, Package, User
+
+import time, unittest
+
+class FavoritesTestCase(GrumpyTestCase):
+
+    def make_users(self):
+        self.db.session.add_all([\
+            User('user1@gentoo.org', 'http://example.net/openid1'),
+            User('user2@gentoo.org', 'http://example.net/openid2')])
+        self.db.session.commit()
+
+    def make_packages(self):
+        p1 = Package('sys-misc', 'testpkg', 'Test Package', \
+                    'Long Description', 'http://example.com/test', \
+                    time.time())
+        p1.ebuilds.append(Ebuild(p1, '1.0', 2, 0, 'x86,amd64', 'doc', ''))
+        p1.ebuilds.append(Ebuild(p1, '1.1', 2, 0, '~x86,~amd64', 'doc', ''))
+        p2 = Package('sys-devel', 'testpkg', 'Test Package 2', \
+                    'Long Description 2', 'http://example.com/test2', \
+                    time.time())
+        p2.ebuilds.append(Ebuild(p1, '0.1', 2, 0, 'x86,amd64', 'doc', ''))
+        self.db.session.add_all([p1, p2])
+        self.db.session.commit()
+
+    def test_favorites(self):
+        with self.app.test_request_context():
+            self.make_users()
+            self.make_packages()
+            assert User.query.count() == 2
+            assert Package.query.count() == 2
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(FavoritesTestCase))
+    return suite