def test_init_arg_exception_obj(self): class Tester(object): def __init__(self, id=None, bad_test="test"): self.id = id self.bad_test = bad_test obj = Tester() with self.assertRaises(AttributeError): AgentProfileDocument(obj)
def test_init_partial(self): doc = AgentProfileDocument(id="test", content_type="test type") self.assertEqual(doc.id, "test") self.assertEqual(doc.content_type, "test type") self.assertTrue(hasattr(doc, "content")) self.assertTrue(hasattr(doc, "etag")) self.assertTrue(hasattr(doc, "timestamp")) self.assertTrue(hasattr(doc, "agent"))
def test_setters_none(self): doc = AgentProfileDocument() doc.id = None doc.content_type = None doc.content = None doc.etag = None doc.timestamp = None doc.agent = None self.assertIsNone(doc.id) self.assertIsNone(doc.content_type) self.assertIsNone(doc.content) self.assertIsNone(doc.etag) self.assertIsNone(doc.timestamp) self.assertIsNone(doc.agent)
def test_setters(self): doc = AgentProfileDocument() doc.id = "test" doc.content_type = "test type" doc.content = bytearray("test bytearray", "utf-8") doc.etag = "test etag" doc.timestamp = "2014-06-23T15:25:00-05:00" doc.agent = self.agent self.assertEqual(doc.id, "test") self.assertEqual(doc.content_type, "test type") self.assertEqual(doc.content, bytearray("test bytearray", "utf-8")) self.assertEqual(doc.etag, "test etag") central = pytz.timezone("US/Central") # UTC -0500 dt = central.localize(datetime(2014, 6, 23, 15, 25)) self.assertEqual(doc.timestamp, dt) self.assertEqual(doc.agent, self.agent)
def test_init_empty(self): doc = AgentProfileDocument() self.assertIsInstance(doc, AgentProfileDocument) self.assertTrue(hasattr(doc, "id")) self.assertIsNone(doc.id) self.assertTrue(hasattr(doc, "content_type")) self.assertIsNone(doc.content_type) self.assertTrue(hasattr(doc, "content")) self.assertIsNone(doc.content) self.assertTrue(hasattr(doc, "etag")) self.assertIsNone(doc.etag) self.assertTrue(hasattr(doc, "timestamp")) self.assertIsNone(doc.timestamp) self.assertTrue(hasattr(doc, "agent")) self.assertIsNone(doc.agent)
def test_init_arg_exception_dict(self): d = {"bad_test": "test", "id": "ok"} with self.assertRaises(AttributeError): AgentProfileDocument(d)
def test_init_kwarg_exception(self): with self.assertRaises(AttributeError): AgentProfileDocument(bad_test="test")
def test_agent_setter(self): doc = AgentProfileDocument() doc.agent = {"mbox": "mailto:[email protected]"} self.assertIsInstance(doc.agent, Agent) self.assertEqual(doc.agent.mbox, self.agent.mbox)