Produits Entr'ouvert » w.c.s.

import cPickle

from utilities import create_temporary_pub, clean_temporary_pub

from qommon.storage import StorableObject

from wcs.roles import Role

def setup_module(module):

    pub = create_temporary_pub()

    pub.cfg['language'] = {'language': 'en'}

def teardown_module(module):

    clean_temporary_pub()

def test_slug():

    role = Role(name='Hello world')

    role.store()

    assert role.slug == 'hello-world'

def test_migrate():

    role = Role(name='Hello world')

    role.store()

    obj = cPickle.load(open(role.get_object_filename()))

    del obj.slug

    cPickle.dump(obj, open(role.get_object_filename(), 'w'))

    assert cPickle.load(open(role.get_object_filename())).slug is None

    assert Role.get(role.id).slug == 'hello-world'

    slug = None

    def migrate(self):

        changed = False

        if not self.slug:

            # .store() will take care of setting the slug

            changed = True

        if changed:

            self.store()

    def store(self):

        if self.slug is None:

            # set slug if it's not yet there

            self.slug = qommon.misc.simplify(self.name)

        super(Role, self).store()