diff options
author | Priit Laes <plaes@plaes.org> | 2010-07-27 11:08:19 +0300 |
---|---|---|
committer | Priit Laes <plaes@plaes.org> | 2010-07-27 11:08:19 +0300 |
commit | d73bce2c75c6be7a289669b674624db152266969 (patch) | |
tree | 50aa7e54f5d5cbc66eb05c72d582bc1d1f39a2f6 | |
parent | Added simple unittest for Ebuilds and Packages (diff) | |
download | gsoc2010-grumpy-d73bce2c75c6be7a289669b674624db152266969.tar.gz gsoc2010-grumpy-d73bce2c75c6be7a289669b674624db152266969.tar.bz2 gsoc2010-grumpy-d73bce2c75c6be7a289669b674624db152266969.zip |
Added test stub for User favorites packages
-rw-r--r-- | grumpy/testsuite/__init__.py | 3 | ||||
-rw-r--r-- | grumpy/testsuite/favorites.py | 48 |
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 |