Python make_flask_test_suite示例

示例#1

        for i in c.collection_children:
            p(i, level + 1)


class WebSearchCollectionTest(FlaskSQLAlchemyTest):
    @fixture.with_data(ExternalcollectionData, CollectionData,
                       CollectionCollectionData)
    def test_loading_collection_tree(data, self):
        from invenio.modules.search.models import Collection
        print
        p(Collection.query.order_by(Collection.id).first())
        print

    @fixture.with_data(ExternalcollectionData, CollectionData)
    def test_external_collection(data, self):
        from invenio.modules.search.models import Collection
        print
        print '----# EXTERNAL COLLECTIONS ----'
        for c in Collection.query.all():
            print c.name, ': ',
            print len([a for a in c._externalcollections])
        print '-------------------------------'
        print 'Total: ', sum(
            map(len, [c._externalcollections for c in Collection.query.all()]))


TEST_SUITE = make_flask_test_suite(WebSearchCollectionTest)

if __name__ == "__main__":
    run_test_suite(TEST_SUITE)

示例#2

文件： test_accounts.py 项目： PXke/invenio-demosite

        new_passwd = db.session.query(User.password).filter(User.id == users.admin.id).one()
        assert users.admin.password != new_passwd

        # Invalid password for admin.
        response = self.login(users.admin.nickname, users.admin.password)
        assert 'logout' not in response.data

        # Valid credentials.
        response = self.login(users.admin.nickname, NEW_PASSWORD)
        assert users.admin.nickname in response.data
        self.logout()


class UserGroupTest(FlaskSQLAlchemyTest):

    @fixture.with_data(UserData, UsergroupData, UserUsergroupData)
    def test_group_relation_consistency(data, self):
        from invenio.modules.accounts.models import User, Usergroup
        orig_len = len(dict(data.UserUsergroupData))
        user_len = sum(len(u.usergroups) for u in User.query.all())
        ugrp_len = sum(len(g.users) for g in Usergroup.query.all())

        assert orig_len == user_len
        assert user_len == ugrp_len


TEST_SUITE = make_flask_test_suite(WebAccountTest, UserGroupTest)

if __name__ == "__main__":
    run_test_suite(TEST_SUITE)

示例#3

文件： test_search.py 项目： mvesper/invenio-demosite

        for i in c.collection_children:
            p(i, level + 1)


class WebSearchCollectionTest(FlaskSQLAlchemyTest):
    @fixture.with_data(ExternalcollectionData, CollectionData, CollectionCollectionData)
    def test_loading_collection_tree(data, self):
        from invenio.modules.search.models import Collection

        print
        p(Collection.query.order_by(Collection.id).first())
        print

    @fixture.with_data(ExternalcollectionData, CollectionData)
    def test_external_collection(data, self):
        from invenio.modules.search.models import Collection

        print
        print "----# EXTERNAL COLLECTIONS ----"
        for c in Collection.query.all():
            print c.name, ": ",
            print len([a for a in c._externalcollections])
        print "-------------------------------"
        print "Total: ", sum(map(len, [c._externalcollections for c in Collection.query.all()]))


TEST_SUITE = make_flask_test_suite(WebSearchCollectionTest)

if __name__ == "__main__":
    run_test_suite(TEST_SUITE)

示例#4

文件： test_messages.py 项目： hachreak/invenio-demosite

        time.sleep(1)

        response = self.client.get("/yourmessages/", follow_redirects=True)
        assert 'sent_now' in response.data
        assert 'sent_later' in response.data

        # Avoid problems with foreign keys.
        UserMsgMESSAGE.query.delete(synchronize_session='fetch')
        MsgMESSAGE.query.delete(synchronize_session='fetch')
        db.session.commit()

    @fixture.with_data(UserData)
    def test_with_fixture(data, self):
        from invenio_accounts.models import User
        u = User.query.all()
        assert len(u) == len(dict(data.UserData))
        db.session.commit()

    def test_without_fixture(self):
        from invenio_accounts.models import User
        u = User.query.all()
        assert len(u) == 0


TEST_SUITE = make_flask_test_suite(MsgMESSAGETest)

if __name__ == "__main__":
    run_test_suite(TEST_SUITE)

示例#5

        new_passwd = db.session.query(
            User.password).filter(User.id == users.admin.id).one()
        assert users.admin.password != new_passwd

        # Invalid password for admin.
        response = self.login(users.admin.nickname, users.admin.password)
        assert 'logout' not in response.data

        # Valid credentials.
        response = self.login(users.admin.nickname, NEW_PASSWORD)
        assert users.admin.nickname in response.data
        self.logout()


class UserGroupTest(FlaskSQLAlchemyTest):
    @fixture.with_data(UserData, UsergroupData, UserUsergroupData)
    def test_group_relation_consistency(data, self):
        from invenio.modules.accounts.models import User, Usergroup
        orig_len = len(dict(data.UserUsergroupData))
        user_len = sum(len(u.usergroups) for u in User.query.all())
        ugrp_len = sum(len(g.users) for g in Usergroup.query.all())

        assert orig_len == user_len
        assert user_len == ugrp_len


TEST_SUITE = make_flask_test_suite(WebAccountTest, UserGroupTest)

if __name__ == "__main__":
    run_test_suite(TEST_SUITE)

示例#6

文件： test_messages.py 项目： PXke/invenio-demosite

        time.sleep(1)

        response = self.client.get("/yourmessages/", follow_redirects=True)
        assert 'sent_now' in response.data
        assert 'sent_later' in response.data

        # Avoid problems with foreign keys.
        UserMsgMESSAGE.query.delete(synchronize_session='fetch')
        MsgMESSAGE.query.delete(synchronize_session='fetch')
        db.session.commit()

    @fixture.with_data(UserData)
    def test_with_fixture(data, self):
        from invenio.modules.accounts.models import User
        u = User.query.all()
        assert len(u) == len(dict(data.UserData))
        db.session.commit()

    def test_without_fixture(self):
        from invenio.modules.accounts.models import User
        u = User.query.all()
        assert len(u) == 0


TEST_SUITE = make_flask_test_suite(MsgMESSAGETest)

if __name__ == "__main__":
    run_test_suite(TEST_SUITE)