def test_init_partial(self): doc = Document(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"))
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): Document(obj)
def test_init_empty(self): doc = Document() self.assertIsInstance(doc, Document) 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)
def test_init_all(self): doc = Document(id="test", content_type="test type", content=bytearray("test bytearray", "utf-8"), etag="test etag", timestamp="2014-06-23T15:25:00-05:00") 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.datetime(2014, 6, 23, 15, 25)) self.assertEqual(doc.timestamp, dt)
def test_setters_none(self): doc = Document() doc.id = None doc.content_type = None doc.content = None doc.etag = None doc.timestamp = None self.assertIsNone(doc.id) self.assertIsNone(doc.content_type) self.assertIsNone(doc.content) self.assertIsNone(doc.etag) self.assertIsNone(doc.timestamp)
def test_setters(self): doc = Document() 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" 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.datetime(2014, 6, 23, 15, 25)) self.assertEqual(doc.timestamp, dt)
def test_init_arg_exception_dict(self): d = {"bad_test": "test", "id": "ok"} with self.assertRaises(AttributeError): Document(d)
def test_init_kwarg_exception(self): with self.assertRaises(AttributeError): Document(bad_test="test")
def test_content_timestamp(self): doc = Document() dt = pytz.utc.localize(datetime.datetime.utcnow()) doc.timestamp = dt self.assertEqual(doc.timestamp, dt)
def test_content_setter(self): doc = Document() doc.content = "test bytearray" self.assertEqual(doc.content, bytearray("test bytearray", "utf-8"))