improvements of test coverage in fields

This commit is contained in:
Bastien Gerard
2020-12-18 00:05:28 +01:00
parent 9a3bca8ab6
commit 98d64f41c6
8 changed files with 169 additions and 112 deletions

View File

@@ -3,7 +3,7 @@ from enum import Enum
from bson import InvalidDocument
import pytest
from mongoengine import *
from mongoengine import Document, EnumField, ValidationError
from tests.utils import MongoDBTestCase, get_as_pymongo
@@ -45,6 +45,11 @@ class TestStringEnumField(MongoDBTestCase):
m.save()
assert m.status == Status.DONE
m.status = "wrong"
assert m.status == "wrong"
with pytest.raises(ValidationError):
m.validate()
def test_set_default(self):
class ModelWithDefault(Document):
status = EnumField(Status, default=Status.DONE)