def to_son(self):
field = UUIDField()
uuid = uuid4()
expect(field.to_son(uuid)).to_equal(uuid)
expect(field.to_son(str(uuid))).to_equal(uuid)
def test_validate_enforces_uuid(self):
field = UUIDField()
uuid = uuid4()
expect(field.validate("123")).to_be_false()
expect(field.validate(uuid)).to_be_true()
expect(field.validate(str(uuid))).to_be_true()
def test_is_empty(self):
field = UUIDField()
uuid = uuid4()
expect(field.is_empty(uuid)).to_be_false()
expect(field.is_empty("")).to_be_true()
expect(field.is_empty(None)).to_be_true()
def test_to_son(self):
field = UUIDField()
uuid = uuid4()
expect(field.to_son(uuid)).to_equal(uuid)
expect(field.to_son(str(uuid))).to_equal(uuid)
expect(field.to_son(None)).to_be_null()
class User(Document):
__collection__ = 'users'
id = UUIDField()
name = StringField()
email = EmailField(required=True, unique=True)
password = StringField(required=True)
date_added = DateTimeField(required=True, auto_now_on_insert=True)
class Issue(Document):
__collection__ = 'issues'
id = UUIDField()
title = StringField(required=True)
content = StringField(required=True)
level = StringField(required=True)
date_added = DateTimeField(required=True, auto_now_on_insert=True)
date_updated = DateTimeField(required=True, auto_now_on_update=True)
user = ReferenceField(User)
class LoadedPost(Document):
uuid = UUIDField(default=uuid4)
def test_create_uuid_field(self):
field = UUIDField(db_field="test")
expect(field.db_field).to_equal("test")
class CreateDeviceAction(Document):
__collection__ = "create_device_actions"
uid = UUIDField()
serial_number = StringField()
class AssignmentInfoAction(Document):
__collection__ = "assignment_info"
device_uid = UUIDField()
patient_uid = UUIDField()
assignment_type = IntField()
def test_none_from_son(self):
field = UUIDField()
expect(field.from_son(None)).to_be_null()
class CreatePatientAction(Document):
__collection__ = "create_patient_actions"
uid = UUIDField()
first_name = StringField()
last_name = StringField()
class EditPatientAction(Document):
__collection__ = "edit_patient_actions"
patientUid = UUIDField()
new_first_name = StringField()
new_last_name = StringField()
contact_info = EmbeddedDocumentField(embedded_document_type=Document)
