What’s Not Supported

ReferenceField is not generically supported and probably will never be. You can, and should, provide an application-specific strategy for these fields. This permits you to ensure that the referential-integrity constraints needed by your application are satisfied. Don’t forget that MongoEngine expects the documents to have been saved to the database before you try to reference them. You can use the hypothesis_mongoengine.helpers.mark_saved function to make a document appear as if saved.

DictField is not generically supported and probably will never be. MapField is supported generically and should be preferred to DictField when the values are homogenous. When writing custom strategies for a DictField, you can use the hypothesis_mongoengine.strategies.mongodb_keys strategy to generate the keys in the absence of more specific application knowledge about the keys.

DynamicDocument (and DynamicEmbeddedDocument) currently generate only the explicitly-specified fields.

DynamicField is normally used internally by DynamicDocument, but if you have a model which references it explicitly, it won’t be handled generically.

Handling Custom Fields

If you have a custom field in use in your application, you can register a strategy to generate examples for it using the field_strategy decorator.

For example, a strategy for the EnumField from extras-mongoengine could look like this:

from extras_mongoengine.fields import EnumField
from hypothesis import strategies
from hypothesis_mongoengine.strategies import field_strategy

@field_strategy(EnumField)
def my_custom_strat(field):
    return strategies.sampled_from(field.enum)

The fields are looked up in the registry by equality of the classes, so if you have a hierarchy of custom fields, you must register the leaf types. You can, however, stack the decorator several times if you need to:

from extras_mongoengine.fields import EnumField, IntEnumField, StringEnumField
from hypothesis import strategies
from hypothesis_mongoengine.strategies import field_strategy

@field_strategy(EnumField)
@field_strategy(IntEnumField)
@field_strategy(StringEnumField)
def my_custom_strat(field):
    return strategies.sampled_from(field.enum)