UUIDModels for testing purposes

This commit is contained in:
JMP 2019-02-26 17:35:50 +01:00
parent bdc6fb05fe
commit 5ff0867bf9

View file

@ -8,7 +8,12 @@ from django.utils.encoding import python_2_unicode_compatible
from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ugettext_lazy as _
from model_utils import Choices from model_utils import Choices
from model_utils.fields import SplitField, MonitorField, StatusField from model_utils.fields import (
SplitField,
MonitorField,
StatusField,
UUIDField,
)
from model_utils.managers import ( from model_utils.managers import (
QueryManager, QueryManager,
InheritanceManager, InheritanceManager,
@ -19,6 +24,7 @@ from model_utils.models import (
StatusModel, StatusModel,
TimeFramedModel, TimeFramedModel,
TimeStampedModel, TimeStampedModel,
UUIDModel,
) )
from tests.fields import MutableField from tests.fields import MutableField
from tests.managers import CustomSoftDeleteManager from tests.managers import CustomSoftDeleteManager
@ -159,8 +165,8 @@ class Post(models.Model):
objects = models.Manager() objects = models.Manager()
public = QueryManager(published=True) public = QueryManager(published=True)
public_confirmed = QueryManager(models.Q(published=True) & public_confirmed = QueryManager(models.Q(published=True)
models.Q(confirmed=True)) & models.Q(confirmed=True))
public_reversed = QueryManager(published=True).order_by("-order") public_reversed = QueryManager(published=True).order_by("-order")
class Meta: class Meta:
@ -340,6 +346,7 @@ class StringyDescriptor(object):
""" """
Descriptor that returns a string version of the underlying integer value. Descriptor that returns a string version of the underlying integer value.
""" """
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
@ -393,3 +400,11 @@ class JoinItemForeignKey(models.Model):
on_delete=models.CASCADE on_delete=models.CASCADE
) )
objects = JoinManager() objects = JoinManager()
class CustomUUIDModel(UUIDModel):
pass
class CustomNotPrimaryUUIDModel(models.Model):
uuid = UUIDField(primary_key=False)