Creates formset data to replicate data posted by Django forms. This helps to generate data while writing test cases

test_data_generator.py

def instantiate_formset_data(formset_class, data, instances=[], initial=None, action="add"):
    """
    Input: 
    Formset_class 
    data - Python Dict type form data
    Instances - [] if its a CreateForm, or list of instances in case of EditForm
    Initial - Initial  data if any
    Action = define edit/Add action
    
    Output: data --> Formset style Data
    """
    prefix = formset_class().prefix
    formset_data = {}
    for i, instance in enumerate(instances):
        formset_data['{}-{}-{}'.format(prefix, i, "id")] = [instance.id]
    if not instances:
        formset_data['{}-{}-{}'.format(prefix, 0, "id")] = [""]


    for i, form_data in enumerate(data):
        for name, value in form_data.items():
            formset_data['{}-{}-{}'.format(prefix, i, name)] = value
    formset_data['{}-TOTAL_FORMS'.format(prefix)] = len(data)
    formset_data['{}-INITIAL_FORMS'.format(prefix)] = 1 if action == "edit" else 0
    return formset_data