From 442faca9151d54d31e1a90280e8a067c08f3b8eb Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Sat, 2 Mar 2019 17:32:38 -0400 Subject: [PATCH] Update test models to generate random PK Signed-off-by: Roberto Rosario --- mayan/apps/common/tests/mixins.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/mayan/apps/common/tests/mixins.py b/mayan/apps/common/tests/mixins.py index 16401b70c7..00edab587f 100644 --- a/mayan/apps/common/tests/mixins.py +++ b/mayan/apps/common/tests/mixins.py @@ -239,11 +239,24 @@ class TestModelTestMixin(object): class Meta: pass + def save(instance, *args, **kwargs): + if instance.pk: + return models.Model.self(instance, *args, **kwargs) + else: + instance.pk = RandomPrimaryKeyModelMonkeyPatchMixin.get_unique_primary_key( + model=instance._meta.model + ) + instance.id = instance.pk + + return instance.save_base(force_insert=True) + if options is not None: for key, value in options.items(): setattr(Meta, key, value) - attrs = {'__module__': self.__class__.__module__, 'Meta': Meta} + attrs = { + '__module__': self.__class__.__module__, 'save': save, 'Meta': Meta + } if fields: attrs.update(fields)