70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
import pytest
|
|
from bson.int64 import Int64
|
|
|
|
from mongoengine import *
|
|
from mongoengine.connection import get_db
|
|
from tests.utils import get_as_pymongo, MongoDBTestCase
|
|
|
|
|
|
class TestLongField(MongoDBTestCase):
|
|
def test_storage(self):
|
|
class Person(Document):
|
|
value = LongField()
|
|
|
|
Person.drop_collection()
|
|
person = Person(value=5000)
|
|
person.save()
|
|
assert get_as_pymongo(person) == {"_id": person.id, "value": 5000}
|
|
|
|
def test_construction_does_not_fail_with_invalid_value(self):
|
|
class Person(Document):
|
|
value = LongField()
|
|
|
|
person = Person(value="not_an_int")
|
|
assert person.value == "not_an_int"
|
|
|
|
def test_long_field_is_considered_as_int64(self):
|
|
"""
|
|
Tests that long fields are stored as long in mongo, even if long
|
|
value is small enough to be an int.
|
|
"""
|
|
|
|
class TestLongFieldConsideredAsInt64(Document):
|
|
some_long = LongField()
|
|
|
|
doc = TestLongFieldConsideredAsInt64(some_long=42).save()
|
|
db = get_db()
|
|
assert isinstance(
|
|
db.test_long_field_considered_as_int64.find()[0]["some_long"], Int64
|
|
)
|
|
assert isinstance(doc.some_long, int)
|
|
|
|
def test_long_validation(self):
|
|
"""Ensure that invalid values cannot be assigned to long fields."""
|
|
|
|
class TestDocument(Document):
|
|
value = LongField(min_value=0, max_value=110)
|
|
|
|
TestDocument(value=50).validate()
|
|
|
|
with pytest.raises(ValidationError):
|
|
TestDocument(value=-1).validate()
|
|
|
|
with pytest.raises(ValidationError):
|
|
TestDocument(value=120).validate()
|
|
|
|
with pytest.raises(ValidationError):
|
|
TestDocument(value="ten").validate()
|
|
|
|
def test_long_ne_operator(self):
|
|
class TestDocument(Document):
|
|
long_fld = LongField()
|
|
|
|
TestDocument.drop_collection()
|
|
|
|
TestDocument(long_fld=None).save()
|
|
TestDocument(long_fld=1).save()
|
|
|
|
assert TestDocument.objects(long_fld__ne=None).count() == 1
|
|
assert TestDocument.objects(long_fld__ne=1).count() == 1
|