def test_lab_to_dict(self): lab_doc = modl.LabDocument("foo") lab_doc.name = "bar" lab_doc.created_on = "now" lab_doc.updated_on = "now" lab_doc.id = "bar" lab_doc.address = { "street_1": "a", "street_2": "b", "city": "c", "country": "d", "zipcode": "e", "longitude": "f", "latitude": "h" } lab_doc.private = True lab_doc.token = "token" lab_doc.contact = { "name": "foo", "surname": "bar", "telephone": "1234", "mobile": "1234", "email": "*****@*****.**" } expected = { "name": "bar", "created_on": "now", "updated_on": "now", "_id": "bar", "address": { "street_1": "a", "street_2": "b", "city": "c", "country": "d", "zipcode": "e", "longitude": "f", "latitude": "h" }, "private": True, "token": "token", "contact": { "name": "foo", "surname": "bar", "telephone": "1234", "mobile": "1234", "email": "*****@*****.**" }, "version": None, } self.assertDictEqual(expected, lab_doc.to_dict())
def test_set_contact_missing_email(self): lab_doc = modl.LabDocument("foo") contact = {"name": "foo", "surname": "bar"} self.assertRaises(ValueError, setattr, lab_doc, "contact", contact)
def test_set_contact_only_surname(self): lab_doc = modl.LabDocument("foo") contact = {"surname": "bar"} self.assertRaises(ValueError, setattr, lab_doc, "contact", contact)
def test_set_contact_wrong_type(self): lab_doc = modl.LabDocument("foo") self.assertRaises(TypeError, setattr, lab_doc, "contact", "") self.assertRaises(TypeError, setattr, lab_doc, "contact", []) self.assertRaises(TypeError, setattr, lab_doc, "contact", ())
def test_set_contact_missing_valid_fields(self): lab_doc = modl.LabDocument("foo") self.assertRaises(ValueError, setattr, lab_doc, "contact", {})
def test_set_address_wrong_type(self): lab_doc = modl.LabDocument("foo") self.assertRaises(TypeError, setattr, lab_doc, "address", "") self.assertRaises(TypeError, setattr, lab_doc, "address", []) self.assertRaises(TypeError, setattr, lab_doc, "address", ())
def test_model_collection(self): lab_doc = modl.LabDocument("") self.assertEqual(lab_doc.collection, "lab")
def test_is_valid_base_class(self): self.assertIsInstance(modl.LabDocument(""), modb.BaseDocument)