Update test models to generate random PK

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-03-02 17:32:38 -04:00
parent 0f84b7a723
commit 442faca915

View File

@@ -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)