def test_can_find_by_id(self):
doc = MongoMintDocument(db.connection, "mint_object_upsert")
genius = "Spencer Reid"
model1 = dict(name=genius)
doc.upsert(model1)
id = model1["_id"]
assert(type(id) == ObjectId)
# Triial case, already right type:
model2 = doc.find_by_id(id)
assert(model2["name"] == genius)
# Validate can load by string
# Convert to string first
model3 = doc.find_by_id(str(id))
assert(model3["name"] == genius)
doc.collection.drop()
def test_can_do_object_upsert(self):
doc = MongoMintDocument(db.connection, "mint_object_upsert")
model1 = dict(name="John")
assert("_id" not in model1)
doc.upsert(model1)
assert("_id" in model1)
model1["name"] = "Jenniffer"
doc.upsert(model1)
# JCL todo wrap in method
model2 = doc.collection.find_one({"_id" : model1["_id"]})
assert(model2["_id"] == model1["_id"])
assert(model2["name"] == "Jenniffer")
doc.collection.drop()