save a field as JSON using factoryboy

Question

I am trying to create an instance of the model in which one of the field is JSONField. When creating the instance of model via factoryboy, I want to save the field as JSONField in the test database. On simply passing the field as JSON,it gets stored in Unicode data type. I am stuck on this.

Answer 1

You can create dict data using a factory.Dict. You can then control the final form of the data using the dict_factory attribute.

e.g. if you want to have dict data that is serialised to a JSON string, you can do something like this:

import json
import factory


class JSONFactory(factory.DictFactory):
    """
    Use with factory.Dict to make JSON strings.
    """
    @classmethod
    def _generate(cls, create, attrs):
        obj = super()._generate(create, attrs)
        return json.dumps(obj)


class BadgerFactory(factory.Factory):

    form_data = factory.Dict({
        'badger': ['stoat'],
    }, dict_factory=JSONFactory)

    class Meta:
        model = ...