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)