def setUp(self):
"""Set up test."""
from base64 import b64encode, b64decode
from bson.binary import Binary, BINARY_SUBTYPE
class BinaryTest(db.Document):
text = db.BinaryField()
self.model_cls = BinaryTest
self.data = {
"text": {
"data": b64encode(
("This is a test").encode("utf-8")
).decode("utf-8"),
"type": BINARY_SUBTYPE
}
}
self.expected_data = {
"text": Binary(
b64decode(self.data["text"]["data"]),
self.data["text"]["type"]
)
}
self.data = json.dumps(self.data)
self.hook = generate_object_hook(BinaryTest)
def setUp(self):
"""Setup test."""
from base64 import b64encode, b64decode
from bson.binary import Binary, BINARY_SUBTYPE
class BinaryTest(db.Document):
text = db.BinaryField()
self.model_cls = BinaryTest
self.data = {
"text": {
"data":
b64encode(("This is a test").encode("utf-8")).decode("utf-8"),
"type":
BINARY_SUBTYPE
}
}
self.expected_data = {
"text":
Binary(b64decode(self.data["text"]["data"]),
self.data["text"]["type"])
}
self.data = json.dumps(self.data)
self.hook = generate_object_hook(BinaryTest)
def setUp(self):
"""Setup class."""
from bson import ObjectId
class TestModel(db.Document):
user = db.ObjectIdField()
self.model_cls = TestModel
self.hook = generate_object_hook(self.model_cls)
self.data = json.dumps({"user": "******"})
self.expected_data = {"user": ObjectId("56f63a716a8dec7705f36409")}
def setUp(self):
"""Set up class."""
from bson import ObjectId
class TestModel(db.Document):
user = db.ObjectIdField()
self.model_cls = TestModel
self.hook = generate_object_hook(self.model_cls)
self.data = json.dumps({"user": "******"})
self.expected_data = {"user": ObjectId("56f63a716a8dec7705f36409")}
def setUp(self):
"""Setup test."""
from uuid import uuid5, NAMESPACE_DNS
class UUIDModel(db.Document):
uuid = db.UUIDField()
self.model_cls = UUIDModel
uuid = uuid5(NAMESPACE_DNS, "This is a test")
self.expected_data = {"uuid": uuid}
self.data = json.dumps({"uuid": str(uuid)})
self.hook = generate_object_hook(UUIDModel)
def setUp(self):
"""Setup test."""
from datetime import datetime
from dateutil.parser import parse
class DateTime(db.Document):
date = db.DateTimeField()
self.model_cls = DateTime
now = datetime.utcnow()
self.data = json.dumps({"date": now.isoformat()})
self.expected_data = {"date": parse(now.isoformat())}
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Set up test."""
from datetime import datetime
from dateutil.parser import parse
class DateTime(db.Document):
date = db.DateTimeField()
self.model_cls = DateTime
now = datetime.utcnow()
self.data = json.dumps({"date": now.isoformat()})
self.expected_data = {"date": parse(now.isoformat())}
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Set up test."""
from uuid import uuid5, NAMESPACE_DNS
class UUIDModel(db.Document):
uuid = db.UUIDField()
self.model_cls = UUIDModel
uuid = uuid5(NAMESPACE_DNS, "This is a test")
self.expected_data = {
"uuid": uuid
}
self.data = json.dumps({"uuid": str(uuid)})
self.hook = generate_object_hook(UUIDModel)
def setUp(self):
"""Setup class."""
from bson import ObjectId, DBRef
class Source(db.Document):
pass
class Model(db.Document):
src = db.ReferenceField(Source)
self.src_cls = Source
self.model_cls = Model
self.src_id = ObjectId()
self.data = json.dumps({"src": str(self.src_id)})
self.expected_data = {"src": DBRef("source", self.src_id)}
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Set up test."""
from datetime import datetime
class DateTime(db.Document):
date = db.DateTimeField()
self.model_cls = DateTime
now = datetime.utcnow()
# This format shouldn't be supported.
self.data = json.dumps({"date": {
"year": now.year,
"month": now.month,
"date": now.day
}})
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Setup test."""
from datetime import datetime
class DateTime(db.Document):
date = db.DateTimeField()
self.model_cls = DateTime
now = datetime.utcnow()
# This format shouldn't be supported.
self.data = json.dumps(
{"date": {
"year": now.year,
"month": now.month,
"date": now.day
}})
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Set up class."""
from bson import DBRef, ObjectId
class Source(db.Document):
pass
class Model(db.Document):
src = db.ReferenceField(Source, dbref=True)
self.src_cls = Source
self.model_cls = Model
self.src_id = ObjectId()
self.data = json.dumps({
"src": {"collection": "source", "id": str(self.src_id)}
})
self.expected_data = {"src": DBRef("source", self.src_id)}
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Set up test."""
from datetime import datetime, timedelta
from calendar import timegm
class DateTime(db.Document):
date = db.DateTimeField()
self.model_cls = DateTime
now = datetime.utcnow()
epoch_mil = int(timegm(now.timetuple())*1000 + now.microsecond / 1000)
self.data = json.dumps({"date": epoch_mil})
self.expected_data = {
"date": datetime.utcfromtimestamp(
int(epoch_mil / 1000)
) + timedelta(milliseconds=int(epoch_mil % 1000))
}
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Setup test."""
from datetime import datetime, timedelta
from calendar import timegm
class DateTime(db.Document):
date = db.DateTimeField()
self.model_cls = DateTime
now = datetime.utcnow()
epoch_mil = int(
timegm(now.timetuple()) * 1000 + now.microsecond / 1000)
self.data = json.dumps({"date": epoch_mil})
self.expected_data = {
"date":
datetime.utcfromtimestamp(int(epoch_mil / 1000)) +
timedelta(milliseconds=int(epoch_mil % 1000))
}
self.hook = generate_object_hook(self.model_cls)
def setUp(self):
"""Setup class."""
self.hook = generate_object_hook(None)
self.expected_data = {"user": "******"}
self.data = json.dumps(self.expected_data)
def setUp(self):
"""Set up class."""
self.hook = generate_object_hook(None)
self.expected_data = {"user": "******"}
self.data = json.dumps(self.expected_data)