Method setUp

tests/test_signals.py:21–207

(self)

        return signal_output

    def setUp(self):
        connect(db="mongoenginetest")

        class Author(Document):
            # Make the id deterministic for easier testing
            id = SequenceField(primary_key=True)
            name = StringField()

            def __unicode__(self):
                return self.name

            @classmethod
            def pre_init(cls, sender, document, *args, **kwargs):
                signal_output.append("pre_init signal, %s" % cls.__name__)
                signal_output.append(kwargs["values"])

            @classmethod
            def post_init(cls, sender, document, **kwargs):
                signal_output.append(
                    "post_init signal, %s, document._created = %s"
                    % (document, document._created)
                )

            @classmethod
            def pre_save(cls, sender, document, **kwargs):
                signal_output.append("pre_save signal, %s" % document)
                signal_output.append(kwargs)

            @classmethod
            def pre_save_post_validation(cls, sender, document, **kwargs):
                signal_output.append("pre_save_post_validation signal, %s" % document)
                if kwargs.pop("created", False):
                    signal_output.append("Is created")
                else:
                    signal_output.append("Is updated")
                signal_output.append(kwargs)

            @classmethod
            def post_save(cls, sender, document, **kwargs):
                dirty_keys = list(document._delta()[0].keys()) + list(
                    document._delta()[1].keys()
                )
                signal_output.append("post_save signal, %s" % document)
                signal_output.append("post_save dirty keys, %s" % dirty_keys)
                if kwargs.pop("created", False):
                    signal_output.append("Is created")
                else:
                    signal_output.append("Is updated")
                signal_output.append(kwargs)

            @classmethod
            def pre_delete(cls, sender, document, **kwargs):
                signal_output.append("pre_delete signal, %s" % document)
                signal_output.append(kwargs)

            @classmethod
            def post_delete(cls, sender, document, **kwargs):
                signal_output.append("post_delete signal, %s" % document)